mirror of https://github.com/astaxie/beego.git
353 lines
10 KiB
Go
353 lines
10 KiB
Go
// Beego (http://beego.me/)
|
|
// @description beego is an open-source, high-performance web framework for the Go programming language.
|
|
// @link http://github.com/astaxie/beego for the canonical source repository
|
|
// @license http://github.com/astaxie/beego/blob/master/LICENSE
|
|
// @authors astaxie
|
|
|
|
package beego
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/astaxie/beego/config"
|
|
"github.com/astaxie/beego/logs"
|
|
"github.com/astaxie/beego/session"
|
|
"github.com/astaxie/beego/utils"
|
|
)
|
|
|
|
var (
|
|
BeeApp *App // beego application
|
|
AppName string
|
|
AppPath string
|
|
workPath string
|
|
AppConfigPath string
|
|
StaticDir map[string]string
|
|
TemplateCache map[string]*template.Template // template caching map
|
|
StaticExtensionsToGzip []string // files with should be compressed with gzip (.js,.css,etc)
|
|
HttpAddr string
|
|
HttpPort int
|
|
HttpTLS bool
|
|
HttpCertFile string
|
|
HttpKeyFile string
|
|
RecoverPanic bool // flag of auto recover panic
|
|
AutoRender bool // flag of render template automatically
|
|
ViewsPath string
|
|
RunMode string // run mode, "dev" or "prod"
|
|
AppConfig config.ConfigContainer
|
|
GlobalSessions *session.Manager // global session mananger
|
|
SessionOn bool // flag of starting session auto. default is false.
|
|
SessionProvider string // default session provider, memory, mysql , redis ,etc.
|
|
SessionName string // the cookie name when saving session id into cookie.
|
|
SessionGCMaxLifetime int64 // session gc time for auto cleaning expired session.
|
|
SessionSavePath string // if use mysql/redis/file provider, define save path to connection info.
|
|
SessionHashFunc string // session hash generation func.
|
|
SessionHashKey string // session hash salt string.
|
|
SessionCookieLifeTime int // the life time of session id in cookie.
|
|
SessionAutoSetCookie bool // auto setcookie
|
|
UseFcgi bool
|
|
MaxMemory int64
|
|
EnableGzip bool // flag of enable gzip
|
|
DirectoryIndex bool // flag of display directory index. default is false.
|
|
EnableHotUpdate bool // flag of hot update checking by app self. default is false.
|
|
HttpServerTimeOut int64
|
|
ErrorsShow bool // flag of show errors in page. if true, show error and trace info in page rendered with error template.
|
|
XSRFKEY string // xsrf hash salt string.
|
|
EnableXSRF bool // flag of enable xsrf.
|
|
XSRFExpire int // the expiry of xsrf value.
|
|
CopyRequestBody bool // flag of copy raw request body in context.
|
|
TemplateLeft string
|
|
TemplateRight string
|
|
BeegoServerName string // beego server name exported in response header.
|
|
EnableAdmin bool // flag of enable admin module to log every request info.
|
|
AdminHttpAddr string // http server configurations for admin module.
|
|
AdminHttpPort int
|
|
FlashName string // name of the flash variable found in response header and cookie
|
|
FlashSeperator string // used to seperate flash key:value
|
|
)
|
|
|
|
func init() {
|
|
// create beego application
|
|
BeeApp = NewApp()
|
|
|
|
workPath, _ = os.Getwd()
|
|
workPath, _ = filepath.Abs(workPath)
|
|
// initialize default configurations
|
|
AppPath, _ = filepath.Abs(filepath.Dir(os.Args[0]))
|
|
|
|
AppConfigPath = filepath.Join(AppPath, "conf", "app.conf")
|
|
|
|
if workPath != AppPath {
|
|
if utils.FileExists(AppConfigPath) {
|
|
os.Chdir(AppPath)
|
|
} else {
|
|
AppConfigPath = filepath.Join(workPath, "conf", "app.conf")
|
|
}
|
|
}
|
|
|
|
StaticDir = make(map[string]string)
|
|
StaticDir["/static"] = "static"
|
|
|
|
StaticExtensionsToGzip = []string{".css", ".js"}
|
|
|
|
TemplateCache = make(map[string]*template.Template)
|
|
|
|
// set this to 0.0.0.0 to make this app available to externally
|
|
HttpAddr = ""
|
|
HttpPort = 8080
|
|
|
|
AppName = "beego"
|
|
|
|
RunMode = "dev" //default runmod
|
|
|
|
AutoRender = true
|
|
|
|
RecoverPanic = true
|
|
|
|
ViewsPath = "views"
|
|
|
|
SessionOn = false
|
|
SessionProvider = "memory"
|
|
SessionName = "beegosessionID"
|
|
SessionGCMaxLifetime = 3600
|
|
SessionSavePath = ""
|
|
SessionHashFunc = "sha1"
|
|
SessionHashKey = "beegoserversessionkey"
|
|
SessionCookieLifeTime = 0 //set cookie default is the brower life
|
|
SessionAutoSetCookie = true
|
|
|
|
UseFcgi = false
|
|
|
|
MaxMemory = 1 << 26 //64MB
|
|
|
|
EnableGzip = false
|
|
|
|
HttpServerTimeOut = 0
|
|
|
|
ErrorsShow = true
|
|
|
|
XSRFKEY = "beegoxsrf"
|
|
XSRFExpire = 0
|
|
|
|
TemplateLeft = "{{"
|
|
TemplateRight = "}}"
|
|
|
|
BeegoServerName = "beegoServer"
|
|
|
|
EnableAdmin = false
|
|
AdminHttpAddr = "127.0.0.1"
|
|
AdminHttpPort = 8088
|
|
|
|
FlashName = "BEEGO_FLASH"
|
|
FlashSeperator = "BEEGOFLASH"
|
|
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
|
|
// init BeeLogger
|
|
BeeLogger = logs.NewLogger(10000)
|
|
err := BeeLogger.SetLogger("console", "")
|
|
if err != nil {
|
|
fmt.Println("init console log error:", err)
|
|
}
|
|
|
|
err = ParseConfig()
|
|
if err != nil && !os.IsNotExist(err) {
|
|
// for init if doesn't have app.conf will not panic
|
|
Info(err)
|
|
}
|
|
}
|
|
|
|
// ParseConfig parsed default config file.
|
|
// now only support ini, next will support json.
|
|
func ParseConfig() (err error) {
|
|
AppConfig, err = config.NewConfig("ini", AppConfigPath)
|
|
if err != nil {
|
|
AppConfig = config.NewFakeConfig()
|
|
return err
|
|
} else {
|
|
HttpAddr = AppConfig.String("HttpAddr")
|
|
|
|
if v, err := AppConfig.Int("HttpPort"); err == nil {
|
|
HttpPort = v
|
|
}
|
|
|
|
if maxmemory, err := AppConfig.Int64("MaxMemory"); err == nil {
|
|
MaxMemory = maxmemory
|
|
}
|
|
|
|
if appname := AppConfig.String("AppName"); appname != "" {
|
|
AppName = appname
|
|
}
|
|
|
|
if runmode := AppConfig.String("RunMode"); runmode != "" {
|
|
RunMode = runmode
|
|
}
|
|
|
|
if autorender, err := AppConfig.Bool("AutoRender"); err == nil {
|
|
AutoRender = autorender
|
|
}
|
|
|
|
if autorecover, err := AppConfig.Bool("RecoverPanic"); err == nil {
|
|
RecoverPanic = autorecover
|
|
}
|
|
|
|
if views := AppConfig.String("ViewsPath"); views != "" {
|
|
ViewsPath = views
|
|
}
|
|
|
|
if sessionon, err := AppConfig.Bool("SessionOn"); err == nil {
|
|
SessionOn = sessionon
|
|
}
|
|
|
|
if sessProvider := AppConfig.String("SessionProvider"); sessProvider != "" {
|
|
SessionProvider = sessProvider
|
|
}
|
|
|
|
if sessName := AppConfig.String("SessionName"); sessName != "" {
|
|
SessionName = sessName
|
|
}
|
|
|
|
if sesssavepath := AppConfig.String("SessionSavePath"); sesssavepath != "" {
|
|
SessionSavePath = sesssavepath
|
|
}
|
|
|
|
if sesshashfunc := AppConfig.String("SessionHashFunc"); sesshashfunc != "" {
|
|
SessionHashFunc = sesshashfunc
|
|
}
|
|
|
|
if sesshashkey := AppConfig.String("SessionHashKey"); sesshashkey != "" {
|
|
SessionHashKey = sesshashkey
|
|
}
|
|
|
|
if sessMaxLifeTime, err := AppConfig.Int("SessionGCMaxLifetime"); err == nil && sessMaxLifeTime != 0 {
|
|
int64val, _ := strconv.ParseInt(strconv.Itoa(sessMaxLifeTime), 10, 64)
|
|
SessionGCMaxLifetime = int64val
|
|
}
|
|
|
|
if sesscookielifetime, err := AppConfig.Int("SessionCookieLifeTime"); err == nil && sesscookielifetime != 0 {
|
|
SessionCookieLifeTime = sesscookielifetime
|
|
}
|
|
|
|
if usefcgi, err := AppConfig.Bool("UseFcgi"); err == nil {
|
|
UseFcgi = usefcgi
|
|
}
|
|
|
|
if enablegzip, err := AppConfig.Bool("EnableGzip"); err == nil {
|
|
EnableGzip = enablegzip
|
|
}
|
|
|
|
if directoryindex, err := AppConfig.Bool("DirectoryIndex"); err == nil {
|
|
DirectoryIndex = directoryindex
|
|
}
|
|
|
|
if hotupdate, err := AppConfig.Bool("HotUpdate"); err == nil {
|
|
EnableHotUpdate = hotupdate
|
|
}
|
|
|
|
if timeout, err := AppConfig.Int64("HttpServerTimeOut"); err == nil {
|
|
HttpServerTimeOut = timeout
|
|
}
|
|
|
|
if errorsshow, err := AppConfig.Bool("ErrorsShow"); err == nil {
|
|
ErrorsShow = errorsshow
|
|
}
|
|
|
|
if copyrequestbody, err := AppConfig.Bool("CopyRequestBody"); err == nil {
|
|
CopyRequestBody = copyrequestbody
|
|
}
|
|
|
|
if xsrfkey := AppConfig.String("XSRFKEY"); xsrfkey != "" {
|
|
XSRFKEY = xsrfkey
|
|
}
|
|
|
|
if enablexsrf, err := AppConfig.Bool("EnableXSRF"); err == nil {
|
|
EnableXSRF = enablexsrf
|
|
}
|
|
|
|
if expire, err := AppConfig.Int("XSRFExpire"); err == nil {
|
|
XSRFExpire = expire
|
|
}
|
|
|
|
if tplleft := AppConfig.String("TemplateLeft"); tplleft != "" {
|
|
TemplateLeft = tplleft
|
|
}
|
|
|
|
if tplright := AppConfig.String("TemplateRight"); tplright != "" {
|
|
TemplateRight = tplright
|
|
}
|
|
|
|
if httptls, err := AppConfig.Bool("HttpTLS"); err == nil {
|
|
HttpTLS = httptls
|
|
}
|
|
|
|
if certfile := AppConfig.String("HttpCertFile"); certfile != "" {
|
|
HttpCertFile = certfile
|
|
}
|
|
|
|
if keyfile := AppConfig.String("HttpKeyFile"); keyfile != "" {
|
|
HttpKeyFile = keyfile
|
|
}
|
|
|
|
if serverName := AppConfig.String("BeegoServerName"); serverName != "" {
|
|
BeegoServerName = serverName
|
|
}
|
|
|
|
if flashname := AppConfig.String("FlashName"); flashname != "" {
|
|
FlashName = flashname
|
|
}
|
|
|
|
if flashseperator := AppConfig.String("FlashSeperator"); flashseperator != "" {
|
|
FlashSeperator = flashseperator
|
|
}
|
|
|
|
if sd := AppConfig.String("StaticDir"); sd != "" {
|
|
for k := range StaticDir {
|
|
delete(StaticDir, k)
|
|
}
|
|
sds := strings.Fields(sd)
|
|
for _, v := range sds {
|
|
if url2fsmap := strings.SplitN(v, ":", 2); len(url2fsmap) == 2 {
|
|
StaticDir["/"+strings.TrimRight(url2fsmap[0], "/")] = url2fsmap[1]
|
|
} else {
|
|
StaticDir["/"+strings.TrimRight(url2fsmap[0], "/")] = url2fsmap[0]
|
|
}
|
|
}
|
|
}
|
|
|
|
if sgz := AppConfig.String("StaticExtensionsToGzip"); sgz != "" {
|
|
extensions := strings.Split(sgz, ",")
|
|
if len(extensions) > 0 {
|
|
StaticExtensionsToGzip = []string{}
|
|
for _, ext := range extensions {
|
|
if len(ext) == 0 {
|
|
continue
|
|
}
|
|
extWithDot := ext
|
|
if extWithDot[:1] != "." {
|
|
extWithDot = "." + extWithDot
|
|
}
|
|
StaticExtensionsToGzip = append(StaticExtensionsToGzip, extWithDot)
|
|
}
|
|
}
|
|
}
|
|
|
|
if enableadmin, err := AppConfig.Bool("EnableAdmin"); err == nil {
|
|
EnableAdmin = enableadmin
|
|
}
|
|
|
|
if adminhttpaddr := AppConfig.String("AdminHttpAddr"); adminhttpaddr != "" {
|
|
AdminHttpAddr = adminhttpaddr
|
|
}
|
|
|
|
if adminhttpport, err := AppConfig.Int("AdminHttpPort"); err == nil {
|
|
AdminHttpPort = adminhttpport
|
|
}
|
|
}
|
|
return nil
|
|
}
|