feat(transport): add endpoint option for server (#2904)

1 year ago · bffc1a0989
parent 32b1d13f90
commit bffc1a0989
2 changed files with 14 additions and 0 deletions
--- a/transport/grpc/server.go
+++ b/transport/grpc/server.go
@ -45,6 +45,13 @@ func Address(addr string) ServerOption {
 	}
 }

+// Endpoint with server address.
+func Endpoint(endpoint *url.URL) ServerOption {
+	return func(s *Server) {
+		s.endpoint = endpoint
+	}
+}
+
 // Timeout with server timeout.
 func Timeout(timeout time.Duration) ServerOption {
 	return func(s *Server) {
--- a/transport/http/server.go
+++ b/transport/http/server.go
@ -42,6 +42,13 @@ func Address(addr string) ServerOption {
 	}
 }

+// Endpoint with server address.
+func Endpoint(endpoint *url.URL) ServerOption {
+	return func(s *Server) {
+		s.endpoint = endpoint
+	}
+}
+
 // Timeout with server timeout.
 func Timeout(timeout time.Duration) ServerOption {
 	return func(s *Server) {