|
@@ -1,6 +1,8 @@
|
|
|
package router
|
|
|
|
|
|
import (
|
|
|
+ "net/http"
|
|
|
+
|
|
|
"github.com/gin-contrib/sessions"
|
|
|
"github.com/gin-contrib/sessions/memstore"
|
|
|
"github.com/gin-gonic/gin"
|
|
@@ -11,5 +13,19 @@ var Router = gin.Default()
|
|
|
func init() {
|
|
|
store := memstore.NewStore([]byte("jy_data_extract"))
|
|
|
Router.Use(sessions.Sessions("mysession", store))
|
|
|
- Router.Run(":9090")
|
|
|
+}
|
|
|
+
|
|
|
+func AuthMiddleWare() gin.HandlerFunc {
|
|
|
+ return func(c *gin.Context) {
|
|
|
+ session := sessions.Default(c)
|
|
|
+ if session.Get("name") != nil {
|
|
|
+ c.Next()
|
|
|
+ return
|
|
|
+ }
|
|
|
+ c.JSON(http.StatusUnauthorized, gin.H{
|
|
|
+ "error": "Unauthorized",
|
|
|
+ })
|
|
|
+ c.Abort()
|
|
|
+ return
|
|
|
+ }
|
|
|
}
|