Add for get gateway host

baetyl · Aug 8, 2023 · f7e9c68 · f7e9c68
1 parent b8d4a34
commit f7e9c68
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/context/context.go b/context/context.go
@@ -74,6 +74,7 @@ type Context interface {
 	NewBrokerClient(mqtt.ClientConfig) (*mqtt.Client, error)
 	// NewSystemBrokerClient creates a new system broker client.
 	NewSystemBrokerClient([]mqtt.QOSTopic) (*mqtt.Client, error)
+	GetGatewayHost() string
 }
 
 type ctx struct {
@@ -334,3 +335,7 @@ func (c *ctx) NewCoreHttpClient() (*http.Client, error) {
 	ops.Address = getCoreInscureAdddress()
 	return http.NewClient(ops), nil
 }
+
+func (c *ctx) GetGatewayHost() string {
+	return GatewayHost()
+}
diff --git a/context/context_test.go b/context/context_test.go
@@ -187,6 +187,9 @@ func TestContext_CheckSystemCert(t *testing.T) {
 	cc, err = ctx.NewCoreHttpsClient()
 	assert.NoError(t, err)
 	assert.NotNil(t, cc)
+
+	gateway := ctx.GetGatewayHost()
+	assert.Equal(t, gateway, fmt.Sprintf("%s.%s", "baetyl-gateway", baetylEdgeSystemNamespace))
 }
 
 func initCert(t *testing.T) string {

diff --git a/context/env.go b/context/env.go
@@ -120,6 +120,13 @@ func FunctionHost() string {
 	return fmt.Sprintf("%s.%s", "baetyl-function", baetylEdgeSystemNamespace)
 }
 
+func GatewayHost() string {
+	if RunMode() == RunModeNative {
+		return localHost
+	}
+	return fmt.Sprintf("%s.%s", "baetyl-gateway", baetylEdgeSystemNamespace)
+}
+
 func getBrokerAddress() string {
 	return fmt.Sprintf("%s://%s:%s", "ssl", BrokerHost(), BrokerPort())
 }