栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

grpc使用多个中间件的转换

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

grpc使用多个中间件的转换

package main

import (
	"context"
	"fmt"
	"log"
)


type UnaryHandler func(ctx context.Context, req interface{}) (interface{}, error)
type UnaryServerInterceptor func(ctx context.Context, req interface{}, handler UnaryHandler) (resp interface{}, err error)
func HelloInterceptor(ctx context.Context, req interface{}, handler UnaryHandler) (interface{}, error){
	log.Println("hello")
	resp, err := handler(ctx, req)
	log.Println("see you")
	return resp, err
}

func HelloInterceptor1(ctx context.Context, req interface{}, handler UnaryHandler) (interface{}, error){
	log.Println("hello1")
	resp, err := handler(ctx, req)
	log.Println("see you1")
	return resp, err
}
func ChainUnaryServer(interceptors ...UnaryServerInterceptor) UnaryServerInterceptor {
	n := len(interceptors)
	return func(ctx context.Context, req interface{}, handler UnaryHandler) (resp interface{}, err error) {
		chainer := func(currentInter UnaryServerInterceptor, currentHandler UnaryHandler) UnaryHandler {
			return func(currentCtx context.Context, currentReq interface{}) (interface{}, error) {
				return currentInter(currentCtx, currentReq, currentHandler)
			}
		}
		chainerHandler := handler
		for i := n - 1; i >= 0; i -- {
			chainerHandler = chainer(interceptors[i], chainerHandler)
		}
		return chainerHandler(ctx, req)
	}
}
func main() {
	interceptor := ChainUnaryServer(HelloInterceptor, HelloInterceptor1)
	i, err := interceptor(context.Background(), nil, func(ctx context.Context, req interface{}) (interface{}, error) {
		return nil, nil
	})
	if err != nil {
		return
	}
	fmt.Println(i)
}

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/644638.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号