You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
460 lines
16 KiB
460 lines
16 KiB
package config
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/spf13/viper"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var (
|
|
_, b, _, _ = runtime.Caller(0)
|
|
// Root folder of this project
|
|
Root = filepath.Join(filepath.Dir(b), "../../..")
|
|
)
|
|
|
|
var Config config
|
|
|
|
type callBackConfig struct {
|
|
Enable bool `yaml:"enable"`
|
|
CallbackTimeOut int `yaml:"callbackTimeOut"`
|
|
CallbackFailedContinue bool `yaml:"callbackFailedContinue"`
|
|
}
|
|
|
|
type config struct {
|
|
ServerIP string `yaml:"serverip"`
|
|
ServerVersion string `yaml:"serverversion"`
|
|
Api struct {
|
|
GinPort []int `yaml:"openImApiPort"`
|
|
}
|
|
CmsApi struct {
|
|
GinPort []int `yaml:"openImCmsApiPort"`
|
|
}
|
|
Sdk struct {
|
|
WsPort []int `yaml:"openImSdkWsPort"`
|
|
}
|
|
Credential struct {
|
|
Tencent struct {
|
|
AppID string `yaml:"appID"`
|
|
Region string `yaml:"region"`
|
|
Bucket string `yaml:"bucket"`
|
|
SecretID string `yaml:"secretID"`
|
|
SecretKey string `yaml:"secretKey"`
|
|
}
|
|
Ali struct {
|
|
RegionID string `yaml:"regionID"`
|
|
AccessKeyID string `yaml:"accessKeyID"`
|
|
AccessKeySecret string `yaml:"accessKeySecret"`
|
|
StsEndpoint string `yaml:"stsEndpoint"`
|
|
OssEndpoint string `yaml:"ossEndpoint"`
|
|
Bucket string `yaml:"bucket"`
|
|
FinalHost string `yaml:"finalHost"`
|
|
StsDurationSeconds int64 `yaml:"stsDurationSeconds"`
|
|
OssRoleArn string `yaml:"OssRoleArn"`
|
|
}
|
|
Minio struct {
|
|
Bucket string `yaml:"bucket"`
|
|
Location string `yaml:"location"`
|
|
Endpoint string `yaml:"endpoint"`
|
|
AccessKeyID string `yaml:"accessKeyID"`
|
|
SecretAccessKey string `yaml:"secretAccessKey"`
|
|
EndpointInner string `yaml:"endpointInner"`
|
|
EndpointInnerEnable bool `yaml:"endpointInnerEnable"`
|
|
} `yaml:"minio"`
|
|
}
|
|
|
|
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"`
|
|
}
|
|
Mongo struct {
|
|
DBUri string `yaml:"dbUri"` // 当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"`
|
|
}
|
|
Redis struct {
|
|
DBAddress string `yaml:"dbAddress"`
|
|
DBMaxIdle int `yaml:"dbMaxIdle"`
|
|
DBMaxActive int `yaml:"dbMaxActive"`
|
|
DBIdleTimeout int `yaml:"dbIdleTimeout"`
|
|
DBPassWord string `yaml:"dbPassWord"`
|
|
}
|
|
RpcPort struct {
|
|
OpenImUserPort []int `yaml:"openImUserPort"`
|
|
openImFriendPort []int `yaml:"openImFriendPort"`
|
|
RpcMessagePort []int `yaml:"rpcMessagePort"`
|
|
RpcPushMessagePort []int `yaml:"rpcPushMessagePort"`
|
|
OpenImGroupPort []int `yaml:"openImGroupPort"`
|
|
RpcModifyUserInfoPort []int `yaml:"rpcModifyUserInfoPort"`
|
|
RpcGetTokenPort []int `yaml:"rpcGetTokenPort"`
|
|
}
|
|
RpcRegisterName struct {
|
|
OpenImStatisticsName string `yaml:"OpenImStatisticsName"`
|
|
OpenImUserName string `yaml:"openImUserName"`
|
|
OpenImFriendName string `yaml:"openImFriendName"`
|
|
OpenImOfflineMessageName string `yaml:"openImOfflineMessageName"`
|
|
OpenImPushName string `yaml:"openImPushName"`
|
|
OpenImOnlineMessageRelayName string `yaml:"openImOnlineMessageRelayName"`
|
|
OpenImGroupName string `yaml:"openImGroupName"`
|
|
OpenImAuthName string `yaml:"openImAuthName"`
|
|
OpenImMessageCMSName string `yaml:"openImMessageCMSName"`
|
|
OpenImAdminCMSName string `yaml:"openImAdminCMSName"`
|
|
OpenImOfficeName string `yaml:"openImOfficeName"`
|
|
OpenImOrganizationName string `yaml:"openImOrganizationName"`
|
|
OpenImConversationName string `yaml:"openImConversationName"`
|
|
}
|
|
Etcd struct {
|
|
EtcdSchema string `yaml:"etcdSchema"`
|
|
EtcdAddr []string `yaml:"etcdAddr"`
|
|
}
|
|
Log struct {
|
|
StorageLocation string `yaml:"storageLocation"`
|
|
RotationTime int `yaml:"rotationTime"`
|
|
RemainRotationCount uint `yaml:"remainRotationCount"`
|
|
RemainLogLevel uint `yaml:"remainLogLevel"`
|
|
ElasticSearchSwitch bool `yaml:"elasticSearchSwitch"`
|
|
ElasticSearchAddr []string `yaml:"elasticSearchAddr"`
|
|
ElasticSearchUser string `yaml:"elasticSearchUser"`
|
|
ElasticSearchPassword string `yaml:"elasticSearchPassword"`
|
|
}
|
|
ModuleName struct {
|
|
LongConnSvrName string `yaml:"longConnSvrName"`
|
|
MsgTransferName string `yaml:"msgTransferName"`
|
|
PushName string `yaml:"pushName"`
|
|
}
|
|
LongConnSvr struct {
|
|
WebsocketPort []int `yaml:"openImWsPort"`
|
|
WebsocketMaxConnNum int `yaml:"websocketMaxConnNum"`
|
|
WebsocketMaxMsgLen int `yaml:"websocketMaxMsgLen"`
|
|
WebsocketTimeOut int `yaml:"websocketTimeOut"`
|
|
}
|
|
|
|
Push struct {
|
|
Tpns struct {
|
|
Ios struct {
|
|
AccessID string `yaml:"accessID"`
|
|
SecretKey string `yaml:"secretKey"`
|
|
}
|
|
Android struct {
|
|
AccessID string `yaml:"accessID"`
|
|
SecretKey string `yaml:"secretKey"`
|
|
}
|
|
Enable bool `yaml:"enable"`
|
|
}
|
|
Jpns struct {
|
|
AppKey string `yaml:"appKey"`
|
|
MasterSecret string `yaml:"masterSecret"`
|
|
PushUrl string `yaml:"pushUrl"`
|
|
PushIntent string `yaml:"pushIntent"`
|
|
Enable bool `yaml:"enable"`
|
|
}
|
|
Getui struct {
|
|
PushUrl string `yaml:"pushUrl"`
|
|
AppKey string `yaml:"appKey"`
|
|
Enable bool `yaml:"enable"`
|
|
Intent string `yaml:"intent"`
|
|
MasterSecret string `yaml:"masterSecret"`
|
|
}
|
|
}
|
|
Manager struct {
|
|
AppManagerUid []string `yaml:"appManagerUid"`
|
|
Secrets []string `yaml:"secrets"`
|
|
}
|
|
|
|
Kafka struct {
|
|
Ws2mschat struct {
|
|
Addr []string `yaml:"addr"`
|
|
Topic string `yaml:"topic"`
|
|
}
|
|
Ms2pschat struct {
|
|
Addr []string `yaml:"addr"`
|
|
Topic string `yaml:"topic"`
|
|
}
|
|
ConsumerGroupID struct {
|
|
MsgToMongo string `yaml:"msgToMongo"`
|
|
MsgToMySql string `yaml:"msgToMySql"`
|
|
MsgToPush string `yaml:"msgToPush"`
|
|
}
|
|
}
|
|
Secret string `yaml:"secret"`
|
|
MultiLoginPolicy int `yaml:"multiloginpolicy"`
|
|
ChatPersistenceMysql bool `yaml:"chatPersistenceMysql"`
|
|
|
|
TokenPolicy struct {
|
|
AccessSecret string `yaml:"accessSecret"`
|
|
AccessExpire int64 `yaml:"accessExpire"`
|
|
}
|
|
MessageVerify struct {
|
|
FriendVerify bool `yaml:"friendVerify"`
|
|
}
|
|
IOSPush struct {
|
|
PushSound string `yaml:"pushSound"`
|
|
BadgeCount bool `yaml:"badgeCount"`
|
|
}
|
|
|
|
Callback struct {
|
|
CallbackUrl string `yaml:"callbackUrl"`
|
|
CallbackBeforeSendSingleMsg callBackConfig `yaml:"callbackbeforeSendSingleMsg"`
|
|
CallbackAfterSendSingleMsg callBackConfig `yaml:"callbackAfterSendSingleMsg"`
|
|
CallbackBeforeSendGroupMsg callBackConfig `yaml:"callbackBeforeSendGroupMsg"`
|
|
CallbackAfterSendGroupMsg callBackConfig `yaml:"callbackAfterSendGroupMsg"`
|
|
CallbackWordFilter callBackConfig `yaml:"callbackWordFilter"`
|
|
} `yaml:"callback"`
|
|
Notification struct {
|
|
///////////////////////group/////////////////////////////
|
|
GroupCreated struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupCreated"`
|
|
|
|
GroupInfoSet struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupInfoSet"`
|
|
|
|
JoinGroupApplication struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"joinGroupApplication"`
|
|
|
|
MemberQuit struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"memberQuit"`
|
|
|
|
GroupApplicationAccepted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupApplicationAccepted"`
|
|
|
|
GroupApplicationRejected struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupApplicationRejected"`
|
|
|
|
GroupOwnerTransferred struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupOwnerTransferred"`
|
|
|
|
MemberKicked struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"memberKicked"`
|
|
|
|
MemberInvited struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"memberInvited"`
|
|
|
|
MemberEnter struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"memberEnter"`
|
|
|
|
GroupDismissed struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupDismissed"`
|
|
|
|
GroupMuted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupMuted"`
|
|
|
|
GroupCancelMuted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupCancelMuted"`
|
|
|
|
GroupMemberMuted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupMemberMuted"`
|
|
|
|
GroupMemberCancelMuted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupMemberCancelMuted"`
|
|
GroupMemberInfoSet struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"groupMemberInfoSet"`
|
|
OrganizationChanged struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"organizationChanged"`
|
|
|
|
////////////////////////user///////////////////////
|
|
UserInfoUpdated struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"userInfoUpdated"`
|
|
|
|
//////////////////////friend///////////////////////
|
|
FriendApplication struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendApplicationAdded"`
|
|
FriendApplicationApproved struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendApplicationApproved"`
|
|
|
|
FriendApplicationRejected struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendApplicationRejected"`
|
|
|
|
FriendAdded struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendAdded"`
|
|
|
|
FriendDeleted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendDeleted"`
|
|
FriendRemarkSet struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"friendRemarkSet"`
|
|
BlackAdded struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"blackAdded"`
|
|
BlackDeleted struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"blackDeleted"`
|
|
ConversationOptUpdate struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"conversationOptUpdate"`
|
|
ConversationSetPrivate struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips struct {
|
|
OpenTips string `yaml:"openTips"`
|
|
CloseTips string `yaml:"closeTips"`
|
|
} `yaml:"defaultTips"`
|
|
} `yaml:"conversationSetPrivate"`
|
|
WorkMomentsNotification struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"workMomentsNotification"`
|
|
JoinDepartmentNotification struct {
|
|
Conversation PConversation `yaml:"conversation"`
|
|
OfflinePush POfflinePush `yaml:"offlinePush"`
|
|
DefaultTips PDefaultTips `yaml:"defaultTips"`
|
|
} `yaml:"joinDepartmentNotification"`
|
|
}
|
|
Demo struct {
|
|
Port []int `yaml:"openImDemoPort"`
|
|
AliSMSVerify struct {
|
|
AccessKeyID string `yaml:"accessKeyId"`
|
|
AccessKeySecret string `yaml:"accessKeySecret"`
|
|
SignName string `yaml:"signName"`
|
|
VerificationCodeTemplateCode string `yaml:"verificationCodeTemplateCode"`
|
|
}
|
|
SuperCode string `yaml:"superCode"`
|
|
CodeTTL int `yaml:"codeTTL"`
|
|
Mail struct {
|
|
Title string `yaml:"title"`
|
|
SenderMail string `yaml:"senderMail"`
|
|
SenderAuthorizationCode string `yaml:"senderAuthorizationCode"`
|
|
SmtpAddr string `yaml:"smtpAddr"`
|
|
SmtpPort int `yaml:"smtpPort"`
|
|
}
|
|
TestDepartMentID string `yaml:"testDepartMentID"`
|
|
}
|
|
Rtc struct {
|
|
Port int `yaml:"port"`
|
|
Address string `yaml:"address"`
|
|
} `yaml:"rtc"`
|
|
}
|
|
type PConversation struct {
|
|
ReliabilityLevel int `yaml:"reliabilityLevel"`
|
|
UnreadCount bool `yaml:"unreadCount"`
|
|
}
|
|
|
|
type POfflinePush struct {
|
|
PushSwitch bool `yaml:"switch"`
|
|
Title string `yaml:"title"`
|
|
Desc string `yaml:"desc"`
|
|
Ext string `yaml:"ext"`
|
|
}
|
|
type PDefaultTips struct {
|
|
Tips string `yaml:"tips"`
|
|
}
|
|
|
|
func init() {
|
|
//path, _ := os.Getwd()
|
|
//bytes, err := ioutil.ReadFile(path + "/config/config.yaml")
|
|
// if we cd Open-IM-Server/src/utils and run go test
|
|
// it will panic cannot find config/config.yaml
|
|
|
|
cfgName := os.Getenv("CONFIG_NAME")
|
|
if len(cfgName) == 0 {
|
|
cfgName = Root + "/config/config.yaml"
|
|
}
|
|
|
|
viper.SetConfigFile(cfgName)
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
bytes, err := ioutil.ReadFile(cfgName)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
if err = yaml.Unmarshal(bytes, &Config); err != nil {
|
|
panic(err.Error())
|
|
}
|
|
}
|