update

mygrpc/grpc.go

@@ -17,6 +17,7 @@ const (
 
 const (
 	ServicePrefix = "grpc@"
+	ExposeHttpTag = "expose-http"
 )
 
 const (

mygrpc/grpcsr/server.go

@@ -53,6 +53,7 @@ type Server struct {
 
 	useDefaultBufferCfg bool
 	delayStopMs         int
+	exposeHttp          bool
 
 	serviceRegInfo *myregistry.ServiceInfo
 }
@@ -82,6 +83,12 @@ func WithDelayStopMs(v int) Opt {
 	}
 }
 
+func WithExposeHttp(v bool) Opt {
+	return func(server *Server) {
+		server.exposeHttp = v
+	}
+}
+
 func SetFlag() {
 	pflag.Int("grpc.port", 0, "listen port, 0 is random port")
 	pflag.String("grpc.log", "true", "enable request log")
@@ -206,6 +213,11 @@ func (s *Server) Run(ctx context.Context) error {
 			ServiceName: s.serviceName,
 			Ip:          svcIp,
 			Port:        port,
+			Extend:      map[string]string{},
+		}
+
+		if s.exposeHttp {
+			s.serviceRegInfo.Extend["tag"] = mygrpc.ExposeHttpTag
 		}
 
 		err = s.reg.Register(s.serviceRegInfo)

mygrpc/mybalancer/random_balancer/random_balancer.go

@@ -19,7 +19,7 @@
 // Package roundrobin defines a roundrobin balancer. Roundrobin balancer is
 // installed as one of the default balancers in gRPC, users don't need to
 // explicitly install this balancer.
-package random
+package random_balancer
 
 import (
 	"google.golang.org/grpc/balancer"