elton-concurrent-limiter

The middleware has been archived, please use the middleware of elton.

Concurrent limiter for elton. It support to get lock value from five ways. Client IP, QueryString, Request Header, Route Params and Post Body.

• IP The key's name is :ip
• QueryString The key's name has prefix q:
• Request Header The key's name has prefix h:
• Route Params The key's name has prefix p:
• Post Body The other's key

package main

import (
	"bytes"
	"sync"
	"time"

	"github.com/vicanso/elton"

	concurrentLimiter "github.com/vicanso/elton-concurrent-limiter"
)

func main() {

	e := elton.New()
	m := new(sync.Map)
	limit := concurrentLimiter.New(concurrentLimiter.Config{
		Keys: []string{
			":ip",
			"h:X-Token",
			"q:type",
			"p:id",
			"account",
		},
		Lock: func(key string, c *elton.Context) (success bool, unlock func(), err error) {
			_, loaded := m.LoadOrStore(key, true)
			// the key not exists
			if !loaded {
				success = true
				unlock = func() {
					m.Delete(key)
				}
			}
			return
		},
	})

	e.POST("/login", limit, func(c *elton.Context) (err error) {
		time.Sleep(3 * time.Second)
		c.BodyBuffer = bytes.NewBufferString("hello world")
		return
	})

	err := e.ListenAndServe(":3000")
	if err != nil {
		panic(err)
	})
}

curl -XPOST 'http://127.0.0.1:7001/login'