1 Star 0 Fork 41

fhaoquan/lessgo

forked from andeyalee/lessgo 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
config.go 7.91 KB
一键复制 编辑 原始数据 按行查看 历史
package lessgo
import (
"fmt"
"os"
"path/filepath"
"reflect"
"strings"
confpkg "github.com/henrylee2cn/lessgo/config"
"github.com/henrylee2cn/lessgo/logs"
)
type (
// Config is the main struct for Config
config struct {
AppName string // Application name
Info Info // Application info
Debug bool // enable/disable debug mode.
CrossDomain bool
MaxMemoryMB int64 // 文件上传默认内存缓存大小,单位MB
Listen Listen
Session SessionConfig
Log LogConfig
FileCache FileCacheConfig
}
Info struct {
Version string
Description string
Email string
TermsOfServiceUrl string
License string
LicenseUrl string
}
// Listen holds for http and https related config
Listen struct {
Address string
ReadTimeout int64
WriteTimeout int64
EnableTLS bool
TLSAddress string
HTTPSKeyFile string
HTTPSCertFile string
}
// SessionConfig holds session related config
SessionConfig struct {
SessionOn bool
SessionProvider string
SessionName string
SessionGCMaxLifetime int64
SessionProviderConfig string
SessionCookieLifeTime int
SessionAutoSetCookie bool
SessionDomain string
EnableSidInHttpHeader bool // enable store/get the sessionId into/from http headers
SessionNameInHttpHeader string
EnableSidInUrlQuery bool // enable get the sessionId from Url Query params
}
// LogConfig holds Log related config
LogConfig struct {
Level int
AsyncChan int64
}
FileCacheConfig struct {
CacheSecond int64 // 静态资源缓存监测频率与缓存动态释放的最大时长,单位秒,默认600秒
SingleFileAllowMB int64 // 允许的最大文件,单位MB
MaxCapMB int64 // 最大缓存总量,单位MB
}
)
// 项目固定目录文件名称
const (
BIZ_HANDLER_DIR = "biz_handler"
BIZ_MODEL_DIR = "biz_model"
BIZ_VIEW_DIR = "biz_view"
SYS_HANDLER_DIR = "sys_handler"
SYS_MODEL_DIR = "sys_model"
SYS_VIEW_DIR = "sys_view"
STATIC_DIR = "static"
IMG_DIR = STATIC_DIR + "/img"
JS_DIR = STATIC_DIR + "/js"
CSS_DIR = STATIC_DIR + "/css"
TPL_DIR = STATIC_DIR + "/tpl"
PLUGIN_DIR = STATIC_DIR + "/plugin"
UPLOADS_DIR = "uploads"
COMMON_DIR = "common"
MIDDLEWARE_DIR = "middleware"
ROUTER_DIR = "router"
TPL_EXT = ".tpl"
STATIC_HTML_EXT = ".html"
CONFIG_DIR = "config"
APPCONFIG_FILE = CONFIG_DIR + "/app.config"
ROUTERCONFIG_FILE = CONFIG_DIR + "/virtrouter.config"
LOG_FILE = "logger/lessgo.log"
)
const (
MB = 1 << 20
)
func newConfig() *config {
return &config{
AppName: "lessgo",
Info: Info{
Version: "0.4.0",
Description: "A simple, stable, efficient and flexible web framework.",
// Host: "127.0.0.1:8080",
Email: "henrylee_cn@foxmail.com",
TermsOfServiceUrl: "https://github.com/henrylee2cn/lessgo",
License: "MIT",
LicenseUrl: "https://github.com/henrylee2cn/lessgo/raw/master/doc/LICENSE",
},
Debug: true,
CrossDomain: false,
MaxMemoryMB: 64, // 64MB
Listen: Listen{
Address: "0.0.0.0:8080",
ReadTimeout: 0,
WriteTimeout: 0,
EnableTLS: false,
TLSAddress: "0.0.0.0:10443",
HTTPSCertFile: "",
HTTPSKeyFile: "",
},
Session: SessionConfig{
SessionOn: false,
SessionProvider: "memory",
SessionName: "lessgosessionID",
SessionGCMaxLifetime: 3600,
SessionProviderConfig: "",
SessionCookieLifeTime: 0, //set cookie default is the browser life
SessionAutoSetCookie: true,
SessionDomain: "",
EnableSidInHttpHeader: false, // enable store/get the sessionId into/from http headers
SessionNameInHttpHeader: "Lessgosessionid",
EnableSidInUrlQuery: false, // enable get the sessionId from Url Query params
},
FileCache: FileCacheConfig{
CacheSecond: 600, // 600s
SingleFileAllowMB: 64, // 64MB
MaxCapMB: 256, // 256MB
},
Log: LogConfig{
Level: logs.DEBUG,
AsyncChan: 1000,
},
}
}
func (this *config) LoadMainConfig() (err error) {
fname := APPCONFIG_FILE
iniconf, err := confpkg.NewConfig("ini", fname)
if err == nil {
os.Remove(fname)
ReadSingleConfig("system", this, iniconf)
ReadSingleConfig("filecache", &this.FileCache, iniconf)
ReadSingleConfig("info", &this.Info, iniconf)
ReadSingleConfig("listen", &this.Listen, iniconf)
ReadSingleConfig("log", &this.Log, iniconf)
ReadSingleConfig("session", &this.Session, iniconf)
}
os.MkdirAll(filepath.Dir(fname), 0777)
f, err := os.Create(fname)
if err != nil {
return err
}
f.Close()
iniconf, err = confpkg.NewConfig("ini", fname)
if err != nil {
return err
}
WriteSingleConfig("system", this, iniconf)
WriteSingleConfig("filecache", &this.FileCache, iniconf)
WriteSingleConfig("info", &this.Info, iniconf)
WriteSingleConfig("listen", &this.Listen, iniconf)
WriteSingleConfig("log", &this.Log, iniconf)
WriteSingleConfig("session", &this.Session, iniconf)
return iniconf.SaveConfigFile(fname)
}
func ReadSingleConfig(section string, p interface{}, iniconf confpkg.Configer) {
pt := reflect.TypeOf(p)
if pt.Kind() != reflect.Ptr {
return
}
pt = pt.Elem()
if pt.Kind() != reflect.Struct {
return
}
pv := reflect.ValueOf(p).Elem()
for i := 0; i < pt.NumField(); i++ {
pf := pv.Field(i)
if !pf.CanSet() {
continue
}
name := pt.Field(i).Name
fullname := getfullname(section, name)
switch pf.Kind() {
case reflect.String:
str := iniconf.DefaultString(fullname, pf.String())
switch name {
case "TableFix", "ColumnFix":
pf.SetString(strings.ToLower(str))
default:
pf.SetString(str)
}
case reflect.Int, reflect.Int64:
num := int64(iniconf.DefaultInt64(fullname, pf.Int()))
switch fullname {
case "system::maxmemorymb":
if num >= 0 {
pf.SetInt(num)
}
case "filecache::cachesecond", "filecache::singlefileallowmb", "filecache::maxcapmb",
"listen::readtimeout", "listen::writetimeout",
"session::sessiongcmaxlifetime", "session::sessioncookielifetime":
if num > 0 {
pf.SetInt(num)
}
case "log::asyncchan":
if num >= 0 {
pf.SetInt(num)
}
case "log::level":
str := logLevelString(int(num))
str2 := iniconf.DefaultString(fullname, str)
num = int64(logLevelInt(str2))
if num != -10 {
pf.SetInt(num)
}
default:
pf.SetInt(num)
}
case reflect.Bool:
pf.SetBool(iniconf.DefaultBool(fullname, pf.Bool()))
}
}
}
func WriteSingleConfig(section string, p interface{}, iniconf confpkg.Configer) {
pt := reflect.TypeOf(p)
if pt.Kind() != reflect.Ptr {
return
}
pt = pt.Elem()
if pt.Kind() != reflect.Struct {
return
}
pv := reflect.ValueOf(p).Elem()
for i := 0; i < pt.NumField(); i++ {
pf := pv.Field(i)
if !pf.CanSet() {
continue
}
fullname := getfullname(section, pt.Field(i).Name)
switch pf.Kind() {
case reflect.String, reflect.Int, reflect.Int64, reflect.Bool:
switch fullname {
case "log::level":
iniconf.Set(fullname, logLevelString(int(pf.Int())))
default:
iniconf.Set(fullname, fmt.Sprint(pf.Interface()))
}
}
}
}
// section name and key name case insensitive
func getfullname(section, name string) string {
if section == "" {
return strings.ToLower(name)
}
return strings.ToLower(section + "::" + name)
}
func logLevelInt(l string) int {
switch strings.ToLower(l) {
case "debug":
return logs.DEBUG
case "info":
return logs.INFO
case "warn":
return logs.WARN
case "error":
return logs.ERROR
case "fatal":
return logs.FATAL
case "off":
return logs.OFF
}
return -10
}
func logLevelString(l int) string {
switch l {
case logs.DEBUG:
return "debug"
case logs.INFO:
return "info"
case logs.WARN:
return "warn"
case logs.ERROR:
return "error"
case logs.FATAL:
return "fatal"
case logs.OFF:
return "off"
}
return "error"
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/fhq/lessgo.git
git@gitee.com:fhq/lessgo.git
fhq
lessgo
lessgo
master

搜索帮助