Go-Micro服务熔断初步使用
简介Go-Micro服务熔断初步使用
1、服务熔断库安装
go get github.com/afex/hystrix-go/hystrix
2、客户端熔断添加
package main
import (
"context"
"github.com/afex/hystrix-go/hystrix"
"github.com/gin-gonic/gin"
"github.com/micro/go-micro"
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/web"
"github.com/micro/go-plugins/registry/consul"
"log"
services "micro-home/service/model"
"strconv"
)
func InitMiddleware(service services.StudentListService) gin.HandlerFunc {
return func(context *gin.Context) {
context.Keys = make(map[string]interface{})
context.Keys["student_service"] = service
context.Next()
}
}
//装饰器
type logWrapper struct {
client.Client
}
//每次调用都会执行这个方法
func (c *logWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
log.Print("请求")
return c.Client.Call(ctx, req, rsp)
}
func NewLogWrapper(c client.Client) client.Client {
return &logWrapper{c}
}
func main() {
register := consul.NewRegistry(func(options *registry.Options) {
options.Addrs = []string{
"192.168.1.171:8500",
"192.168.1.177:8500",
"192.168.1.178:8500",
}
})
//微服务需要注册
homeService := micro.NewService(
micro.Name("micro-home-client"),
micro.Registry(register),
micro.WrapClient(NewLogWrapper), //在注册时只需要传入方法名即可,底层会自动给这个方法传入client
)
//实例化服务
studentListService := services.NewStudentListService("student-service", homeService.Client())
routers := gin.Default()
routers.Use(InitMiddleware(studentListService)) //使用中间件,将服务传递到context中
routers.GET("/", func(context *gin.Context) {
service := context.Keys["student_service"].(services.StudentListService)
number := context.DefaultQuery("number", "0")
num, err := strconv.Atoi(number)
if err != nil {
num = 1
}
if num <= 0 {
num = 1
}
//熔断配置
hystrix.ConfigureCommand("student-service", hystrix.CommandConfig{
Timeout: 1000,
})
var studentListResponse *services.StudentListResponse
//熔断处理
err = hystrix.Do("student-service", func() error {
studentListResponse, err = service.GetStudentListService(context, &services.StudentListRequest{
Num: int32(num),
})
return err
}, nil)//fallback这里为nil,可以自己定义需要返回的处理,如果这里调用服务出现问题会直接抛出异常,如果不想抛出异常,可以实现对应的方法,下面我们会处理
if err != nil {
context.JSON(400, gin.H{
"status": "",
"data": err.Error(),
})
} else {
context.JSON(200, gin.H{
"status": studentListResponse.Status,
"data": studentListResponse.Data,
})
}
})
service := web.NewService(
web.Name("home-service-api"),
web.Address(":8081"),
web.Handler(routers),
web.Registry(register),
)
service.Run()
}
3、将服务接口处理时间延迟,可以睡眠几秒钟
4、熔断返回处理
package main
import (
"context"
"github.com/afex/hystrix-go/hystrix"
"github.com/gin-gonic/gin"
"github.com/micro/go-micro"
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/web"
"github.com/micro/go-plugins/registry/consul"
"log"
services "micro-home/service/model"
"strconv"
)
func InitMiddleware(service services.StudentListService) gin.HandlerFunc {
return func(context *gin.Context) {
context.Keys = make(map[string]interface{})
context.Keys["student_service"] = service
context.Next()
}
}
//装饰器
type logWrapper struct {
client.Client
}
//每次调用都会执行这个方法
func (c *logWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
log.Print("请求")
return c.Client.Call(ctx, req, rsp)
}
func NewLogWrapper(c client.Client) client.Client {
return &logWrapper{c}
}
func main() {
register := consul.NewRegistry(func(options *registry.Options) {
options.Addrs = []string{
"192.168.1.171:8500",
"192.168.1.177:8500",
"192.168.1.178:8500",
}
})
//微服务需要注册
homeService := micro.NewService(
micro.Name("micro-home-client"),
micro.Registry(register),
micro.WrapClient(NewLogWrapper), //在注册时只需要传入方法名即可,底层会自动给这个方法传入client
)
//实例化服务
studentListService := services.NewStudentListService("student-service", homeService.Client())
routers := gin.Default()
routers.Use(InitMiddleware(studentListService)) //使用中间件,将服务传递到context中
routers.GET("/", func(context *gin.Context) {
service := context.Keys["student_service"].(services.StudentListService)
number := context.DefaultQuery("number", "0")
num, err := strconv.Atoi(number)
if err != nil {
num = 1
}
if num <= 0 {
num = 1
}
hystrix.ConfigureCommand("student-service", hystrix.CommandConfig{
Timeout: 1000,
})
var studentListResponse *services.StudentListResponse
err = hystrix.Do("student-service", func() error {
studentListResponse, err = service.GetStudentListService(context, &services.StudentListRequest{
Num: int32(num),
})
return err
}, func(e error) error {//熔断处理,当发生熔断后,就返回如下数据
var studentModel []*services.StudentModel
studentModel = append(studentModel, &services.StudentModel{
Id: 0,
Name: "熔断返回",
Sex: "女",
})
studentListResponse = &services.StudentListResponse{
Status: 200,
Data: studentModel,
}
return nil
})
if err != nil {
context.JSON(400, gin.H{
"status": "",
"data": err.Error(),
})
} else {
context.JSON(200, gin.H{
"status": studentListResponse.Status,
"data": studentListResponse.Data,
})
}
})
service := web.NewService(
web.Name("home-service-api"),
web.Address(":8081"),
web.Handler(routers),
web.Registry(register),
)
service.Run()
}
5、可以使用go-micro的wrapper来对熔断进行封装