-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
84 lines (69 loc) · 2.43 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package main
import (
"baseapp/controller"
"baseapp/tool"
"fmt"
_ "fmt"
"net/http"
"strings"
"github.com/wonderivan/logger"
"github.com/gin-gonic/gin"
)
func main() {
cfg, err := tool.ParseConfig("./config/app.json")
fmt.Println(cfg)
if err != nil {
panic(err.Error())
}
engine := gin.Default()
_, err = tool.OrmEngine(cfg)
if err != nil {
logger.Error(err.Error())
return
}
tool.InitRedisStore() //初始化redis配置
engine.Use(Cors())
//集成session
tool.InitSession(engine)
// engine.GET("/hello", func(context *gin.Context){
// fmt.Println("123")
// context.Writer.Write([]byte("helloooo\n"))
// })
registerRouter(engine)
engine.Run(":"+cfg.AppPort)
}
func registerRouter(router *gin.Engine){
new(controller.HelloController).Router(router)
new(controller.MemberController).Router(router)
}
func Cors() gin.HandlerFunc {
return func(context *gin.Context) {
method := context.Request.Method
origin := context.Request.Header.Get("Origin")
var headerKeys []string
for key, _ := range context.Request.Header {
headerKeys = append(headerKeys, key)
}
headerStr := strings.Join(headerKeys, ",")
if headerStr != "" {
headerStr = fmt.Sprintf("access-control-allow-origin, access-control-allow-headers, %s", headerStr)
} else {
headerStr = "access-control-allow-origin, access-control-allow-headers"
}
if origin != "" {
context.Writer.Header().Set("Access-Control-Allow-Origin", "*")
context.Header("Access-Control-Allow-Origin", "*") // 设置允许访问所有域
context.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE")
context.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Token,session,X_Requested_With,Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, Cache-Control, Content-Type, Pragma")
context.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified,Pragma,FooBar")
context.Header("Access-Control-Max-Age", "172800")
context.Header("Access-Control-Allow-Credentials", "false")
context.Set("content-type", "application/json") //// 设置返回格式是json
}
if method == "OPTIONS" {
context.JSON(http.StatusOK, "Options Request!")
}
//处理请求
context.Next()
}
}