mirror of
https://github.com/astaxie/beego.git
synced 2024-12-23 03:30:49 +00:00
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package web
|
|
|
|
import (
|
|
context2 "context"
|
|
"encoding/json"
|
|
"mime"
|
|
"net/http"
|
|
"path/filepath"
|
|
|
|
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
|
"github.com/astaxie/beego/pkg/infrastructure/session"
|
|
"github.com/astaxie/beego/pkg/server/web/context"
|
|
)
|
|
|
|
// register MIME type with content type
|
|
func registerMime() error {
|
|
for k, v := range mimemaps {
|
|
mime.AddExtensionType(k, v)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// register default error http handlers, 404,401,403,500 and 503.
|
|
func registerDefaultErrorHandler() error {
|
|
m := map[string]func(http.ResponseWriter, *http.Request){
|
|
"401": unauthorized,
|
|
"402": paymentRequired,
|
|
"403": forbidden,
|
|
"404": notFound,
|
|
"405": methodNotAllowed,
|
|
"500": internalServerError,
|
|
"501": notImplemented,
|
|
"502": badGateway,
|
|
"503": serviceUnavailable,
|
|
"504": gatewayTimeout,
|
|
"417": invalidxsrf,
|
|
"422": missingxsrf,
|
|
"413": payloadTooLarge,
|
|
}
|
|
for e, h := range m {
|
|
if _, ok := ErrorMaps[e]; !ok {
|
|
ErrorHandler(e, h)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerSession() error {
|
|
if BConfig.WebConfig.Session.SessionOn {
|
|
var err error
|
|
sessionConfig, err := AppConfig.String(nil, "sessionConfig")
|
|
conf := new(session.ManagerConfig)
|
|
if sessionConfig == "" || err != nil {
|
|
conf.CookieName = BConfig.WebConfig.Session.SessionName
|
|
conf.EnableSetCookie = BConfig.WebConfig.Session.SessionAutoSetCookie
|
|
conf.Gclifetime = BConfig.WebConfig.Session.SessionGCMaxLifetime
|
|
conf.Secure = BConfig.Listen.EnableHTTPS
|
|
conf.CookieLifeTime = BConfig.WebConfig.Session.SessionCookieLifeTime
|
|
conf.ProviderConfig = filepath.ToSlash(BConfig.WebConfig.Session.SessionProviderConfig)
|
|
conf.DisableHTTPOnly = BConfig.WebConfig.Session.SessionDisableHTTPOnly
|
|
conf.Domain = BConfig.WebConfig.Session.SessionDomain
|
|
conf.EnableSidInHTTPHeader = BConfig.WebConfig.Session.SessionEnableSidInHTTPHeader
|
|
conf.SessionNameInHTTPHeader = BConfig.WebConfig.Session.SessionNameInHTTPHeader
|
|
conf.EnableSidInURLQuery = BConfig.WebConfig.Session.SessionEnableSidInURLQuery
|
|
} else {
|
|
if err = json.Unmarshal([]byte(sessionConfig), conf); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if GlobalSessions, err = session.NewManager(BConfig.WebConfig.Session.SessionProvider, conf); err != nil {
|
|
return err
|
|
}
|
|
go GlobalSessions.GC()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerTemplate() error {
|
|
defer lockViewPaths()
|
|
if err := AddViewPath(BConfig.WebConfig.ViewsPath); err != nil {
|
|
if BConfig.RunMode == DEV {
|
|
logs.Warn(err)
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerGzip() error {
|
|
if BConfig.EnableGzip {
|
|
context.InitGzip(
|
|
AppConfig.DefaultInt(context2.Background(), "gzipMinLength", -1),
|
|
AppConfig.DefaultInt(context2.Background(), "gzipCompressLevel", -1),
|
|
AppConfig.DefaultStrings(context2.Background(), "includedMethods", []string{"GET"}),
|
|
)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func registerCommentRouter() error {
|
|
if BConfig.RunMode == DEV {
|
|
if err := parserPkg(filepath.Join(WorkPath, BConfig.WebConfig.CommentRouterPath)); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|