Merge pull request #10 from nilorg/develop

Develop
nilorg · Dec 4, 2018 · 2e6bc51 · 2e6bc51
2 parents 93bda91 + da06ecf
commit 2e6bc51
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/grpc/server.go b/grpc/server.go
@@ -10,8 +10,8 @@ import (
 )
 
 var (
-	// DefaultServer 默认server
-	DefaultServer *Server
+	// defaultServer 默认server
+	defaultServer *Server
 )
 
 // Server 服务端
@@ -68,11 +68,16 @@ func (s *Server) Close() {
 
 // Start 启动Grpc
 func Start(address string, tls bool, certFile, keyFile string) {
-	DefaultServer = NewServer(address, tls, certFile, keyFile)
-	DefaultServer.Start()
+	defaultServer = NewServer(address, tls, certFile, keyFile)
+	defaultServer.Start()
+}
+
+// GetSrv 获取rpc server
+func GetSrv() *grpc.Server {
+	return defaultServer.GetSrv()
 }
 
 // Close 关闭Grpc
 func Close() {
-	DefaultServer.Close()
+	defaultServer.Close()
 }