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.
538 lines
20 KiB
538 lines
20 KiB
// 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 (
|
|
"github.com/openimsdk/tools/db/mongoutil"
|
|
"github.com/openimsdk/tools/db/redisutil"
|
|
"github.com/openimsdk/tools/mq/kafka"
|
|
"github.com/openimsdk/tools/s3/cos"
|
|
"github.com/openimsdk/tools/s3/minio"
|
|
"github.com/openimsdk/tools/s3/oss"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type CacheConfig struct {
|
|
Topic string `mapstructure:"topic"`
|
|
SlotNum int `mapstructure:"slotNum"`
|
|
SlotSize int `mapstructure:"slotSize"`
|
|
SuccessExpire int `mapstructure:"successExpire"`
|
|
FailedExpire int `mapstructure:"failedExpire"`
|
|
}
|
|
|
|
type LocalCache struct {
|
|
User CacheConfig `mapstructure:"user"`
|
|
Group CacheConfig `mapstructure:"group"`
|
|
Friend CacheConfig `mapstructure:"friend"`
|
|
Conversation CacheConfig `mapstructure:"conversation"`
|
|
}
|
|
|
|
type Log struct {
|
|
StorageLocation string `mapstructure:"storageLocation"`
|
|
RotationTime uint `mapstructure:"rotationTime"`
|
|
RemainRotationCount uint `mapstructure:"remainRotationCount"`
|
|
RemainLogLevel int `mapstructure:"remainLogLevel"`
|
|
IsStdout bool `mapstructure:"isStdout"`
|
|
IsJson bool `mapstructure:"isJson"`
|
|
WithStack bool `mapstructure:"withStack"`
|
|
}
|
|
|
|
type Minio struct {
|
|
Bucket string `mapstructure:"bucket"`
|
|
AccessKeyID string `mapstructure:"accessKeyID"`
|
|
SecretAccessKey string `mapstructure:"secretAccessKey"`
|
|
SessionToken string `mapstructure:"sessionToken"`
|
|
InternalAddress string `mapstructure:"internalAddress"`
|
|
ExternalAddress string `mapstructure:"externalAddress"`
|
|
PublicRead bool `mapstructure:"publicRead"`
|
|
}
|
|
|
|
type Mongo struct {
|
|
URI string `mapstructure:"uri"`
|
|
Address []string `mapstructure:"address"`
|
|
Database string `mapstructure:"database"`
|
|
Username string `mapstructure:"username"`
|
|
Password string `mapstructure:"password"`
|
|
MaxPoolSize int `mapstructure:"maxPoolSize"`
|
|
MaxRetry int `mapstructure:"maxRetry"`
|
|
}
|
|
type Kafka struct {
|
|
Username string `mapstructure:"username"`
|
|
Password string `mapstructure:"password"`
|
|
ProducerAck string `mapstructure:"producerAck"`
|
|
CompressType string `mapstructure:"compressType"`
|
|
Address []string `mapstructure:"address"`
|
|
ToRedisTopic string `mapstructure:"toRedisTopic"`
|
|
ToMongoTopic string `mapstructure:"toMongoTopic"`
|
|
ToPushTopic string `mapstructure:"toPushTopic"`
|
|
ToRedisGroupID string `mapstructure:"toRedisGroupID"`
|
|
ToMongoGroupID string `mapstructure:"toMongoGroupID"`
|
|
ToPushGroupID string `mapstructure:"toPushGroupID"`
|
|
Tls TLSConfig `mapstructure:"tls"`
|
|
}
|
|
type TLSConfig struct {
|
|
EnableTLS bool `mapstructure:"enableTLS"`
|
|
CACrt string `mapstructure:"caCrt"`
|
|
ClientCrt string `mapstructure:"clientCrt"`
|
|
ClientKey string `mapstructure:"clientKey"`
|
|
ClientKeyPwd string `mapstructure:"clientKeyPwd"`
|
|
InsecureSkipVerify bool `mapstructure:"insecureSkipVerify"`
|
|
}
|
|
|
|
type API struct {
|
|
Api struct {
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"api"`
|
|
Prometheus struct {
|
|
Enable bool `mapstructure:"enable"`
|
|
Ports []int `mapstructure:"ports"`
|
|
GrafanaURL string `mapstructure:"grafanaURL"`
|
|
} `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type CronTask struct {
|
|
ChatRecordsClearTime string `mapstructure:"chatRecordsClearTime"`
|
|
RetainChatRecords int `mapstructure:"retainChatRecords"`
|
|
}
|
|
|
|
type OfflinePushConfig struct {
|
|
Enable bool `mapstructure:"enable"`
|
|
Title string `mapstructure:"title"`
|
|
Desc string `mapstructure:"desc"`
|
|
Ext string `mapstructure:"ext"`
|
|
}
|
|
|
|
type NotificationConfig struct {
|
|
IsSendMsg bool `mapstructure:"isSendMsg"`
|
|
ReliabilityLevel int `mapstructure:"reliabilityLevel"`
|
|
UnreadCount bool `mapstructure:"unreadCount"`
|
|
OfflinePush OfflinePushConfig `mapstructure:"offlinePush"`
|
|
}
|
|
|
|
type Notification struct {
|
|
GroupCreated NotificationConfig `mapstructure:"groupCreated"`
|
|
GroupInfoSet NotificationConfig `mapstructure:"groupInfoSet"`
|
|
JoinGroupApplication NotificationConfig `mapstructure:"joinGroupApplication"`
|
|
MemberQuit NotificationConfig `mapstructure:"memberQuit"`
|
|
GroupApplicationAccepted NotificationConfig `mapstructure:"groupApplicationAccepted"`
|
|
GroupApplicationRejected NotificationConfig `mapstructure:"groupApplicationRejected"`
|
|
GroupOwnerTransferred NotificationConfig `mapstructure:"groupOwnerTransferred"`
|
|
MemberKicked NotificationConfig `mapstructure:"memberKicked"`
|
|
MemberInvited NotificationConfig `mapstructure:"memberInvited"`
|
|
MemberEnter NotificationConfig `mapstructure:"memberEnter"`
|
|
GroupDismissed NotificationConfig `mapstructure:"groupDismissed"`
|
|
GroupMuted NotificationConfig `mapstructure:"groupMuted"`
|
|
GroupCancelMuted NotificationConfig `mapstructure:"groupCancelMuted"`
|
|
GroupMemberMuted NotificationConfig `mapstructure:"groupMemberMuted"`
|
|
GroupMemberCancelMuted NotificationConfig `mapstructure:"groupMemberCancelMuted"`
|
|
GroupMemberInfoSet NotificationConfig `mapstructure:"groupMemberInfoSet"`
|
|
GroupMemberSetToAdmin NotificationConfig `yaml:"groupMemberSetToAdmin"`
|
|
GroupMemberSetToOrdinary NotificationConfig `yaml:"groupMemberSetToOrdinaryUser"`
|
|
GroupInfoSetAnnouncement NotificationConfig `mapstructure:"groupInfoSetAnnouncement"`
|
|
GroupInfoSetName NotificationConfig `mapstructure:"groupInfoSetName"`
|
|
FriendApplicationAdded NotificationConfig `mapstructure:"friendApplicationAdded"`
|
|
FriendApplicationApproved NotificationConfig `mapstructure:"friendApplicationApproved"`
|
|
FriendApplicationRejected NotificationConfig `mapstructure:"friendApplicationRejected"`
|
|
FriendAdded NotificationConfig `mapstructure:"friendAdded"`
|
|
FriendDeleted NotificationConfig `mapstructure:"friendDeleted"`
|
|
FriendRemarkSet NotificationConfig `mapstructure:"friendRemarkSet"`
|
|
BlackAdded NotificationConfig `mapstructure:"blackAdded"`
|
|
BlackDeleted NotificationConfig `mapstructure:"blackDeleted"`
|
|
FriendInfoUpdated NotificationConfig `mapstructure:"friendInfoUpdated"`
|
|
UserInfoUpdated NotificationConfig `mapstructure:"userInfoUpdated"`
|
|
UserStatusChanged NotificationConfig `mapstructure:"userStatusChanged"`
|
|
ConversationChanged NotificationConfig `mapstructure:"conversationChanged"`
|
|
ConversationSetPrivate NotificationConfig `mapstructure:"conversationSetPrivate"`
|
|
}
|
|
|
|
type Prometheus struct {
|
|
Enable bool `mapstructure:"enable"`
|
|
Ports []int `mapstructure:"ports"`
|
|
}
|
|
|
|
type MsgGateway struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
LongConnSvr struct {
|
|
Ports []int `mapstructure:"ports"`
|
|
WebsocketMaxConnNum int `mapstructure:"websocketMaxConnNum"`
|
|
WebsocketMaxMsgLen int `mapstructure:"websocketMaxMsgLen"`
|
|
WebsocketTimeout int `mapstructure:"websocketTimeout"`
|
|
} `mapstructure:"longConnSvr"`
|
|
MultiLoginPolicy int `mapstructure:"multiLoginPolicy"`
|
|
}
|
|
|
|
type MsgTransfer struct {
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type Push struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
MaxConcurrentWorkers int `mapstructure:"maxConcurrentWorkers"`
|
|
Enable string `mapstructure:"enable"`
|
|
GeTui struct {
|
|
PushUrl string `mapstructure:"pushUrl"`
|
|
MasterSecret string `mapstructure:"masterSecret"`
|
|
AppKey string `mapstructure:"appKey"`
|
|
Intent string `mapstructure:"intent"`
|
|
ChannelID string `mapstructure:"channelID"`
|
|
ChannelName string `mapstructure:"channelName"`
|
|
} `mapstructure:"geTui"`
|
|
FCM struct {
|
|
ServiceAccount string `mapstructure:"serviceAccount"`
|
|
} `mapstructure:"fcm"`
|
|
JPNS struct {
|
|
AppKey string `mapstructure:"appKey"`
|
|
MasterSecret string `mapstructure:"masterSecret"`
|
|
PushURL string `mapstructure:"pushURL"`
|
|
PushIntent string `mapstructure:"pushIntent"`
|
|
} `mapstructure:"jpns"`
|
|
IOSPush struct {
|
|
PushSound string `mapstructure:"pushSound"`
|
|
BadgeCount bool `mapstructure:"badgeCount"`
|
|
Production bool `mapstructure:"production"`
|
|
} `mapstructure:"iosPush"`
|
|
}
|
|
|
|
type Auth struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
TokenPolicy struct {
|
|
Expire int64 `mapstructure:"expire"`
|
|
} `mapstructure:"tokenPolicy"`
|
|
}
|
|
|
|
type Conversation struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type Friend struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type Group struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type Msg struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
FriendVerify bool `mapstructure:"friendVerify"`
|
|
}
|
|
|
|
type Third struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
Object struct {
|
|
Enable string `mapstructure:"enable"`
|
|
Cos Cos `mapstructure:"cos"`
|
|
Oss Oss `mapstructure:"oss"`
|
|
Kodo struct {
|
|
Endpoint string `mapstructure:"endpoint"`
|
|
Bucket string `mapstructure:"bucket"`
|
|
BucketURL string `mapstructure:"bucketURL"`
|
|
AccessKeyID string `mapstructure:"accessKeyID"`
|
|
AccessKeySecret string `mapstructure:"accessKeySecret"`
|
|
SessionToken string `mapstructure:"sessionToken"`
|
|
PublicRead bool `mapstructure:"publicRead"`
|
|
} `mapstructure:"kodo"`
|
|
Aws struct {
|
|
Endpoint string `mapstructure:"endpoint"`
|
|
Region string `mapstructure:"region"`
|
|
Bucket string `mapstructure:"bucket"`
|
|
AccessKeyID string `mapstructure:"accessKeyID"`
|
|
AccessKeySecret string `mapstructure:"accessKeySecret"`
|
|
PublicRead bool `mapstructure:"publicRead"`
|
|
} `mapstructure:"aws"`
|
|
} `mapstructure:"object"`
|
|
}
|
|
type Cos struct {
|
|
BucketURL string `mapstructure:"bucketURL"`
|
|
SecretID string `mapstructure:"secretID"`
|
|
SecretKey string `mapstructure:"secretKey"`
|
|
SessionToken string `mapstructure:"sessionToken"`
|
|
PublicRead bool `mapstructure:"publicRead"`
|
|
}
|
|
type Oss struct {
|
|
Endpoint string `mapstructure:"endpoint"`
|
|
Bucket string `mapstructure:"bucket"`
|
|
BucketURL string `mapstructure:"bucketURL"`
|
|
AccessKeyID string `mapstructure:"accessKeyID"`
|
|
AccessKeySecret string `mapstructure:"accessKeySecret"`
|
|
SessionToken string `mapstructure:"sessionToken"`
|
|
PublicRead bool `mapstructure:"publicRead"`
|
|
}
|
|
|
|
type User struct {
|
|
RPC struct {
|
|
RegisterIP string `mapstructure:"registerIP"`
|
|
ListenIP string `mapstructure:"listenIP"`
|
|
Ports []int `mapstructure:"ports"`
|
|
} `mapstructure:"rpc"`
|
|
Prometheus Prometheus `mapstructure:"prometheus"`
|
|
}
|
|
|
|
type Redis struct {
|
|
Address []string `mapstructure:"address"`
|
|
Username string `mapstructure:"username"`
|
|
Password string `mapstructure:"password"`
|
|
EnablePipeline bool `mapstructure:"enablePipeline"`
|
|
ClusterMode bool `mapstructure:"clusterMode"`
|
|
DB int `mapstructure:"db"`
|
|
MaxRetry int `mapstructure:"MaxRetry"`
|
|
}
|
|
|
|
type BeforeConfig struct {
|
|
Enable bool `mapstructure:"enable"`
|
|
Timeout int `mapstructure:"timeout"`
|
|
FailedContinue bool `mapstructure:"failedContinue"`
|
|
}
|
|
|
|
type AfterConfig struct {
|
|
Enable bool `mapstructure:"enable"`
|
|
Timeout int `mapstructure:"timeout"`
|
|
}
|
|
|
|
type Share struct {
|
|
Secret string `mapstructure:"secret"`
|
|
RpcRegisterName RpcRegisterName `mapstructure:"rpcRegisterName"`
|
|
IMAdminUserID []string `mapstructure:"imAdminUserID"`
|
|
}
|
|
type RpcRegisterName struct {
|
|
User string `mapstructure:"user"`
|
|
Friend string `mapstructure:"friend"`
|
|
Msg string `mapstructure:"msg"`
|
|
Push string `mapstructure:"push"`
|
|
MessageGateway string `mapstructure:"messageGateway"`
|
|
Group string `mapstructure:"group"`
|
|
Auth string `mapstructure:"auth"`
|
|
Conversation string `mapstructure:"conversation"`
|
|
Third string `mapstructure:"third"`
|
|
}
|
|
|
|
func (r *RpcRegisterName) GetServiceNames() []string {
|
|
return []string{
|
|
r.User,
|
|
r.Friend,
|
|
r.Msg,
|
|
r.Push,
|
|
r.MessageGateway,
|
|
r.Group,
|
|
r.Auth,
|
|
r.Conversation,
|
|
r.Third,
|
|
}
|
|
}
|
|
|
|
// FullConfig stores all configurations for before and after events
|
|
type Webhooks struct {
|
|
URL string `mapstructure:"url"`
|
|
BeforeSendSingleMsg BeforeConfig `mapstructure:"beforeSendSingleMsg"`
|
|
BeforeUpdateUserInfoEx BeforeConfig `mapstructure:"beforeUpdateUserInfoEx"`
|
|
AfterUpdateUserInfoEx AfterConfig `mapstructure:"afterUpdateUserInfoEx"`
|
|
AfterSendSingleMsg AfterConfig `mapstructure:"afterSendSingleMsg"`
|
|
BeforeSendGroupMsg BeforeConfig `mapstructure:"beforeSendGroupMsg"`
|
|
BeforeMsgModify BeforeConfig `mapstructure:"beforeMsgModify"`
|
|
AfterSendGroupMsg AfterConfig `mapstructure:"afterSendGroupMsg"`
|
|
AfterUserOnline AfterConfig `mapstructure:"afterUserOnline"`
|
|
AfterUserOffline AfterConfig `mapstructure:"afterUserOffline"`
|
|
AfterUserKickOff AfterConfig `mapstructure:"afterUserKickOff"`
|
|
BeforeOfflinePush BeforeConfig `mapstructure:"beforeOfflinePush"`
|
|
BeforeOnlinePush BeforeConfig `mapstructure:"beforeOnlinePush"`
|
|
BeforeGroupOnlinePush BeforeConfig `mapstructure:"beforeGroupOnlinePush"`
|
|
BeforeAddFriend BeforeConfig `mapstructure:"beforeAddFriend"`
|
|
BeforeUpdateUserInfo BeforeConfig `mapstructure:"beforeUpdateUserInfo"`
|
|
AfterUpdateUserInfo AfterConfig `mapstructure:"afterUpdateUserInfo"`
|
|
BeforeCreateGroup BeforeConfig `mapstructure:"beforeCreateGroup"`
|
|
AfterCreateGroup AfterConfig `mapstructure:"afterCreateGroup"`
|
|
BeforeMemberJoinGroup BeforeConfig `mapstructure:"beforeMemberJoinGroup"`
|
|
BeforeSetGroupMemberInfo BeforeConfig `mapstructure:"beforeSetGroupMemberInfo"`
|
|
AfterSetGroupMemberInfo AfterConfig `mapstructure:"afterSetGroupMemberInfo"`
|
|
AfterQuitGroup AfterConfig `mapstructure:"afterQuitGroup"`
|
|
AfterKickGroupMember AfterConfig `mapstructure:"afterKickGroupMember"`
|
|
AfterDismissGroup AfterConfig `mapstructure:"afterDismissGroup"`
|
|
BeforeApplyJoinGroup BeforeConfig `mapstructure:"beforeApplyJoinGroup"`
|
|
AfterGroupMsgRead AfterConfig `mapstructure:"afterGroupMsgRead"`
|
|
AfterSingleMsgRead AfterConfig `mapstructure:"afterSingleMsgRead"`
|
|
BeforeUserRegister BeforeConfig `mapstructure:"beforeUserRegister"`
|
|
AfterUserRegister AfterConfig `mapstructure:"afterUserRegister"`
|
|
AfterTransferGroupOwner AfterConfig `mapstructure:"afterTransferGroupOwner"`
|
|
BeforeSetFriendRemark BeforeConfig `mapstructure:"beforeSetFriendRemark"`
|
|
AfterSetFriendRemark AfterConfig `mapstructure:"afterSetFriendRemark"`
|
|
AfterGroupMsgRevoke AfterConfig `mapstructure:"afterGroupMsgRevoke"`
|
|
AfterJoinGroup AfterConfig `mapstructure:"afterJoinGroup"`
|
|
BeforeInviteUserToGroup BeforeConfig `mapstructure:"beforeInviteUserToGroup"`
|
|
AfterSetGroupInfo AfterConfig `mapstructure:"afterSetGroupInfo"`
|
|
BeforeSetGroupInfo BeforeConfig `mapstructure:"beforeSetGroupInfo"`
|
|
AfterRevokeMsg AfterConfig `mapstructure:"afterRevokeMsg"`
|
|
BeforeAddBlack BeforeConfig `mapstructure:"beforeAddBlack"`
|
|
AfterAddFriend AfterConfig `mapstructure:"afterAddFriend"`
|
|
BeforeAddFriendAgree BeforeConfig `mapstructure:"beforeAddFriendAgree"`
|
|
AfterDeleteFriend AfterConfig `mapstructure:"afterDeleteFriend"`
|
|
BeforeImportFriends BeforeConfig `mapstructure:"beforeImportFriends"`
|
|
AfterImportFriends AfterConfig `mapstructure:"afterImportFriends"`
|
|
AfterRemoveBlack AfterConfig `mapstructure:"afterRemoveBlack"`
|
|
}
|
|
|
|
type ZooKeeper struct {
|
|
Schema string `mapstructure:"schema"`
|
|
Address []string `mapstructure:"address"`
|
|
Username string `mapstructure:"username"`
|
|
Password string `mapstructure:"password"`
|
|
}
|
|
|
|
type Discovery struct {
|
|
Enable string `mapstructure:"enable"`
|
|
Etcd Etcd `mapstructure:"etcd"`
|
|
ZooKeeper ZooKeeper `mapstructure:"zooKeeper"`
|
|
}
|
|
|
|
type Etcd struct {
|
|
RootDirectory string `mapstructure:"rootDirectory"`
|
|
Address []string `mapstructure:"address"`
|
|
Username string `mapstructure:"username"`
|
|
Password string `mapstructure:"password"`
|
|
}
|
|
|
|
func (m *Mongo) Build() *mongoutil.Config {
|
|
return &mongoutil.Config{
|
|
Uri: m.URI,
|
|
Address: m.Address,
|
|
Database: m.Database,
|
|
Username: m.Username,
|
|
Password: m.Password,
|
|
MaxPoolSize: m.MaxPoolSize,
|
|
MaxRetry: m.MaxRetry,
|
|
}
|
|
}
|
|
|
|
func (r *Redis) Build() *redisutil.Config {
|
|
return &redisutil.Config{
|
|
ClusterMode: r.ClusterMode,
|
|
Address: r.Address,
|
|
Username: r.Username,
|
|
Password: r.Password,
|
|
DB: r.DB,
|
|
MaxRetry: r.MaxRetry,
|
|
}
|
|
}
|
|
|
|
func (k *Kafka) Build() *kafka.Config {
|
|
return &kafka.Config{
|
|
Username: k.Username,
|
|
Password: k.Password,
|
|
ProducerAck: k.ProducerAck,
|
|
CompressType: k.CompressType,
|
|
Addr: k.Address,
|
|
TLS: kafka.TLSConfig{
|
|
EnableTLS: k.Tls.EnableTLS,
|
|
CACrt: k.Tls.CACrt,
|
|
ClientCrt: k.Tls.ClientCrt,
|
|
ClientKey: k.Tls.ClientKey,
|
|
ClientKeyPwd: k.Tls.ClientKeyPwd,
|
|
InsecureSkipVerify: k.Tls.InsecureSkipVerify,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (m *Minio) Build() *minio.Config {
|
|
formatEndpoint := func(address string) string {
|
|
if strings.HasPrefix(address, "http://") || strings.HasPrefix(address, "https://") {
|
|
return address
|
|
}
|
|
return "http://" + address
|
|
}
|
|
return &minio.Config{
|
|
Bucket: m.Bucket,
|
|
AccessKeyID: m.AccessKeyID,
|
|
SecretAccessKey: m.SecretAccessKey,
|
|
SessionToken: m.SessionToken,
|
|
PublicRead: m.PublicRead,
|
|
Endpoint: formatEndpoint(m.InternalAddress),
|
|
SignEndpoint: formatEndpoint(m.ExternalAddress),
|
|
}
|
|
}
|
|
func (c *Cos) Build() *cos.Config {
|
|
return &cos.Config{
|
|
BucketURL: c.BucketURL,
|
|
SecretID: c.SecretID,
|
|
SecretKey: c.SecretKey,
|
|
SessionToken: c.SessionToken,
|
|
PublicRead: c.PublicRead,
|
|
}
|
|
}
|
|
|
|
func (o *Oss) Build() *oss.Config {
|
|
return &oss.Config{
|
|
Endpoint: o.Endpoint,
|
|
Bucket: o.Bucket,
|
|
BucketURL: o.BucketURL,
|
|
AccessKeyID: o.AccessKeyID,
|
|
AccessKeySecret: o.AccessKeySecret,
|
|
SessionToken: o.SessionToken,
|
|
PublicRead: o.PublicRead,
|
|
}
|
|
}
|
|
|
|
func (l *CacheConfig) Failed() time.Duration {
|
|
return time.Second * time.Duration(l.FailedExpire)
|
|
}
|
|
|
|
func (l *CacheConfig) Success() time.Duration {
|
|
return time.Second * time.Duration(l.SuccessExpire)
|
|
}
|
|
|
|
func (l *CacheConfig) Enable() bool {
|
|
return l.Topic != "" && l.SlotNum > 0 && l.SlotSize > 0
|
|
}
|