完成complete方法

This commit is contained in:
chenjiekun
2023-03-01 18:09:23 +08:00
parent 1c0e58563d
commit bf7143c930
6 changed files with 186 additions and 13 deletions

View File

@@ -0,0 +1,52 @@
package proxy_mirror
import (
"github.com/eolinker/eosc/eocontext"
dubbo2_context "github.com/eolinker/eosc/eocontext/dubbo2-context"
grpc_context "github.com/eolinker/eosc/eocontext/grpc-context"
http_service "github.com/eolinker/eosc/eocontext/http-context"
log "github.com/eolinker/goku-api-gateway/goku-log"
)
type proxyMirrorCompleteHandler struct {
orgComplete eocontext.CompleteHandler
mirrorComplete eocontext.CompleteHandler
}
func newMirrorHandler(eoCtx eocontext.EoContext, proxyCfg *Config) (eocontext.CompleteHandler, error) {
handler := &proxyMirrorCompleteHandler{
orgComplete: eoCtx.GetComplete(),
}
if _, success := eoCtx.(http_service.IHttpContext); success {
handler.mirrorComplete = newHttpMirrorComplete(proxyCfg)
} else if _, success = eoCtx.(grpc_context.IGrpcContext); success {
handler.mirrorComplete = newGrpcMirrorComplete(proxyCfg)
} else if _, success = eoCtx.(dubbo2_context.IDubbo2Context); success {
handler.mirrorComplete = newDubbo2MirrorComplete(proxyCfg)
} else if _, success = eoCtx.(http_service.IWebsocketContext); success {
handler.mirrorComplete = newWebsocketMirrorComplete(proxyCfg)
} else {
return nil, ErrUnsupportedType
}
return handler, nil
}
func (p *proxyMirrorCompleteHandler) Complete(ctx eocontext.EoContext) error {
//执行镜像请求的Complete
cloneCtx, err := ctx.Clone()
if err != nil {
return err
}
go func() {
err = p.mirrorComplete.Complete(cloneCtx)
if err != nil {
log.Error(err)
}
}()
//执行原始Complete
return p.orgComplete.Complete(ctx)
}