复制项目

2026-01-14 22:16:44 +08:00
parent e2577b8cee
commit e50142a3b9
691 changed files with 97009 additions and 1 deletions
--- a/pkg/common/cmd/msg_gateway.go
+++ b/pkg/common/cmd/msg_gateway.go
@@ -0,0 +1,105 @@
+// Copyright © 2023 OpenIM. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"git.imall.cloud/openim/open-im-server-deploy/internal/msggateway"
+	"git.imall.cloud/openim/open-im-server-deploy/pkg/common/config"
+	"git.imall.cloud/openim/open-im-server-deploy/pkg/common/startrpc"
+	"git.imall.cloud/openim/open-im-server-deploy/version"
+
+	"github.com/openimsdk/tools/log"
+	"github.com/openimsdk/tools/system/program"
+	"github.com/spf13/cobra"
+)
+
+type MsgGatewayCmd struct {
+	*RootCmd
+	ctx              context.Context
+	configMap        map[string]any
+	msgGatewayConfig *msggateway.Config
+}
+
+func NewMsgGatewayCmd() *MsgGatewayCmd {
+	var msgGatewayConfig msggateway.Config
+	ret := &MsgGatewayCmd{msgGatewayConfig: &msgGatewayConfig}
+	ret.configMap = map[string]any{
+		config.OpenIMMsgGatewayCfgFileName: &msgGatewayConfig.MsgGateway,
+		config.ShareFileName:               &msgGatewayConfig.Share,
+		config.RedisConfigFileName:         &msgGatewayConfig.RedisConfig,
+		config.WebhooksConfigFileName:      &msgGatewayConfig.WebhooksConfig,
+		config.DiscoveryConfigFilename:     &msgGatewayConfig.Discovery,
+	}
+	ret.RootCmd = NewRootCmd(program.GetProcessName(), WithConfigMap(ret.configMap))
+	ret.ctx = context.WithValue(context.Background(), "version", version.Version)
+	ret.Command.RunE = func(cmd *cobra.Command, args []string) error {
+		return ret.runE()
+	}
+	return ret
+}
+
+func (m *MsgGatewayCmd) Exec() error {
+	return m.Execute()
+}
+
+func (m *MsgGatewayCmd) runE() error {
+	m.msgGatewayConfig.Index = config.Index(m.Index())
+	rpc := m.msgGatewayConfig.MsgGateway.RPC
+	// 从配置读取 Prometheus，避免空配置导致的下标越界
+	prometheus := m.msgGatewayConfig.MsgGateway.Prometheus
+
+	b, err := json.Marshal(prometheus)
+	if err != nil {
+		return err
+	}
+	fmt.Println(string(b))
+	log.CInfo(m.ctx, "prometheus", "prometheus", string(b))
+	// 调试日志：打印关键启动参数
+	log.CInfo(m.ctx, "msg-gateway starting",
+		"autoSetPorts", rpc.AutoSetPorts,
+		"rpcPorts", rpc.Ports,
+		"prometheusEnable", prometheus.Enable,
+		"prometheusPorts", prometheus.Ports,
+		"index", int(m.msgGatewayConfig.Index),
+		"listenIP", rpc.ListenIP,
+		"registerIP", rpc.RegisterIP,
+	)
+
+	if !rpc.AutoSetPorts && (len(rpc.Ports) == 0) {
+		log.ZWarn(m.ctx, "rpc ports is empty while autoSetPorts=false", nil)
+	}
+	if prometheus.Enable && len(prometheus.Ports) == 0 {
+		log.ZWarn(m.ctx, "prometheus enabled but ports is empty", nil)
+	}
+	return startrpc.Start(
+		m.ctx, &m.msgGatewayConfig.Discovery,
+		&m.msgGatewayConfig.MsgGateway.CircuitBreaker,
+		&m.msgGatewayConfig.MsgGateway.RateLimiter,
+		&prometheus,
+		rpc.ListenIP, rpc.RegisterIP,
+		rpc.AutoSetPorts,
+		rpc.Ports, int(m.msgGatewayConfig.Index),
+		m.msgGatewayConfig.Discovery.RpcService.MessageGateway,
+		nil,
+		m.msgGatewayConfig,
+		[]string{},
+		[]string{},
+		msggateway.Start,
+	)
+}