mirror of
https://github.com/gotoeasy/glogcenter.git
synced 2025-09-15 12:58:34 +08:00
371 lines
15 KiB
Go
371 lines
15 KiB
Go
/**
|
||
* 系统配置
|
||
* 1）统一管理系统的全部配置信息
|
||
* 2）所有配置都有默认值以便直接使用
|
||
* 3）所有配置都可以通过环境变量设定覆盖，方便自定义配置，方便容器化部署
|
||
*/
|
||
package conf
|
||
|
||
import (
|
||
"runtime"
|
||
"sort"
|
||
"strings"
|
||
|
||
"github.com/gotoeasy/glang/cmn"
|
||
)
|
||
|
||
const LogTransferAdd = "/v1/log/transferAdd"
|
||
const SysUserTransferSave = "/v1/sysuser/transferSave"
|
||
const SysUserTransferChgPsw = "/v1/sysuser/transferChgPsw"
|
||
const SysUserTransferDel = "/v1/sysuser/transferDel"
|
||
const UserTransferLogin = "/v1/user/transferLogin"
|
||
|
||
var storeRoot string = "/glogcenter" // 【固定】容器化缘故，不适合修改
|
||
var serverPort string = "8080" // 【固定】容器化缘故，不适合修改
|
||
var contextPath string = "/glc" // 【固定】容器化缘故，不适合修改
|
||
var storeChanLength int
|
||
var maxIdleTime int
|
||
var storeNameAutoAddDate bool
|
||
var serverUrl string
|
||
var serverIp string
|
||
var enableSecurityKey bool
|
||
var securityKey string
|
||
var headerSecurityKey string
|
||
var enableAmqpConsume bool
|
||
var enableWebGzip bool
|
||
var amqpAddr string
|
||
var amqpQueueName string
|
||
var amqpJsonFormat bool
|
||
var saveDays int
|
||
var enableLogin bool
|
||
var username string
|
||
var password string
|
||
var sessionTimeout int
|
||
var clusterMode bool
|
||
var clusterUrls []string
|
||
var enableBackup bool
|
||
var glcGroup string
|
||
var minioUrl string
|
||
var minioUser string
|
||
var minioPassword string
|
||
var minioBucket string
|
||
var enableUploadMinio bool
|
||
var goMaxProcess int
|
||
var goMaxProcessIdx int
|
||
var enableCors bool
|
||
var pageSize int
|
||
var nearSearchSize int
|
||
var mulitLineSearch bool
|
||
var testMode bool
|
||
var tokenSalt string
|
||
var aryWhite []string
|
||
var aryBlack []string
|
||
var ipAddCity bool
|
||
var enableChatAi bool
|
||
|
||
func init() {
|
||
UpdateConfigByEnv()
|
||
}
|
||
|
||
func UpdateConfigByEnv() {
|
||
// 读取环境变量初始化配置，各配置都有默认值
|
||
storeChanLength = cmn.GetEnvInt("GLC_STORE_CHAN_LENGTH", 64) // 【X】存储通道长度
|
||
maxIdleTime = cmn.GetEnvInt("GLC_MAX_IDLE_TIME", 300) // 【X】最大闲置时间（秒）,超过闲置时间将自动关闭，0时表示不关闭
|
||
storeNameAutoAddDate = cmn.GetEnvBool("GLC_STORE_NAME_AUTO_ADD_DATE", true) // 存储名是否自动添加日期（日志量大通常按日单位区分存储），默认true
|
||
serverUrl = cmn.GetEnvStr("GLC_SERVER_URL", "") // 服务URL，默认“”，集群配置时自动获取地址可能不对，可通过这个设定
|
||
serverIp = cmn.GetEnvStr("GLC_SERVER_IP", "") // 【X】服务IP，默认“”，当“”时会自动获取
|
||
enableSecurityKey = cmn.GetEnvBool("GLC_ENABLE_SECURITY_KEY", false) // web服务是否开启API秘钥校验，默认false
|
||
headerSecurityKey = cmn.GetEnvStr("GLC_HEADER_SECURITY_KEY", "X-GLC-AUTH") // web服务API秘钥的header键名
|
||
securityKey = cmn.GetEnvStr("GLC_SECURITY_KEY", "glogcenter") // web服务API秘钥
|
||
enableWebGzip = cmn.GetEnvBool("GLC_ENABLE_WEB_GZIP", false) // web服务是否开启Gzip
|
||
enableAmqpConsume = cmn.GetEnvBool("GLC_ENABLE_AMQP_CONSUME", false) // 是否开启rabbitMq消费者接收日志
|
||
amqpAddr = cmn.GetEnvStr("GLC_AMQP_ADDR", "") // rabbitMq连接地址，例："amqp://user:password@ip:port/"
|
||
amqpQueueName = cmn.GetEnvStr("GLC_AMQP_QUEUE_NAME", "glc-log-queue") // rabbitMq队列名
|
||
amqpJsonFormat = cmn.GetEnvBool("GLC_AMQP_JSON_FORMAT", true) // rabbitMq消息文本是否为json格式，默认true
|
||
saveDays = cmn.GetEnvInt("GLC_SAVE_DAYS", 180) // 日志分仓时的保留天数(0~1200)，0表示不自动删除，默认180天
|
||
enableLogin = cmn.GetEnvBool("GLC_ENABLE_LOGIN", false) // 是否开启用户密码登录，默认“false”
|
||
sessionTimeout = cmn.GetEnvInt("GLC_SESSION_TIMEOUT", 30) // 登录会话超时时间，默认“30”分钟
|
||
username = cmn.GetEnvStr("GLC_USERNAME", "glc") // 登录用户名，默认“glc”
|
||
password = cmn.GetEnvStr("GLC_PASSWORD", "GLogCenter100%666") // 登录密码，默认“GLogCenter100%666”
|
||
tokenSalt = cmn.GetEnvStr("GLC_TOKEN_SALT", "") // 令牌盐，默认“”
|
||
aryWhite = cmn.Split(cmn.GetEnvStr("GLC_WHITE_LIST", ""), ",") // IP或区域白名单，逗号分隔，默认“”
|
||
aryBlack = cmn.Split(cmn.GetEnvStr("GLC_BLACK_LIST", ""), ",") // IP或区域黑名单，逗号分隔，单个*代表全部，内网地址不受限制，默认“”
|
||
ipAddCity = cmn.GetEnvBool("GLC_IP_ADD_CITY", false) // IP是否要自动附加城市信息，默认false
|
||
clusterMode = cmn.GetEnvBool("GLC_CLUSTER_MODE", false) // 是否开启集群模式，默认false
|
||
splitUrls(cmn.GetEnvStr("GLC_CLUSTER_URLS", "")) // 从服务器地址，多个时逗号分开，默认“”
|
||
enableBackup = cmn.GetEnvBool("GLC_ENABLE_BACKUP", false) // 【X】是否开启备份，默认false
|
||
glcGroup = cmn.GetEnvStr("GLC_GROUP", "default") // 【X】日志中心分组名，默认“default”
|
||
minioUrl = cmn.GetEnvStr("GLC_MINIO_URL", "") // 【X】MINIO地址，默认“”
|
||
minioUser = cmn.GetEnvStr("GLC_MINIO_USER", "") // 【X】MINIO用户名，默认“”
|
||
minioPassword = cmn.GetEnvStr("GLC_MINIO_PASS", "") // 【X】MINIO密码，默认“”
|
||
minioBucket = cmn.GetEnvStr("GLC_MINIO_BUCKET", "") // 【X】MINIO桶名，默认“”
|
||
enableUploadMinio = cmn.GetEnvBool("GLC_ENABLE_UPLOAD_MINIO", false) // 【X】是否开启上传备份至MINIO服务器，默认false
|
||
goMaxProcess = getGoMaxProcessConf(cmn.GetEnvInt("GLC_GOMAXPROCS", -1)) // 使用的最大CPU数量，默认是最大CPU数量（设定值不在实际数量范围是按最大看待）
|
||
goMaxProcessIdx = getGoMaxProcessConf(cmn.GetEnvInt("GLC_GOMAXPROCS_IDX", -1)) // 创建索引使用的最大协程数量，默认是最大CPU数量（设定值不在实际数量范围是按最大看待）
|
||
enableCors = cmn.GetEnvBool("GLC_ENABLE_CORS", false) // 是否允许跨域，默认false
|
||
pageSize = getPageSizeConf(cmn.GetEnvInt("GLC_PAGE_SIZE", 100)) // 每次检索件数，默认100（有效范围1~1000）
|
||
nearSearchSize = getNearSearchSizeConf(cmn.GetEnvInt("GLC_NEAR_SEARCH_SIZE", 200)) // 定位相邻检索的查询件数，默认200（有效范围50~1000）
|
||
mulitLineSearch = cmn.GetEnvBool("GLC_SEARCH_MULIT_LINE", false) // 是否检索日志的全部行（日志可能有换行），默认false仅第一行
|
||
testMode = cmn.GetEnvBool("GLC_TEST_MODE", false) // 是否测试模式，默认false
|
||
enableChatAi = cmn.GetEnvBool("GLC_ENABLE_CHATAI", true) // 是否开启GLC智能助手，默认true
|
||
}
|
||
|
||
// 取配置： 是否开启GLC智能助手
|
||
func IsEnableChatAi() bool {
|
||
return enableChatAi
|
||
}
|
||
|
||
// 取配置： 定位相邻检索的查询件数，可通过环境变量“GLC_NEAR_SEARCH_SIZE”设定，默认200件
|
||
func GetNearSearchSize() int {
|
||
return nearSearchSize
|
||
}
|
||
|
||
// 取配置： IP是否要自动附加城市信息，默认false
|
||
func IsIpAddCity() bool {
|
||
return ipAddCity
|
||
}
|
||
|
||
// 取配置： 登录会话超时时间，可通过环境变量“GLC_SESSION_TIMEOUT”设定，默认“30”分钟
|
||
func GetSessionTimeout() int {
|
||
return sessionTimeout
|
||
}
|
||
|
||
// 取配置： 白名单，可通过环境变量“GLC_WHITE_LIST”设定，默认“”
|
||
func GetWhiteList() []string {
|
||
return aryWhite
|
||
}
|
||
|
||
// 取配置： 黑名单，可通过环境变量“GLC_BLACK_LIST”设定，默认“”
|
||
func GetBlackList() []string {
|
||
return aryBlack
|
||
}
|
||
|
||
// 取配置： 令牌盐，可通过环境变量“GLC_TOKEN_SALT”设定，默认“”
|
||
func GetTokenSalt() string {
|
||
return tokenSalt
|
||
}
|
||
|
||
// 取配置： 是否测试模式，可通过环境变量“GLC_TEST_MODE”设定，默认false
|
||
func IsTestMode() bool {
|
||
return testMode
|
||
}
|
||
|
||
// 取配置： 是否检索日志的全部行，可通过环境变量“GLC_SEARCH_MULIT_LINE”设定，默认false
|
||
func IsMulitLineSearch() bool {
|
||
return mulitLineSearch
|
||
}
|
||
|
||
// 取配置： 每次检索件数，可通过环境变量“GLC_PAGE_SIZE”设定，默认100（有效范围1~1000）
|
||
func GetPageSize() int {
|
||
return pageSize
|
||
}
|
||
func getNearSearchSizeConf(n int) int {
|
||
if n < 50 {
|
||
n = 50
|
||
}
|
||
if n > 1000 {
|
||
n = 1000
|
||
}
|
||
return n
|
||
}
|
||
func getPageSizeConf(n int) int {
|
||
if n < 1 {
|
||
n = 1
|
||
}
|
||
if n > 1000 {
|
||
n = 1000
|
||
}
|
||
return n
|
||
}
|
||
|
||
// 取配置： 是否允许跨域，可通过环境变量“GLC_ENABLE_CROSS”设定，默认false
|
||
func IsEnableCors() bool {
|
||
return enableCors
|
||
}
|
||
|
||
// 取配置： 创建索引使用的最大协程数量，默认是最大CPU数量（设定值不在实际数量范围是按最大看待）
|
||
func GetGoMaxProcessIdx() int {
|
||
return goMaxProcessIdx
|
||
}
|
||
|
||
// 取配置： 使用的最大CPU数量，可通过环境变量“GLC_GOMAXPROCS”设定，默认最大CPU数量
|
||
func GetGoMaxProcess() int {
|
||
return goMaxProcess
|
||
}
|
||
func getGoMaxProcessConf(n int) int {
|
||
max := runtime.NumCPU()
|
||
if n < 1 || n > max {
|
||
n = max
|
||
}
|
||
return n
|
||
}
|
||
|
||
// 取配置： 服务URL，集群配置时自动获取地址可能不对，可通过环境变量“GLC_ENABLE_BACKUP”设定，默认“”
|
||
func GetServerUrl() string {
|
||
return serverUrl
|
||
}
|
||
|
||
// 取配置： 是否开启MINIO备份，可通过环境变量“GLC_ENABLE_BACKUP”设定，默认false
|
||
func IsEnableBackup() bool {
|
||
return enableBackup
|
||
}
|
||
|
||
// 取配置： 日志中心分组名，可通过环境变量“GLC_GROUP”设定，默认“default”
|
||
func GetGlcGroup() string {
|
||
return glcGroup
|
||
}
|
||
|
||
// 取配置： MINIO地址，可通过环境变量“GLC_MINIO_URL”设定，默认“”
|
||
func GetMinioUrl() string {
|
||
return minioUrl
|
||
}
|
||
|
||
// 取配置： MINIO用户名，可通过环境变量“GLC_MINIO_USER”设定，默认“”
|
||
func GetMinioUser() string {
|
||
return minioUser
|
||
}
|
||
|
||
// 取配置： MINIO密码，可通过环境变量“GLC_MINIO_PASS”设定，默认“”
|
||
func GetMinioPassword() string {
|
||
return minioPassword
|
||
}
|
||
|
||
// 取配置： MINIO桶名，可通过环境变量“GLC_MINIO_BUCKET”设定，默认“”
|
||
func GetMinioBucket() string {
|
||
return minioBucket
|
||
}
|
||
|
||
// 取配置： 是否开启上传备份至MINIO服务器，可通过环境变量“GLC_ENABLE_UPLOAD_MINIO”设定，默认false
|
||
func IsEnableUploadMinio() bool {
|
||
return enableUploadMinio
|
||
}
|
||
|
||
// 取配置： 是否开启转发日志到其他GLC服务，可通过环境变量“GLC_CLUSTER_MODE”设定，默认false
|
||
func IsClusterMode() bool {
|
||
return clusterMode
|
||
}
|
||
|
||
// 取配置： 从服务器地址，可通过环境变量“GLC_SLAVE_HOSTS”设定，默认“”
|
||
func GetClusterUrls() []string {
|
||
return clusterUrls
|
||
}
|
||
|
||
func splitUrls(str string) {
|
||
hosts := strings.Split(str, ";")
|
||
for i := 0; i < len(hosts); i++ {
|
||
url := strings.TrimSpace(hosts[i])
|
||
if url != "" {
|
||
clusterUrls = append(clusterUrls, url)
|
||
}
|
||
}
|
||
|
||
// 倒序
|
||
sort.Slice(clusterUrls, func(i, j int) bool {
|
||
return clusterUrls[i] > clusterUrls[j]
|
||
})
|
||
}
|
||
|
||
// 取配置： 是否开启用户密码登录，可通过环境变量“GLC_ENABLE_LOGIN”设定，默认“false”
|
||
func IsEnableLogin() bool {
|
||
return enableLogin
|
||
}
|
||
|
||
// 取配置： 登录用户名，可通过环境变量“GLC_USERNAME”设定，默认“glc”
|
||
func GetUsername() string {
|
||
return username
|
||
}
|
||
|
||
// 存取配置： 登录用户名，可通过环境变量“GLC_PASSWORD”设定，默认“glogcenter”
|
||
func GetPassword() string {
|
||
return password
|
||
}
|
||
func SetPassword(psw string) {
|
||
password = psw
|
||
}
|
||
|
||
// 取配置： 日志分仓时的保留天数(0~180)，0表示不自动删除，可通过环境变量“GLC_SAVE_DAYS”设定，默认180天
|
||
func GetSaveDays() int {
|
||
if saveDays < 0 {
|
||
saveDays = 0
|
||
}
|
||
if saveDays > 1200 {
|
||
saveDays = 1200
|
||
}
|
||
return saveDays
|
||
}
|
||
|
||
// 取配置： rabbitMq消息文本是否为json格式，可通过环境变量“GLC_AMQP_JSON_FORMAT”设定，默认值“true”
|
||
func IsAmqpJsonFormat() bool {
|
||
return amqpJsonFormat
|
||
}
|
||
|
||
// 取配置： rabbitMq连接地址，可通过环境变量“GLC_AMQP_ADDR”设定，默认值“”
|
||
func GetAmqpQueueName() string {
|
||
return amqpQueueName
|
||
}
|
||
|
||
// 取配置： rabbitMq连接地址，可通过环境变量“GLC_AMQP_ADDR”设定，默认值“”
|
||
func GetAmqpAddr() string {
|
||
return amqpAddr
|
||
}
|
||
|
||
// 取配置： 是否开启rabbitMq消费者接收日志，可通过环境变量“GLC_ENABLE_AMQP_CONSUME”设定，默认值“false”
|
||
func IsEnableAmqpConsume() bool {
|
||
return enableAmqpConsume
|
||
}
|
||
|
||
// 取配置： web服务API秘钥的header键名，可通过环境变量“GLC_HEADER_SECURITY_KEY”设定，默认值“X-GLC-AUTH”
|
||
func IsEnableWebGzip() bool {
|
||
return enableWebGzip
|
||
}
|
||
|
||
// 取配置： web服务API秘钥的header键名，可通过环境变量“GLC_HEADER_SECURITY_KEY”设定，默认值“X-GLC-AUTH”
|
||
func IsEnableSecurityKey() bool {
|
||
return enableSecurityKey
|
||
}
|
||
|
||
// 取配置： web服务API秘钥的header键名，可通过环境变量“GLC_HEADER_SECURITY_KEY”设定，默认值“X-GLC-AUTH”
|
||
func GetHeaderSecurityKey() string {
|
||
return headerSecurityKey
|
||
}
|
||
|
||
// 取配置： web服务API秘钥，可通过环境变量“GLC_SECURITY_KEY”设定，默认值“glogcenter”
|
||
func GetSecurityKey() string {
|
||
return securityKey
|
||
}
|
||
|
||
// 取配置： web服务端口，可通过环境变量“GLC_CONTEXT_PATH”设定，默认值“8080”
|
||
func GetContextPath() string {
|
||
return contextPath
|
||
}
|
||
|
||
// 取配置： 服务IP，可通过环境变量“GLC_SERVER_IP”设定，默认值“”，自动获取
|
||
func GetServerIp() string {
|
||
return serverIp
|
||
}
|
||
|
||
// 取配置： web服务端口，可通过环境变量“GLC_SERVER_PORT”设定，默认值“8080”
|
||
func GetServerPort() string {
|
||
return serverPort
|
||
}
|
||
|
||
// 取配置：存储根目录，可通过环境变量“GLC_STORE_ROOT”设定，默认值“/glogcenter”
|
||
func GetStorageRoot() string {
|
||
return storeRoot
|
||
}
|
||
|
||
// 取配置：存储通道长度，可通过环境变量“GLC_STORE_CHAN_LENGTH”设定，默认值“64”
|
||
func GetStoreChanLength() int {
|
||
return storeChanLength
|
||
}
|
||
|
||
// 取配置：最大闲置时间（秒），可通过环境变量“GLC_MAX_IDLE_TIME”设定，默认值“180”，超过闲置时间将自动关闭存储器，0时表示不关闭
|
||
func GetMaxIdleTime() int {
|
||
return maxIdleTime
|
||
}
|
||
|
||
// 取配置：存储名是否自动添加日期（日志量大通常按日单位区分存储），可通过环境变量“GLC_STORE_NAME_AUTO_ADD_DATE”设定，默认值“true”
|
||
func IsStoreNameAutoAddDate() bool {
|
||
return storeNameAutoAddDate
|
||
}
|