update

mygrpc/grpc.go

@@ -0,0 +1,41 @@
+package mygrpc
+
+import (
+	"context"
+	"google.golang.org/grpc/metadata"
+	"google.golang.org/grpc/peer"
+	"net"
+)
+
+const (
+	DefaultReadBufferSize  = 256 * 1024 // 256kb
+	DefaultWriteBufferSize = 256 * 1024
+
+	// DefaultWindowSize 滑动窗口 16mb 手动设置滑动窗口大小, 尝试提升吞吐量, 减少动态计算可能导致的cpu波动
+	DefaultWindowSize = 16 * 1024 * 1024
+)
+
+const (
+	HeaderClientIP    = "grpc-client-ip"
+	HeaderServiceName = "grpc-service-name"
+)
+
+func ClientIP(ctx context.Context) string {
+	// 先从自定义的header获取
+	md, ok := metadata.FromIncomingContext(ctx)
+	if ok {
+		if v := md.Get(HeaderClientIP); len(v) > 0 {
+			return v[0]
+		}
+	}
+
+	p, ok := peer.FromContext(ctx)
+	if ok {
+		switch v := p.Addr.(type) {
+		case *net.TCPAddr:
+			return v.IP.String()
+		}
+	}
+
+	return ""
+}