feat: update go.mod

.gitignore

@@ -2,3 +2,4 @@
 *.txt
 *.exe
 node_modules/*
+example/server/server

ctx_keys.go

@@ -1,5 +1,11 @@
 package gmicro
 
+import (
+	"context"
+
+	"google.golang.org/grpc/metadata"
+)
+
 // CtxKey ctx key type.
 type CtxKey string
 
@@ -8,11 +14,64 @@ const (
 	XRequestID CtxKey = "x-request-id"
 
 	// GRPCClientIP grpc client_ip
-	GRPCClientIP CtxKey = "grpc_client_ip"
+	GRPCClientIP CtxKey = "client-ip"
 
 	// RequestMethod request method
 	RequestMethod CtxKey = "request_method"
 
 	// RequestURI request uri
 	RequestURI CtxKey = "request_uri"
 )
+
+// String returns string
+func (c CtxKey) String() string {
+	return string(c)
+}
+
+// GetIncomingMD returns metadata.MD from incoming ctx
+// get request metadata
+// this method is mainly used at the server end to get the relevant metadata data
+func GetIncomingMD(ctx context.Context) metadata.MD {
+	md, ok := metadata.FromIncomingContext(ctx)
+	if !ok {
+		return metadata.MD{}
+	}
+
+	return md
+}
+
+// GetOutgoingMD returns metadata.MD from outgoing ctx
+// Use this method when you pass ctx to a downstream service
+func GetOutgoingMD(ctx context.Context) metadata.MD {
+	md, ok := metadata.FromOutgoingContext(ctx)
+	if !ok {
+		return metadata.MD{}
+	}
+
+	return md
+}
+
+// GetSliceFromMD returns []string from md
+func GetSliceFromMD(md metadata.MD, key CtxKey) []string {
+	return md.Get(key.String())
+}
+
+// GetSliceFromMD returns string from md
+func GetStringFromMD(md metadata.MD, key CtxKey) string {
+	values := md.Get(key.String())
+	if len(values) > 0 {
+		return values[0]
+	}
+
+	return ""
+}
+
+// SetCtxValue returns ctx when you set key/value into ctx
+func SetCtxValue(ctx context.Context, key CtxKey, val interface{}) context.Context {
+	return context.WithValue(ctx, key.String(), val)
+}
+
+// GetCtxValue returns ctx when you set key/value into ctx
+func GetCtxValue(ctx context.Context, key CtxKey) interface{} {
+	return ctx.Value(key)
+}

example/clients/go/client.go

@@ -5,12 +5,14 @@ import (
 	"log"
 	"os"
 
+	"github.com/daheige/gmicro/v2"
 	"github.com/daheige/gmicro/v2/example/clients/go/pb"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/metadata"
 )
 
-const (
+var (
 	address = "localhost:8081" // grpc server and http gateway on share port
 	// address = "localhost:50051" // grpc server port without http gateway
 	// address     = "localhost:50050" // nginx grpc_pass port
@@ -46,7 +48,14 @@ func main() {
 		name = os.Args[1]
 	}
 
-	res, err := c.SayHello(context.Background(), &pb.HelloReq{
+	requestID := gmicro.Uuid()
+	log.Println("x-request-id: ", requestID)
+	md := metadata.New(map[string]string{
+		"x-request-id": requestID,
+	})
+
+	ctx := metadata.NewOutgoingContext(context.Background(), md)
+	res, err := c.SayHello(ctx, &pb.HelloReq{
 		Name: name,
 	})