x

Author: chenquan

internal/health/health.go

@@ -48,6 +48,11 @@ func IsReady() bool {
 	return defaultHealthManager.IsReady()
 }
 
+// MarkReady sets components status to ready.
+func MarkReady() {
+	defaultHealthManager.MarkReady()
+}
+
 // CreateHttpHandler create health http handler base on given probe.
 func CreateHttpHandler(healthResponse string) http.HandlerFunc {
 	return func(w http.ResponseWriter, _ *http.Request) {

zrpc/internal/rpcpubserver_test.go

@@ -4,8 +4,11 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"google.golang.org/grpc"
+
 	"github.com/zeromicro/go-zero/core/discov"
 	"github.com/zeromicro/go-zero/core/netx"
+	"github.com/zeromicro/go-zero/internal/health"
 )
 
 func TestNewRpcPubServer(t *testing.T) {
@@ -15,8 +18,11 @@ func TestNewRpcPubServer(t *testing.T) {
 		ID:   10,
 	}, "")
 	assert.NoError(t, err)
+	health.MarkReady()
 	assert.NotPanics(t, func() {
-		s.Start(nil)
+		s.Start(func(server *grpc.Server) {
+
+		})
 	})
 }