Sentinel

Introduction

Hertz provides hertz-contrib/opensergo, to facilitate the integration of sentinel-golang

Installation

go get github.com/hertz-contrib/opensergo

Config

sentinel-golang

The basic configuration of sentinel-golang can be found at documentation

Server

SentinelServerMiddleware

SentinelServerMiddleware() returns new app.HandlerFunc

Default resource name is {method}:{path}, such as “GET:/api/users/:id”

Default block fallback is returning 429 code

Define your own behavior by WithServerXxx()

Sample Code:

package main

// ...

func main() {
	h := server.Default(server.WithHostPorts(":8081"))
	h.Use(adaptor.SentinelServerMiddleware())
	// ...
}

WithServerResourceExtractor

WithServerResourceExtractor sets the resource extractor of the web requests for server side.

Sample Code:

package main

// ...

func main() {
	h := server.Default(server.WithHostPorts(":8081"))
	h.Use(adaptor.SentinelServerMiddleware(
		// customize resource extractor if required
		// method_path by default
		adaptor.WithServerResourceExtractor(func(c context.Context, ctx *app.RequestContext) string {
			return "server_test"
		}),
	))
	// ...
}

WithServerBlockFallback

WithServerBlockFallback sets the fallback handler when requests are blocked for server side.

Sample Code:

package main

// ...

func main() {
	h := server.Default(server.WithHostPorts(":8081"))
	h.Use(adaptor.SentinelServerMiddleware(
		// customize block fallback if required
		// abort with status 429 by default
		adaptor.WithServerBlockFallback(func(c context.Context, ctx *app.RequestContext) {
			ctx.AbortWithStatusJSON(400, utils.H{
				"err":  "too many request; the quota used up",
				"code": 10222,
			})
		}),
	))
	// ...
}

Client

SentinelClientMiddleware

SentinelClientMiddleware() returns new client.Middleware Default resource name is {method}:{path}, such as “GET:/api/users” Default block fallback is returning blockError Define your own behavior by WithClientXxx()

Sample Code:

package main

// ...

func main() {
	c, err := client.NewClient()
	if err != nil {
		log.Fatalf("Unexpected error: %+v", err)
		return
	}

	c.Use(adaptor.SentinelClientMiddleware())
}

WithClientResourceExtractor

WithClientResourceExtractor sets the resource extractor of the web requests for client side.

Sample Code:

package main

// ...

func main() {
	c, err := client.NewClient()
	if err != nil {
		log.Fatalf("Unexpected error: %+v", err)
		return
	}

	c.Use(adaptor.SentinelClientMiddleware(
		// customize resource extractor if required
		// method_path by default
		adaptor.WithClientResourceExtractor(func(ctx context.Context, request *protocol.Request, response *protocol.Response) string {
			return "client_test"
		}),
	))
}

WithClientBlockFallback

WithClientBlockFallback sets the fallback handler when requests are blocked for client side.

Sample Code:

package main

// ...

func main() {
	c, err := client.NewClient()
	if err != nil {
		log.Fatalf("Unexpected error: %+v", err)
		return
	}

	c.Use(adaptor.SentinelClientMiddleware(
		// customize resource extractor if required
		// method_path by default
		adaptor.WithClientBlockFallback(func(ctx context.Context, req *protocol.Request, resp *protocol.Response, blockError error) error {
			resp.SetStatusCode(http.StatusBadRequest)
			resp.SetBody([]byte("request failed"))
			return blockError
		}),
	))
}

Complete sample code

Full usage examples are available at example


Last modified August 2, 2023 : chore: update readme (#742) (945d264)