save

Author: Maybe-U

go-kit/v2/v2_transport/ZapLogErrorHandler.go

@@ -18,5 +18,8 @@ func NewZapLogErrorHandler(logger *zap.Logger) *LogErrorHandler {
 }
 
 func (h *LogErrorHandler) Handle(ctx context.Context, err error) {
-	h.logger.Warn(fmt.Sprint(ctx.Value(v2_service.ContextReqUUid), zap.Error(err)))
+	if err != nil {
+		h.logger.Warn(fmt.Sprint(ctx.Value(v2_service.ContextReqUUid), zap.Error(err)))
+		ctx.Done()
+	}
 }

go-kit/v2/v2_transport/transport.go

@@ -12,6 +12,7 @@ import (
 	"learning_tools/go-kit/v2/v2_service"
 	"net/http"
 	"strconv"
+	"time"
 )
 
 func NewHttpHandler(endpoint v2_endpoint.EndPointServer, log *zap.Logger) http.Handler {
@@ -21,13 +22,19 @@ func NewHttpHandler(endpoint v2_endpoint.EndPointServer, log *zap.Logger) http.H
 			w.WriteHeader(http.StatusOK)
 			json.NewEncoder(w).Encode(errorWrapper{Error: err.Error()})
 		}), //程序中的全部报错都会走这里面
-		//httptransport.ServerErrorHandler(NewZapLogErrorHandler(log)),
+		httptransport.ServerErrorHandler(NewZapLogErrorHandler(log)),
 		httptransport.ServerBefore(func(ctx context.Context, request *http.Request) context.Context {
+			ctx, _ = context.WithDeadline(ctx, time.Now().Add(1*time.Second))
 			UUID := uuid.NewV5(uuid.Must(uuid.NewV4(), nil), "req_uuid").String()
 			log.Debug("给请求添加uuid", zap.Any("UUID", UUID))
 			ctx = context.WithValue(ctx, v2_service.ContextReqUUid, UUID)
 			return ctx
 		}),
+
+		httptransport.ServerAfter(func(ctx context.Context, resp http.ResponseWriter) context.Context {
+			fmt.Println("hpj---------value", ctx.Value(v2_service.ContextReqUUid))
+			return context.WithValue(ctx, 12, 3223)
+		}),
 	}
 	m := http.NewServeMux()
 	m.Handle("/sum", httptransport.NewServer(
@@ -46,6 +53,7 @@ func decodeHTTPADDRequest(ctx context.Context, r *http.Request) (interface{}, er
 	)
 	in.A, err = strconv.Atoi(r.FormValue("a"))
 	in.B, err = strconv.Atoi(r.FormValue("b"))
+	//err = errors.New("测试encode 出错------")
 	if err != nil {
 		return in, err
 	}

go-kit/v3/go.mod

@@ -0,0 +1,14 @@
+module learning_tools/go-kit/v3
+
+go 1.14
+
+require (
+	github.com/dgrijalva/jwt-go v3.2.0+incompatible
+	github.com/go-kit/kit v0.10.0
+	github.com/natefinch/lumberjack v2.0.0+incompatible // indirect
+	github.com/satori/go.uuid v1.2.0
+	go.uber.org/zap v1.16.0
+	learning_tools/all_packaged_library v0.0.0-00010101000000-000000000000
+)
+
+replace learning_tools/all_packaged_library => ../../all_packaged_library

go-kit/v3/go.sum

@@ -27,10 +27,10 @@ func NewEndPointServer(svc v3_service.Service, log *zap.Logger) EndPointServer {
 	return EndPointServer{AddEndPoint: addEndPoint, LoginEndPoint: loginEndPoint}
 }
 
-func (s EndPointServer) Add(ctx context.Context, in v3_service.Add) v3_service.AddAck {
-	res, _ := s.AddEndPoint(ctx, in)
-	return res.(v3_service.AddAck)
-}
+//func (s EndPointServer) Add(ctx context.Context, in v3_service.Add) v3_service.AddAck {
+//	res, _ := s.AddEndPoint(ctx, in)
+//	return res.(v3_service.AddAck)
+//}
 
 func MakeAddEndPoint(s v3_service.Service) endpoint.Endpoint {
 	return func(ctx context.Context, request interface{}) (response interface{}, err error) {
@@ -40,10 +40,10 @@ func MakeAddEndPoint(s v3_service.Service) endpoint.Endpoint {
 	}
 }
 
-func (s EndPointServer) Login(ctx context.Context, in v3_service.Login) (v3_service.LoginAck, error) {
-	res, err := s.LoginEndPoint(ctx, in)
-	return res.(v3_service.LoginAck), err
-}
+//func (s EndPointServer) Login(ctx context.Context, in v3_service.Login) (v3_service.LoginAck, error) {
+//	res, err := s.LoginEndPoint(ctx, in)
+//	return res.(v3_service.LoginAck), err
+//}
 
 func MakeLoginEndPoint(s v3_service.Service) endpoint.Endpoint {
 	return func(ctx context.Context, request interface{}) (response interface{}, err error) {

go-kit/v3/v3_endpoint/endpoint.go

@@ -25,6 +25,7 @@ func NewLogMiddlewareServer(log *zap.Logger) NewMiddlewareServer {
 }
 
 func (l logMiddlewareServer) TestAdd(ctx context.Context, in Add) (out AddAck) {
+
 	defer func() {
 		l.logger.Debug(fmt.Sprint(ctx.Value(ContextReqUUid)), zap.Any("调用 Login logMiddlewareServer", "TestAdd"), zap.Any("req", in), zap.Any("res", out))
 	}()

go-kit/v3/v3_service/middleware.go

@@ -1,11 +1,16 @@
 package v3_service
 
+type ComRespone struct {
+	Code int `json:"code"`
+}
+
 type Add struct {
 	A int `json:"a"`
 	B int `json:"b"`
 }
 
 type AddAck struct {
+	ComRespone
 	Res int `json:"res"`
 }
 
@@ -15,5 +20,6 @@ type Login struct {
 }
 
 type LoginAck struct {
+	ComRespone
 	Token string `json:"token"`
 }

go-kit/v3/v3_service/model.go

@@ -22,7 +22,7 @@ func NewHttpHandler(endpoint v3_endpoint.EndPointServer, log *zap.Logger) http.H
 			json.NewEncoder(w).Encode(errorWrapper{Error: err.Error()})
 		}), //程序中的全部报错都会走这里面
 		httptransport.ServerBefore(func(ctx context.Context, request *http.Request) context.Context {
-			UUID := uuid.NewV5(uuid.Must(uuid.NewV4()), "req_uuid").String()
+			UUID := uuid.NewV5(uuid.Must(uuid.NewV4(), nil), "req_uuid").String()
 			log.Debug("给请求添加uuid", zap.Any("UUID", UUID))
 			ctx = context.WithValue(ctx, v3_service.ContextReqUUid, UUID)
 			ctx = context.WithValue(ctx, utils.JWT_CONTEXT_KEY, request.Header.Get("Authorization"))