复制项目

pkg/common/config/parse.go

@@ -0,0 +1,107 @@
+// 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 config
+
+import (
+	"os"
+	"path/filepath"
+
+	"gopkg.in/yaml.v3"
+
+	"git.imall.cloud/openim/open-im-server-deploy/pkg/msgprocessor"
+	"git.imall.cloud/openim/protocol/constant"
+	"github.com/openimsdk/tools/errs"
+	"github.com/openimsdk/tools/field"
+)
+
+const (
+	DefaultFolderPath = "../config/"
+)
+
+// return absolude path join ../config/, this is k8s container config path.
+func GetDefaultConfigPath() (string, error) {
+	executablePath, err := os.Executable()
+	if err != nil {
+		return "", errs.WrapMsg(err, "failed to get executable path")
+	}
+
+	configPath, err := field.OutDir(filepath.Join(filepath.Dir(executablePath), "../config/"))
+	if err != nil {
+		return "", errs.WrapMsg(err, "failed to get output directory", "outDir", filepath.Join(filepath.Dir(executablePath), "../config/"))
+	}
+	return configPath, nil
+}
+
+// getProjectRoot returns the absolute path of the project root directory.
+func GetProjectRoot() (string, error) {
+	executablePath, err := os.Executable()
+	if err != nil {
+		return "", errs.Wrap(err)
+	}
+	projectRoot, err := field.OutDir(filepath.Join(filepath.Dir(executablePath), "../../../../.."))
+	if err != nil {
+		return "", errs.Wrap(err)
+	}
+	return projectRoot, nil
+}
+
+func GetOptionsByNotification(cfg NotificationConfig, sendMessage *bool) msgprocessor.Options {
+	opts := msgprocessor.NewOptions()
+
+	if sendMessage != nil {
+		cfg.IsSendMsg = *sendMessage
+	}
+	if cfg.IsSendMsg {
+		opts = msgprocessor.WithOptions(opts, msgprocessor.WithUnreadCount(true))
+	}
+	if cfg.OfflinePush.Enable {
+		opts = msgprocessor.WithOptions(opts, msgprocessor.WithOfflinePush(true))
+	}
+	switch cfg.ReliabilityLevel {
+	case constant.UnreliableNotification:
+	case constant.ReliableNotificationNoMsg:
+		opts = msgprocessor.WithOptions(opts, msgprocessor.WithHistory(true), msgprocessor.WithPersistent())
+	}
+	opts = msgprocessor.WithOptions(opts, msgprocessor.WithSendMsg(cfg.IsSendMsg))
+
+	return opts
+}
+
+// initConfig loads configuration from a specified path into the provided config structure.
+// If the specified config file does not exist, it attempts to load from the project's default "config" directory.
+// It logs informative messages regarding the configuration path being used.
+func initConfig(config any, configName, configFolderPath string) error {
+	configFolderPath = filepath.Join(configFolderPath, configName)
+	_, err := os.Stat(configFolderPath)
+	if err != nil {
+		if !os.IsNotExist(err) {
+			return errs.WrapMsg(err, "stat config path error", "config Folder Path", configFolderPath)
+		}
+		path, err := GetProjectRoot()
+		if err != nil {
+			return err
+		}
+		configFolderPath = filepath.Join(path, "config", configName)
+	}
+	data, err := os.ReadFile(configFolderPath)
+	if err != nil {
+		return errs.WrapMsg(err, "read file error", "config Folder Path", configFolderPath)
+	}
+	if err = yaml.Unmarshal(data, config); err != nil {
+		return errs.WrapMsg(err, "unmarshal yaml error", "config Folder Path", configFolderPath)
+	}
+
+	return nil
+}