1. // @APIVersion 1.0.0
    2. // @Title 认证中心、用户中心
    3. // @Description token、用户、角色、权限
    4. package routers
    6. import (
    7.     "fmt"
    8.     "strings"
    9.     "zhsq_go/code"
    10.     "zhsq_go/user_center/controllers"
    11.     "zhsq_go/user_center/models"
    13.     "github.com/astaxie/beego"
    14.     "github.com/astaxie/beego/context"
    15.     "github.com/astaxie/beego/httplib"
    16. )
    18. func init() {
    19.     // 所有接口访问之前的过滤器
    20.     beego.InsertFilter("/*", beego.BeforeRouter, func(ctx *context.Context) {
    21.         names := beego.AppConfig.Strings("proxy::names")
    22.         for i := range names {
    23.             //如果是静态资源，直接放行
    24.             url := beego.AppConfig.String(fmt.Sprintf("proxy::%s_url", names[i]))
    25.             if strings.HasPrefix(ctx.Request.RequestURI, url) {
    26.                 if strings.Contains(ctx.Request.RequestURI, "swagger") || strings.Contains(ctx.Request.RequestURI, "api-docs") {
    27.                     host := beego.AppConfig.String(fmt.Sprintf("proxy::%s_host", names[i]))
    28.                     request := httplib.Get(fmt.Sprintf("%s%v", host, ctx.Request.RequestURI))
    29.                     con, _ := request.Bytes()
    30.                     if strings.HasSuffix(ctx.Request.RequestURI, ".css") {
    31.                         ctx.Output.Header("Content-Type", "text/css")
    32.                     }
    33.                     ctx.Output.Body(con)
    34.                     return
    35.                 }
    36.             }
    37.         }
    39.         // 校验Token是否有效
    40.         result, userId := models.FilterValidateToken(ctx)
    41.         if !result { //Token无效
    42.             message := code.ApiResult{Code: code.UNAUTHORIZED, Message: "Token 无效！"}
    43.             ctx.Output.JSON(message, false, false)
    44.             return
    45.         } else {
    46.             //判断用户的请求
    47.             if strings.HasPrefix(ctx.Request.RequestURI, "/api/user-center") { //用户中心
    48.                 // fmt.Println("userId :", userId)
    49.                 ctx.Input.SetData("userId", userId)
    50.             } else {
    51.                 for i := range names {
    52.                     //实现代理
    53.                     url := beego.AppConfig.String(fmt.Sprintf("proxy::%s_url", names[i]))
    54.                     if strings.HasPrefix(ctx.Request.RequestURI, url) {
    55.                         // fmt.Println("其他模块接口")
    56.                         host := beego.AppConfig.String(fmt.Sprintf("proxy::%s_host", names[i]))
    57.                         // 请求方式
    58.                         method := ctx.Request.Method
    59.                         // 请求路径
    60.                         url := ctx.Request.RequestURI
    62.                         var req *httplib.BeegoHTTPRequest
    63.                         uri := fmt.Sprintf("%s%v", host, url)
    64.                         // fmt.Println("请求地址：", uri, " 请求方式：", method)
    65.                         if method == "POST" {
    66.                             req = httplib.Post(uri)
    67.                             // Post 请求参数
    68.                             req.Body(ctx.Input.RequestBody)
    69.                         } else if method == "GET" {
    70.                             req = httplib.Get(uri)
    71.                         }
    73.                         req.Param("userId", userId)
    74.                         req.Header("Content-Type", "application/json")
    75.                         response, _ := req.Response()
    76.                         // fmt.Println("response.Status :", response.Status)
    77.                         if strings.Contains(response.Status, "200") {
    78.                             con, _ := req.String()
    79.                             ctx.WriteString(con)
    80.                         } else {
    81.                             ctx.Output.JSON(code.ApiResult{Code: code.FAILED, Message: "操作失败！"}, false, false)
    82.                         }
    83.                     }
    84.                 }
    85.             }
    86.         }
    87.     })
    89.     ns := beego.NewNamespace("/api/user-center",
    90.         beego.NSNamespace("/token",
    91.             beego.NSInclude(
    92.                 &controllers.TokenController{},
    93.             ),
    94.         ),
    95.         beego.NSNamespace("/user",
    96.             beego.NSInclude(
    97.                 &controllers.UserController{},
    98.             ),
    99.         ),
    100.     )
    101.     beego.AddNamespace(ns)
    103. }
    1. appname = user_center
    2. httpaddr = 192.168.2.83
    3. httpport = 80
    4. runmode = dev
    5. autorender = false
    6. copyrequestbody = true
    7. EnableDocs = true
    9. [ignored]
    10. url=/token/authorize;/token/verification;/api/property/company/list
    12. [proxy]
    13. names=profit;property
    14. profit_host=http://127.0.0.1:8081
    15. profit_url=/api/profit
    16. profit_language=golang
    17. property_host=http://127.0.0.1:8090
    18. property_url=/api/property
    19. property_language=java