mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-05 05:12:45 +08:00
9ed6200e45
* pb * fix: Modifying other fields while setting IsPrivateChat does not take effect * fix: quote message error revoke * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * refactoring scheduled tasks * upgrading pkg tools * fix * fix * optimize log output * feat: support GetLastMessage * feat: support GetLastMessage * feat: s3 switch * feat: s3 switch * fix: GetUsersOnline * feat: SendBusinessNotification supported configuration parameters * feat: SendBusinessNotification supported configuration parameters * feat: SendBusinessNotification supported configuration parameters * feat: seq conversion failed without exiting * monolithic * fix: DeleteDoc crash * fix: DeleteDoc crash * fix: monolithic * fix: monolithic * fix: fill send time * fix: fill send time * fix: crash caused by withdrawing messages from users who have left the group * fix: mq * fix: mq * fix: user msg timestamp * fix: mq * 1 * 1 * 1 * 1 * 1 * 1 * 1 * seq read config * seq read config * 1 * 1 * fix: the source message of the reference is withdrawn, and the referenced message is deleted * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1 * 1
305 lines
11 KiB
Go
305 lines
11 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/apistruct"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/authverify"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/discovery/etcd"
|
|
"github.com/openimsdk/open-im-server/v3/version"
|
|
"github.com/openimsdk/tools/apiresp"
|
|
"github.com/openimsdk/tools/errs"
|
|
"github.com/openimsdk/tools/log"
|
|
"github.com/openimsdk/tools/utils/runtimeenv"
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
)
|
|
|
|
const (
|
|
// wait for Restart http call return
|
|
waitHttp = time.Millisecond * 200
|
|
)
|
|
|
|
type ConfigManager struct {
|
|
imAdminUserID []string
|
|
config *config.AllConfig
|
|
client *clientv3.Client
|
|
|
|
configPath string
|
|
}
|
|
|
|
func NewConfigManager(IMAdminUserID []string, cfg *config.AllConfig, client *clientv3.Client, configPath string) *ConfigManager {
|
|
cm := &ConfigManager{
|
|
imAdminUserID: IMAdminUserID,
|
|
config: cfg,
|
|
client: client,
|
|
configPath: configPath,
|
|
}
|
|
return cm
|
|
}
|
|
|
|
func (cm *ConfigManager) CheckAdmin(c *gin.Context) {
|
|
if err := authverify.CheckAdmin(c, cm.imAdminUserID); err != nil {
|
|
apiresp.GinError(c, err)
|
|
c.Abort()
|
|
}
|
|
}
|
|
|
|
func (cm *ConfigManager) GetConfig(c *gin.Context) {
|
|
var req apistruct.GetConfigReq
|
|
if err := c.BindJSON(&req); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
conf := cm.config.Name2Config(req.ConfigName)
|
|
if conf == nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail("config name not found").Wrap())
|
|
return
|
|
}
|
|
b, err := json.Marshal(conf)
|
|
if err != nil {
|
|
apiresp.GinError(c, err)
|
|
return
|
|
}
|
|
apiresp.GinSuccess(c, string(b))
|
|
}
|
|
|
|
func (cm *ConfigManager) GetConfigList(c *gin.Context) {
|
|
var resp apistruct.GetConfigListResp
|
|
resp.ConfigNames = cm.config.GetConfigNames()
|
|
resp.Environment = runtimeenv.RuntimeEnvironment()
|
|
resp.Version = version.Version
|
|
|
|
apiresp.GinSuccess(c, resp)
|
|
}
|
|
|
|
func (cm *ConfigManager) SetConfig(c *gin.Context) {
|
|
if cm.config.Discovery.Enable != config.ETCD {
|
|
apiresp.GinError(c, errs.New("only etcd support set config").Wrap())
|
|
return
|
|
}
|
|
var req apistruct.SetConfigReq
|
|
if err := c.BindJSON(&req); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
var err error
|
|
switch req.ConfigName {
|
|
case cm.config.Discovery.GetConfigFileName():
|
|
err = compareAndSave[config.Discovery](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Kafka.GetConfigFileName():
|
|
err = compareAndSave[config.Kafka](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.LocalCache.GetConfigFileName():
|
|
err = compareAndSave[config.LocalCache](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Log.GetConfigFileName():
|
|
err = compareAndSave[config.Log](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Minio.GetConfigFileName():
|
|
err = compareAndSave[config.Minio](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Mongo.GetConfigFileName():
|
|
err = compareAndSave[config.Mongo](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Notification.GetConfigFileName():
|
|
err = compareAndSave[config.Notification](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.API.GetConfigFileName():
|
|
err = compareAndSave[config.API](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.CronTask.GetConfigFileName():
|
|
err = compareAndSave[config.CronTask](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.MsgGateway.GetConfigFileName():
|
|
err = compareAndSave[config.MsgGateway](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.MsgTransfer.GetConfigFileName():
|
|
err = compareAndSave[config.MsgTransfer](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Push.GetConfigFileName():
|
|
err = compareAndSave[config.Push](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Auth.GetConfigFileName():
|
|
err = compareAndSave[config.Auth](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Conversation.GetConfigFileName():
|
|
err = compareAndSave[config.Conversation](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Friend.GetConfigFileName():
|
|
err = compareAndSave[config.Friend](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Group.GetConfigFileName():
|
|
err = compareAndSave[config.Group](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Msg.GetConfigFileName():
|
|
err = compareAndSave[config.Msg](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Third.GetConfigFileName():
|
|
err = compareAndSave[config.Third](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.User.GetConfigFileName():
|
|
err = compareAndSave[config.User](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Redis.GetConfigFileName():
|
|
err = compareAndSave[config.Redis](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Share.GetConfigFileName():
|
|
err = compareAndSave[config.Share](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
case cm.config.Webhooks.GetConfigFileName():
|
|
err = compareAndSave[config.Webhooks](c, cm.config.Name2Config(req.ConfigName), &req, cm)
|
|
default:
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap())
|
|
return
|
|
}
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
apiresp.GinSuccess(c, nil)
|
|
}
|
|
|
|
func compareAndSave[T any](c *gin.Context, old any, req *apistruct.SetConfigReq, cm *ConfigManager) error {
|
|
conf := new(T)
|
|
err := json.Unmarshal([]byte(req.Data), &conf)
|
|
if err != nil {
|
|
return errs.ErrArgs.WithDetail(err.Error()).Wrap()
|
|
}
|
|
eq := reflect.DeepEqual(old, conf)
|
|
if eq {
|
|
return nil
|
|
}
|
|
data, err := json.Marshal(conf)
|
|
if err != nil {
|
|
return errs.ErrArgs.WithDetail(err.Error()).Wrap()
|
|
}
|
|
_, err = cm.client.Put(c, etcd.BuildKey(req.ConfigName), string(data))
|
|
if err != nil {
|
|
return errs.WrapMsg(err, "save to etcd failed")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cm *ConfigManager) ResetConfig(c *gin.Context) {
|
|
go func() {
|
|
if err := cm.resetConfig(c, true); err != nil {
|
|
log.ZError(c, "reset config err", err)
|
|
}
|
|
}()
|
|
apiresp.GinSuccess(c, nil)
|
|
}
|
|
|
|
func (cm *ConfigManager) resetConfig(c *gin.Context, checkChange bool, ops ...clientv3.Op) error {
|
|
txn := cm.client.Txn(c)
|
|
type initConf struct {
|
|
old any
|
|
new any
|
|
}
|
|
configMap := map[string]*initConf{
|
|
cm.config.Discovery.GetConfigFileName(): {old: &cm.config.Discovery, new: new(config.Discovery)},
|
|
cm.config.Kafka.GetConfigFileName(): {old: &cm.config.Kafka, new: new(config.Kafka)},
|
|
cm.config.LocalCache.GetConfigFileName(): {old: &cm.config.LocalCache, new: new(config.LocalCache)},
|
|
cm.config.Log.GetConfigFileName(): {old: &cm.config.Log, new: new(config.Log)},
|
|
cm.config.Minio.GetConfigFileName(): {old: &cm.config.Minio, new: new(config.Minio)},
|
|
cm.config.Mongo.GetConfigFileName(): {old: &cm.config.Mongo, new: new(config.Mongo)},
|
|
cm.config.Notification.GetConfigFileName(): {old: &cm.config.Notification, new: new(config.Notification)},
|
|
cm.config.API.GetConfigFileName(): {old: &cm.config.API, new: new(config.API)},
|
|
cm.config.CronTask.GetConfigFileName(): {old: &cm.config.CronTask, new: new(config.CronTask)},
|
|
cm.config.MsgGateway.GetConfigFileName(): {old: &cm.config.MsgGateway, new: new(config.MsgGateway)},
|
|
cm.config.MsgTransfer.GetConfigFileName(): {old: &cm.config.MsgTransfer, new: new(config.MsgTransfer)},
|
|
cm.config.Push.GetConfigFileName(): {old: &cm.config.Push, new: new(config.Push)},
|
|
cm.config.Auth.GetConfigFileName(): {old: &cm.config.Auth, new: new(config.Auth)},
|
|
cm.config.Conversation.GetConfigFileName(): {old: &cm.config.Conversation, new: new(config.Conversation)},
|
|
cm.config.Friend.GetConfigFileName(): {old: &cm.config.Friend, new: new(config.Friend)},
|
|
cm.config.Group.GetConfigFileName(): {old: &cm.config.Group, new: new(config.Group)},
|
|
cm.config.Msg.GetConfigFileName(): {old: &cm.config.Msg, new: new(config.Msg)},
|
|
cm.config.Third.GetConfigFileName(): {old: &cm.config.Third, new: new(config.Third)},
|
|
cm.config.User.GetConfigFileName(): {old: &cm.config.User, new: new(config.User)},
|
|
cm.config.Redis.GetConfigFileName(): {old: &cm.config.Redis, new: new(config.Redis)},
|
|
cm.config.Share.GetConfigFileName(): {old: &cm.config.Share, new: new(config.Share)},
|
|
cm.config.Webhooks.GetConfigFileName(): {old: &cm.config.Webhooks, new: new(config.Webhooks)},
|
|
}
|
|
|
|
changedKeys := make([]string, 0, len(configMap))
|
|
for k, v := range configMap {
|
|
err := config.Load(cm.configPath, k, config.EnvPrefixMap[k], v.new)
|
|
if err != nil {
|
|
log.ZError(c, "load config failed", err)
|
|
continue
|
|
}
|
|
equal := reflect.DeepEqual(v.old, v.new)
|
|
if !checkChange || !equal {
|
|
changedKeys = append(changedKeys, k)
|
|
}
|
|
}
|
|
|
|
for _, k := range changedKeys {
|
|
data, err := json.Marshal(configMap[k].new)
|
|
if err != nil {
|
|
log.ZError(c, "marshal config failed", err)
|
|
continue
|
|
}
|
|
ops = append(ops, clientv3.OpPut(etcd.BuildKey(k), string(data)))
|
|
}
|
|
if len(ops) > 0 {
|
|
txn.Then(ops...)
|
|
_, err := txn.Commit()
|
|
if err != nil {
|
|
return errs.WrapMsg(err, "commit etcd txn failed")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cm *ConfigManager) Restart(c *gin.Context) {
|
|
go cm.restart(c)
|
|
apiresp.GinSuccess(c, nil)
|
|
}
|
|
|
|
func (cm *ConfigManager) restart(c *gin.Context) {
|
|
time.Sleep(waitHttp) // wait for Restart http call return
|
|
t := time.Now().Unix()
|
|
_, err := cm.client.Put(c, etcd.BuildKey(etcd.RestartKey), strconv.Itoa(int(t)))
|
|
if err != nil {
|
|
log.ZError(c, "restart etcd put key failed", err)
|
|
}
|
|
}
|
|
|
|
func (cm *ConfigManager) SetEnableConfigManager(c *gin.Context) {
|
|
if cm.config.Discovery.Enable != config.ETCD {
|
|
apiresp.GinError(c, errs.New("only etcd support config manager").Wrap())
|
|
return
|
|
}
|
|
var req apistruct.SetEnableConfigManagerReq
|
|
if err := c.BindJSON(&req); err != nil {
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
|
|
return
|
|
}
|
|
var enableStr string
|
|
if req.Enable {
|
|
enableStr = etcd.Enable
|
|
} else {
|
|
enableStr = etcd.Disable
|
|
}
|
|
resp, err := cm.client.Get(c, etcd.BuildKey(etcd.EnableConfigCenterKey))
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.WrapMsg(err, "getEnableConfigManager failed"))
|
|
return
|
|
}
|
|
if !(resp.Count > 0 && string(resp.Kvs[0].Value) == etcd.Enable) && req.Enable {
|
|
go func() {
|
|
time.Sleep(waitHttp) // wait for Restart http call return
|
|
err := cm.resetConfig(c, false, clientv3.OpPut(etcd.BuildKey(etcd.EnableConfigCenterKey), enableStr))
|
|
if err != nil {
|
|
log.ZError(c, "resetConfig failed", err)
|
|
}
|
|
}()
|
|
} else {
|
|
_, err = cm.client.Put(c, etcd.BuildKey(etcd.EnableConfigCenterKey), enableStr)
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.WrapMsg(err, "setEnableConfigManager failed"))
|
|
return
|
|
}
|
|
}
|
|
|
|
apiresp.GinSuccess(c, nil)
|
|
}
|
|
|
|
func (cm *ConfigManager) GetEnableConfigManager(c *gin.Context) {
|
|
resp, err := cm.client.Get(c, etcd.BuildKey(etcd.EnableConfigCenterKey))
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.WrapMsg(err, "getEnableConfigManager failed"))
|
|
return
|
|
}
|
|
var enable bool
|
|
if resp.Count > 0 && string(resp.Kvs[0].Value) == etcd.Enable {
|
|
enable = true
|
|
}
|
|
apiresp.GinSuccess(c, &apistruct.GetEnableConfigManagerResp{Enable: enable})
|
|
}
|