@ -2,42 +2,17 @@ package config
import (
_ "embed"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"runtime"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
"gopkg.in/yaml.v3"
)
//go:embed version
var Version string
var (
_ , b , _ , _ = runtime . Caller ( 0 )
// Root folder of this project
Root = filepath . Join ( filepath . Dir ( b ) , "../../.." )
)
const (
FileName = "config.yaml"
NotificationFileName = "notification.yaml"
ENV = "CONFIG_NAME"
DefaultFolderPath = "../config/"
ConfKey = "conf"
)
var Config config
type CallBackConfig struct {
Enable bool ` yaml:"enable" `
CallbackTimeOut int ` yaml:" callbackTimeO ut"`
CallbackFailedContinue * bool ` yaml:" callbackF ailedContinue"`
CallbackTimeOut int ` yaml:"timeout" `
CallbackFailedContinue * bool ` yaml:"failedContinue" `
}
type NotificationConf struct {
@ -55,22 +30,74 @@ type POfflinePush struct {
}
type config struct {
ServerIP string ` yaml:"serverip" `
Zookeeper struct {
Schema string ` yaml:"schema" `
ZkAddr [ ] string ` yaml:"address" `
Username string ` yaml:"username" `
Password string ` yaml:"password" `
} ` yaml:"zookeeper" `
RpcRegisterIP string ` yaml:"rpcRegisterIP" `
ListenIP string ` yaml:"listenIP" `
Mysql struct {
Address [ ] string ` yaml:"address" `
Username string ` yaml:"username" `
Password string ` yaml:"password" `
Database string ` yaml:"database" `
MaxOpenConn int ` yaml:"maxOpenConn" `
MaxIdleConn int ` yaml:"maxIdleConn" `
MaxLifeTime int ` yaml:"maxLifeTime" `
LogLevel int ` yaml:"logLevel" `
SlowThreshold int ` yaml:"slowThreshold" `
} ` yaml:"mysql" `
ServerVersion string ` yaml:"serverversion" `
Api struct {
GinPort [ ] int ` yaml:"openImApiPort" `
ListenIP string ` yaml:"listenIP" `
}
Sdk struct {
WsPort [ ] int ` yaml:"openImSdkWsPort" `
DataDir [ ] string ` yaml:"dataDir" `
}
Credential struct {
}
Mongo struct {
Uri string ` yaml:"uri" `
Address [ ] string ` yaml:"address" `
Database string ` yaml:"database" `
Username string ` yaml:"username" `
Password string ` yaml:"password" `
MaxPoolSize int ` yaml:"maxPoolSize" `
} ` yaml:"mongo" `
Redis struct {
Address [ ] string ` yaml:"address" `
Username string ` yaml:"username" `
Password string ` yaml:"password" `
} ` yaml:"redis" `
Kafka struct {
Username string ` yaml:"username" `
Password string ` yaml:"password" `
Addr [ ] string ` yaml:"addr" `
LatestMsgToRedis struct {
Topic string ` yaml:"topic" `
} ` yaml:"latestMsgToRedis" `
MsgToMongo struct {
Topic string ` yaml:"topic" `
} ` yaml:"offlineMsgToMongo" `
MsgToPush struct {
Topic string ` yaml:"topic" `
} ` yaml:"msgToPush" `
MsgToModify struct {
Topic string ` yaml:"topic" `
} ` yaml:"msgToModify" `
ConsumerGroupID struct {
MsgToRedis string ` yaml:"msgToRedis" `
MsgToMongo string ` yaml:"msgToMongo" `
MsgToMySql string ` yaml:"msgToMySql" `
MsgToPush string ` yaml:"msgToPush" `
MsgToModify string ` yaml:"msgToModify" `
} ` yaml:"consumerGroupID" `
} ` yaml:"kafka" `
Rpc struct {
RegisterIP string ` yaml:"registerIP" `
ListenIP string ` yaml:"listenIP" `
} ` yaml:"rpc" `
Api struct {
OpenImApiPort [ ] int ` yaml:"openImApiPort" `
ListenIP string ` yaml:"listenIP" `
} ` yaml:"api" `
Object struct {
Enable string ` yaml:"enable" `
@ -90,7 +117,7 @@ type config struct {
Bucket string ` yaml:"bucket" `
SecretID string ` yaml:"secretID" `
SecretKey string ` yaml:"secretKey" `
}
} ` yaml:"tencent" `
Ali struct {
RegionID string ` yaml:"regionID" `
AccessKeyID string ` yaml:"accessKeyID" `
@ -101,7 +128,7 @@ type config struct {
FinalHost string ` yaml:"finalHost" `
StsDurationSeconds int64 ` yaml:"stsDurationSeconds" `
OssRoleArn string ` yaml:"OssRoleArn" `
}
} ` yaml:"ali" `
Aws struct {
AccessKeyID string ` yaml:"accessKeyID" `
AccessKeySecret string ` yaml:"accessKeySecret" `
@ -112,43 +139,8 @@ type config struct {
ExternalId string ` yaml:"externalId" `
RoleSessionName string ` yaml:"roleSessionName" `
} ` yaml:"aws" `
}
} ` yaml:"object" `
Mysql struct {
DBAddress [ ] string ` yaml:"dbMysqlAddress" `
DBUserName string ` yaml:"dbMysqlUserName" `
DBPassword string ` yaml:"dbMysqlPassword" `
DBDatabaseName string ` yaml:"dbMysqlDatabaseName" `
DBTableName string ` yaml:"DBTableName" `
DBMsgTableNum int ` yaml:"dbMsgTableNum" `
DBMaxOpenConns int ` yaml:"dbMaxOpenConns" `
DBMaxIdleConns int ` yaml:"dbMaxIdleConns" `
DBMaxLifeTime int ` yaml:"dbMaxLifeTime" `
LogLevel int ` yaml:"logLevel" `
SlowThreshold int ` yaml:"slowThreshold" `
}
Mongo struct {
DBUri string ` yaml:"dbUri" `
DBAddress [ ] string ` yaml:"dbAddress" `
DBDirect bool ` yaml:"dbDirect" `
DBTimeout int ` yaml:"dbTimeout" `
DBDatabase string ` yaml:"dbDatabase" `
DBSource string ` yaml:"dbSource" `
DBUserName string ` yaml:"dbUserName" `
DBPassword string ` yaml:"dbPassword" `
DBMaxPoolSize int ` yaml:"dbMaxPoolSize" `
DBRetainChatRecords int ` yaml:"dbRetainChatRecords" `
ChatRecordsClearTime string ` yaml:"chatRecordsClearTime" `
}
Redis struct {
DBAddress [ ] string ` yaml:"dbAddress" `
DBMaxIdle int ` yaml:"dbMaxIdle" `
DBMaxActive int ` yaml:"dbMaxActive" `
DBIdleTimeout int ` yaml:"dbIdleTimeout" `
DBUserName string ` yaml:"dbUserName" `
DBPassWord string ` yaml:"dbPassWord" `
EnableCluster bool ` yaml:"enableCluster" `
}
RpcPort struct {
OpenImUserPort [ ] int ` yaml:"openImUserPort" `
OpenImFriendPort [ ] int ` yaml:"openImFriendPort" `
@ -158,10 +150,10 @@ type config struct {
OpenImAuthPort [ ] int ` yaml:"openImAuthPort" `
OpenImPushPort [ ] int ` yaml:"openImPushPort" `
OpenImConversationPort [ ] int ` yaml:"openImConversationPort" `
OpenImCachePort [ ] int ` yaml:"openImCachePort" `
OpenImRtcPort [ ] int ` yaml:"openImRtcPort" `
OpenImThirdPort [ ] int ` yaml:"openImThirdPort" `
}
} ` yaml:"rpcPort" `
RpcRegisterName struct {
OpenImUserName string ` yaml:"openImUserName" `
OpenImFriendName string ` yaml:"openImFriendName" `
@ -171,128 +163,89 @@ type config struct {
OpenImGroupName string ` yaml:"openImGroupName" `
OpenImAuthName string ` yaml:"openImAuthName" `
OpenImConversationName string ` yaml:"openImConversationName" `
OpenImRtcName string ` yaml:"openImRtcName" `
OpenImThirdName string ` yaml:"openImThirdName" `
}
Zookeeper struct {
Schema string ` yaml:"schema" `
ZkAddr [ ] string ` yaml:"zkAddr" `
UserName string ` yaml:"userName" `
Password string ` yaml:"password" `
} ` yaml:"zookeeper" `
} ` yaml:"rpcRegisterName" `
Log struct {
StorageLocation string ` yaml:"storageLocation" `
RotationTime int ` yaml:"rotationTime" `
RemainRotationCount uint ` yaml:"remainRotationCount" `
RemainLogLevel int ` yaml:"remainLogLevel" `
IsStdout bool ` yaml:"isStdout" `
WithStack bool ` yaml:"withStack" `
IsJson bool ` yaml:"isJson" `
}
ModuleName struct {
LongConnSvrName string ` yaml:"longConnSvrName" `
MsgTransferName string ` yaml:"msgTransferName" `
PushName string ` yaml:"pushName" `
}
WithStack bool ` yaml:"withStack" `
} ` yaml:"log" `
LongConnSvr struct {
WebsocketPort [ ] int ` yaml:"openImWsPort" `
OpenImWsPort [ ] int ` yaml:"openImWsPort" `
WebsocketMaxConnNum int ` yaml:"websocketMaxConnNum" `
WebsocketMaxMsgLen int ` yaml:"websocketMaxMsgLen" `
WebsocketTime Out int ` yaml:"websocketTimeO ut"`
}
WebsocketTime out int ` yaml:"websocketTimeo ut"`
} ` yaml:"longConnSvr" `
Push struct {
Jpns struct {
AppKey string ` yaml:"appKey" `
MasterSecret string ` yaml:"masterSecret" `
PushUrl string ` yaml:"pushUrl" `
PushIntent string ` yaml:"pushIntent" `
Enable bool ` yaml:"enable" `
}
Getui struct {
Enable string ` yaml:"enable" `
GeTui struct {
PushUrl string ` yaml:"pushUrl" `
AppKey string ` yaml:"appKey" `
Enable bool ` yaml:"enable" `
Intent string ` yaml:"intent" `
MasterSecret string ` yaml:"masterSecret" `
ChannelID string ` yaml:"channelID" `
ChannelName string ` yaml:"channelName" `
}
} ` yaml:"geTui" `
Fcm struct {
ServiceAccount string ` yaml:"serviceAccount" `
Enable bool ` yaml:"enable" `
}
} ` yaml:"fcm" `
Jpns struct {
AppKey string ` yaml:"appKey" `
MasterSecret string ` yaml:"masterSecret" `
PushUrl string ` yaml:"pushUrl" `
PushIntent string ` yaml:"pushIntent" `
} ` yaml:"jpns" `
}
Manager struct {
AppManagerUid [ ] string ` yaml:"appManagerUid" `
Nickname [ ] string ` yaml:"nickname" `
}
UserID [ ] string ` yaml:"userID "`
Nickname [ ] string ` yaml:"nickname" `
} ` yaml:"manager" `
Kafka struct {
SASLUserName string ` yaml:"SASLUserName" `
SASLPassword string ` yaml:"SASLPassword" `
Ws2mschat struct {
Addr [ ] string ` yaml:"addr" `
Topic string ` yaml:"topic" `
}
MsgToMongo struct {
Addr [ ] string ` yaml:"addr" `
Topic string ` yaml:"topic" `
}
Ms2pschat struct {
Addr [ ] string ` yaml:"addr" `
Topic string ` yaml:"topic" `
}
MsgToModify struct {
Addr [ ] string ` yaml:"addr" `
Topic string ` yaml:"topic" `
}
ConsumerGroupID struct {
MsgToRedis string ` yaml:"msgToTransfer" `
MsgToMongo string ` yaml:"msgToMongo" `
MsgToMySql string ` yaml:"msgToMySql" `
MsgToPush string ` yaml:"msgToPush" `
MsgToModify string ` yaml:"msgToModify" `
}
}
Secret string ` yaml:"secret" `
MultiLoginPolicy int ` yaml:"multiloginpolicy" `
ChatPersistenceMysql bool ` yaml:"chatpersistencemysql" `
MultiLoginPolicy int ` yaml:"multiLoginPolicy" `
ChatPersistenceMysql bool ` yaml:"chatPersistenceMysql" `
MsgCacheTimeout int ` yaml:"msgCacheTimeout" `
GroupMessageHasReadReceiptEnable bool ` yaml:"groupMessageHasReadReceiptEnable" `
SingleMessageHasReadReceiptEnable bool ` yaml:"singleMessageHasReadReceiptEnable" `
TokenPolicy struct {
RetainChatRecords int ` yaml:"retainChatRecords" `
ChatRecordsClearTime string ` yaml:"chatRecordsClearTime" `
TokenPolicy struct {
AccessSecret string ` yaml:"accessSecret" `
AccessExpire int64 ` yaml:"accessExpire" `
}
} ` yaml:"tokenPolicy" `
MessageVerify struct {
FriendVerify * bool ` yaml:"friendVerify" `
}
} ` yaml:"messageVerify" `
IOSPush struct {
PushSound string ` yaml:"pushSound" `
BadgeCount bool ` yaml:"badgeCount" `
Production bool ` yaml:"production" `
}
} ` yaml:"iosPush" `
Callback struct {
CallbackUrl string ` yaml:" callbackU rl"`
CallbackBeforeSendSingleMsg CallBackConfig ` yaml:" call backB eforeSendSingleMsg"`
CallbackAfterSendSingleMsg CallBackConfig ` yaml:" c allbackA fterSendSingleMsg"`
CallbackBeforeSendGroupMsg CallBackConfig ` yaml:" call backB eforeSendGroupMsg"`
CallbackAfterSendGroupMsg CallBackConfig ` yaml:" c allbackA fterSendGroupMsg"`
CallbackMsgModify CallBackConfig ` yaml:" callbackM sgModify"`
CallbackUserOnline CallBackConfig ` yaml:" callbackU serOnline"`
CallbackUserOffline CallBackConfig ` yaml:" callbackU serOffline"`
CallbackUserKickOff CallBackConfig ` yaml:" callbackU serKickOff"`
CallbackOfflinePush CallBackConfig ` yaml:" callbackO fflinePush"`
CallbackOnlinePush CallBackConfig ` yaml:" callbackO nlinePush"`
CallbackBeforeSuperGroupOnlinePush CallBackConfig ` yaml:" callbackS uperGroupOnlinePush"`
CallbackBeforeAddFriend CallBackConfig ` yaml:" call backB eforeAddFriend"`
CallbackBeforeCreateGroup CallBackConfig ` yaml:" call backB eforeCreateGroup"`
CallbackBeforeMemberJoinGroup CallBackConfig ` yaml:" call backB eforeMemberJoinGroup"`
CallbackBeforeSetGroupMemberInfo CallBackConfig ` yaml:" call backB eforeSetGroupMemberInfo"`
CallbackUrl string ` yaml:"url" `
CallbackBeforeSendSingleMsg CallBackConfig ` yaml:"beforeSendSingleMsg" `
CallbackAfterSendSingleMsg CallBackConfig ` yaml:"afterSendSingleMsg" `
CallbackBeforeSendGroupMsg CallBackConfig ` yaml:"beforeSendGroupMsg" `
CallbackAfterSendGroupMsg CallBackConfig ` yaml:"afterSendGroupMsg" `
CallbackMsgModify CallBackConfig ` yaml:"msgModify" `
CallbackUserOnline CallBackConfig ` yaml:"userOnline" `
CallbackUserOffline CallBackConfig ` yaml:"userOffline" `
CallbackUserKickOff CallBackConfig ` yaml:"userKickOff" `
CallbackOfflinePush CallBackConfig ` yaml:" o fflinePush"`
CallbackOnlinePush CallBackConfig ` yaml:" o nlinePush"`
CallbackBeforeSuperGroupOnlinePush CallBackConfig ` yaml:" s uperGroupOnlinePush"`
CallbackBeforeAddFriend CallBackConfig ` yaml:" beforeAddFriend"`
CallbackBeforeCreateGroup CallBackConfig ` yaml:" beforeCreateGroup"`
CallbackBeforeMemberJoinGroup CallBackConfig ` yaml:" beforeMemberJoinGroup"`
CallbackBeforeSetGroupMemberInfo CallBackConfig ` yaml:" beforeSetGroupMemberInfo"`
} ` yaml:"callback" `
Notification Notification ` yaml:"notification" `
Prometheus struct {
Enable bool ` yaml:"enable" `
@ -308,9 +261,10 @@ type config struct {
MessageTransferPrometheusPort [ ] int ` yaml:"messageTransferPrometheusPort" `
ThirdPrometheusPort [ ] int ` yaml:"thirdPrometheusPort" `
} ` yaml:"prometheus" `
Notification notification ` yaml:"notification" `
}
type N otification struct {
type n otification struct {
GroupCreated NotificationConf ` yaml:"groupCreated" `
GroupInfoSet NotificationConf ` yaml:"groupInfoSet" `
JoinGroupApplication NotificationConf ` yaml:"joinGroupApplication" `
@ -347,82 +301,3 @@ type Notification struct {
ConversationChanged NotificationConf ` yaml:"conversationChanged" `
ConversationSetPrivate NotificationConf ` yaml:"conversationSetPrivate" `
}
func GetServiceNames ( ) [ ] string {
return [ ] string { Config . RpcRegisterName . OpenImUserName , Config . RpcRegisterName . OpenImFriendName , Config . RpcRegisterName . OpenImMsgName ,
Config . RpcRegisterName . OpenImPushName , Config . RpcRegisterName . OpenImMessageGatewayName , Config . RpcRegisterName . OpenImGroupName ,
Config . RpcRegisterName . OpenImAuthName , Config . RpcRegisterName . OpenImConversationName , Config . RpcRegisterName . OpenImRtcName ,
Config . RpcRegisterName . OpenImThirdName }
}
func GetOptionsByNotification ( cfg NotificationConf ) utils . Options {
opts := utils . NewOptions ( )
if cfg . UnreadCount {
opts = utils . WithOptions ( opts , utils . WithUnreadCount ( true ) )
}
if cfg . OfflinePush . Enable {
opts = utils . WithOptions ( opts , utils . WithOfflinePush ( true ) )
}
switch cfg . ReliabilityLevel {
case constant . UnreliableNotification :
case constant . ReliableNotificationNoMsg :
opts = utils . WithOptions ( opts , utils . WithHistory ( true ) , utils . WithPersistent ( ) )
}
opts = utils . WithOptions ( opts , utils . WithSendMsg ( cfg . IsSendMsg ) )
return opts
}
func ( c * config ) unmarshalConfig ( config interface { } , configPath string ) error {
bytes , err := ioutil . ReadFile ( configPath )
if err != nil {
return err
}
if err = yaml . Unmarshal ( bytes , config ) ; err != nil {
return err
}
return nil
}
func ( c * config ) initConfig ( config interface { } , configName , configFolderPath string ) error {
if configFolderPath == "" {
configFolderPath = DefaultFolderPath
}
configPath := filepath . Join ( configFolderPath , configName )
defer func ( ) {
fmt . Println ( "use config" , configPath )
} ( )
_ , err := os . Stat ( configPath )
if err != nil {
if ! os . IsNotExist ( err ) {
return err
}
configPath = filepath . Join ( Root , "config" , configName )
} else {
Root = filepath . Dir ( configPath )
}
return c . unmarshalConfig ( config , configPath )
}
func ( c * config ) RegisterConf2Registry ( registry discoveryregistry . SvcDiscoveryRegistry ) error {
bytes , err := yaml . Marshal ( Config )
if err != nil {
return err
}
return registry . RegisterConf2Registry ( ConfKey , bytes )
}
func ( c * config ) GetConfFromRegistry ( registry discoveryregistry . SvcDiscoveryRegistry ) ( [ ] byte , error ) {
return registry . GetConfFromRegistry ( ConfKey )
}
func InitConfig ( configFolderPath string ) error {
err := Config . initConfig ( & Config , FileName , configFolderPath )
if err != nil {
return err
}
err = Config . initConfig ( & Config . Notification , NotificationFileName , configFolderPath )
if err != nil {
return err
}
return nil
}
withchao
2 years ago