Skip to content

Throttling, rate-limiting and synchronization for distributed applications.

License

Notifications You must be signed in to change notification settings

pjwerneck/bouncer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

70 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

bouncer

Go Report Card Build Status

bouncer logo

This is a simple RPC service to provide throttling, rate-limiting, and synchronization for distributed applications. It's intended as a replacement for makeshift solutions using memcached or Redis.

Examples

Start a local container:

$ docker run -p 5505:5505 pjwerneck/bouncer:latest

"I want to limit something to only one operation per second"

You need a token bucket. Just do this before each operation:

$ curl http://myhost:5505/tokenbucket/myapp/acquire

"How to increase the limit to twenty per second?"

Use size=20:

$ curl http://myhost:5505/tokenbucket/myapp/acquire?size=20

"But I can't have all twenty starting at the same time!"

If you don't want bursts of activity, set interval to 1000/rate:

$ curl http://myhost:5505/tokenbucket/myapp/acquire?interval=50

"What if I have a resource that can be used by only one client at a time?"

Use a semaphore:

$ KEY=$(curl http://myhost:5505/semaphore/myapp/acquire)
$ # do something
$ curl http://myhost:5505/semaphore/myapp/release?key=$KEY

"Now I need to limit it to ten concurrent clients."

Use a semaphore with size=10:

$ KEY=$(curl http://myhost:5505/semaphore/myapp/acquire?size=10)
$ # do something
$ curl http://myhost:5505/semaphore/myapp/release?key=$KEY

"I have some clients that must wait for something else to finish."

You can use an event:

$ # do this on any number of waiting clients
$ curl http://myhost:5505/event/myapp/wait
$ # and this on the client doing the operation, when it's finished
$ curl http://myhost:5505/event/myapp/send?message=wakeup

API

Requests

All requests use the GET method. All arguments are sent using query string parameters. A controller can be easily integrated into anything that can perform an HTTP request.

A request automatically creates a controller if it doesn't exist. If it already exists and the parameters are different, the request updates the controller automatically.

Parameters

The parameters are normalized for all controllers. It's not an error to pass an unneeded parameter, but it's an error to pass an unknown parameter.

All time values are in milliseconds. All numeric values are integers.

expires=[integer]

The expiration time for semaphore and watchdog controllers. Default is 60000, one minute.

interval=[integer]

The time interval between tokenbucket refills. Default is 1000, one second.

key=[alphanumeric]

The unique value used by semaphore to release a hold.

maxwait=[integer]

The max time to wait for a response. The default is -1, wait forever. A value of zero never blocks, returning immediately.

You should use maxwait=0 if you want to return an error immediately instead of waiting for availability.

size=[integer]

The size of tokenbucket and semaphore. The default is 1.

You can use size=0 if you want to stop all activity, but keep in mind that a request with size=0 will always timeout, unless another request with size>0 resizes the controller.

Resizing a controller affects its current state. For instance, if you reduce or increase the size of a tokenbucket, it will take into account the tokens already acquired in the current interval. If you reduce the size of a semaphore, a client won't be able to acquire a hold until the extra clients are released.

Responses

Status codes are very specific so clients can use them to understand the response for valid requests without parsing the response body.

200 OK

For succesful requests that return some value.

204 No Content

For successful requests that don't return any value.

400 Bad Request

An invalid request, e.g., missing a required parameter, invalid value, etc.

408 Request Timeout

The maxwait value was exceeded while waiting for a response.

409 Conflict

The current state of the controller is incompatible with this request.

Token Bucket

The tokenbucket is an implementation of the Token Bucket algorithm. The bucket has a limited size, and every interval the bucket is refilled to capacity with tokens. Each acquire request takes a token out of the bucket, or waits for a token to be added if the bucket is empty.

Acquire

/tokenbucket/<name>/acquire <size=1> <interval=1000> <maxwait=-1>

In most cases you can simply set size to the desired number of requests per second.

The size value must be an integer. If you need a fractional ratio of requests per second, you should reduce the fraction and set size and interval accordingly. Keep in mind that all tokens are added at once, and a naive conversion might result in long wait times. For instance, if you want 10 requests per minute use size=1&interval=6000, not size=10&interval=60000.

Bursts of activity can happen if there are many clients waiting for a refill. If that's undesirable, you can reduce size and interval by a common factor. You can go as far as setting size=1 and use only the interval to control the average rate, but keep in mind that setting high rates this way can result in significantly higher server CPU load.

Responses:

  • 204 No Content, if successful.
  • 408 Request Timeout, if maxwait was exceeded.

Semaphore

A semaphore can be used to control concurrent access to shared resources.

Acquire

/semaphore/<name>/acquire <size=1> <key=?> <expires=60000> <maxwait=-1>

A semaphore has a number of slots equal to size. An acquire request stores the key value in the next available slot. If there are no available slots, the request waits until maxwait.

If key is not provided, a random UUID is generated.

If expires is provided, the hold is automatically released after the given time. You should provide a reasonable value if there's the possibility of a client never releasing it. With a value of zero the hold never expires and must be released explicitly.

Reusing a key that's already being held doesn't result in an error and will return a succesful response, but it doesn't reset the expiration timer.

Responses:

  • 200 OK, if successful. The key value is returned on the response body.
  • 408 Request Timeout, if maxwait was exceeded.

Release

/semaphore/<name>/release <key>

Releases the previously acquired hold. A release always returns immediately.

Responses:

  • 204 No Content, if successful.
  • 409 Conflict, there's no current hold for the given key.

Event

An event can be used to synchronize clients, when you want all of them to start doing something immediately after a signal.

Wait

/event/<name>/wait <maxwait=-1>

Keeps the client waiting for a signal.

Responses:

  • 204 No Content, if the signal was received.
  • 408 Request Timeout, if maxwait was exceeded.

Send

/event/<name>/send

Sends the signal to all waiting requests.

Responses:

  • 204 No Content, if successful
  • 409 Conflict, if already sent.

Watchdog

A watchdog can be used to synchronize clients to do something when a recurring request takes too long.

Wait

/watchdog/<name>/wait <maxwait=-1>

Keeps the client waiting for a signal

Responses:

  • 204 No Content, if the signal was received.
  • 408 Request Timeout, if maxwait was exceeded.

Kick

/watchdog/<name>/kick <expires=60000>

Resets the watchdog timer. A signal will be sent to the clients if another kick isn't received within the expires time.

Responses:

  • 204 No Content always.

Health Check

/.well-known/ready

Returns a 200 OK if the server is running. This is useful for health checks.

Responses:

  • 200 OK always with the text "I'm ready!"

Environment Variables

The following environment variables can be used to customize server behavior.

BOUNCER_HOST

BOUNCER_HOST=0.0.0.0

The IP address the server will listen to. Defaults to 0.0.0.0, i.e. all IP addresses on the host machine.

BOUNCER_PORT

BOUNCER_PORT=5505

The port the server will listen to. Defaults to 5505.

BOUNCER_LOGLEVEL

BOUNCER_LOGLEVEL=INFO

The server log level. Accepted values are:

  • DEBUG
  • INFO
  • NOTICE
  • WARNING
  • ERROR
  • CRITICAL

BOUNCER_READ_TIMEOUT

BOUNCER_READ_TIMEOUT=30

The HTTP server read timeout, in seconds. Defaults to 30.

BOUNCER_WRITE_TIMEOUT

BOUNCER_WRITE_TIMEOUT=30

The HTTP server write timeout, in seconds. Defaults to 30.

If your controllers require a maxwait time that exceeds the default write timeout of 30 seconds, increase the timeout value accordingly, otherwise the server might close the connection before the response is available.

About

Throttling, rate-limiting and synchronization for distributed applications.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published