init-a_router-func.go 8.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. package routers
  2. import (
  3. controllers_func "boiler-go/controllers/func"
  4. "boiler-go/cronjobs"
  5. "boiler-go/locals"
  6. models_table "boiler-go/models/table"
  7. "bytes"
  8. "encoding/json"
  9. "errors"
  10. "io/ioutil"
  11. "runtime"
  12. "strconv"
  13. "strings"
  14. "github.com/dabory/abango-rest"
  15. e "github.com/dabory/abango-rest/etc"
  16. "github.com/labstack/echo"
  17. "github.com/labstack/echo/middleware"
  18. "gopkg.in/robfig/cron.v2"
  19. )
  20. type Route struct {
  21. Method []string
  22. Path string
  23. Handler echo.HandlerFunc
  24. }
  25. var routes []Route
  26. func AddRoute(route Route) {
  27. routes = append(routes, route)
  28. }
  29. func RestRouterInit(ask *abango.AbangoAsk) {
  30. //main.go 에서는 XConfig 값을 받아 올수 없으므로 여기서 세팅
  31. if abango.XConfig["IsCronJob"] == "Yes" {
  32. runtime.GOMAXPROCS(runtime.NumCPU()) //CPU Core 전체를 사용함.
  33. sched := cron.New()
  34. sched.AddFunc(abango.XConfig["CronJobInterval"], cronjobs.MainJob)
  35. // read manaul : https://pkg.go.dev/gopkg.in/robfig/cron.v2
  36. sched.Start()
  37. }
  38. c := echo.New()
  39. // c.Pre(middleware.BodyDump(func(c echo.Context, reqBody, resBody []byte) {
  40. c.Use(middleware.CORSWithConfig(middleware.CORSConfig{
  41. AllowOrigins: []string{"*"},
  42. // AllowOrigins: []string{"http://single.daboryhost.com", "http://fit-vac.com",
  43. AllowHeaders: []string{echo.HeaderOrigin, echo.HeaderContentType, echo.HeaderAccept},
  44. }))
  45. c.Pre(func(next echo.HandlerFunc) echo.HandlerFunc {
  46. return func(c echo.Context) error {
  47. r := c.Request()
  48. uri := r.URL.Path
  49. // var IsUpdateFieldListOn bool
  50. // actPostfix := uri[len(uri)-4:]
  51. // if actPostfix == "-ins" || actPostfix == "-upt" || actPostfix == "-del" { //update이외의 경우 속도증가
  52. // uri = uri[0:len(uri)-4] + "-act" // 이렇게 하면 -ins,-upt,-del 의 경우는 abg.UpdateFieldList 만드는 과정 스킵가능함.
  53. // } else if actPostfix == "-act" {
  54. // IsUpdateFieldListOn = true
  55. // }
  56. var l models_table.DbtLogAccess //변수재사용을 위하여, 실제 로그 기록은 소스 맨 아래에서 AddaRow함.
  57. l.CreatedOn = e.GetNowUnix()
  58. l.FrontIp = c.RealIP()
  59. l.Url = uri
  60. l.FrontHost = r.Header.Get("FrontendHost")
  61. l.RemoteIp = r.Header.Get("RemoteIp")
  62. l.Referer = r.Header.Get("Referer")
  63. e.OkLog(".")
  64. e.OkLog("Start >> " + l.Url + " FROM (" + l.FrontIp + ") " + l.FrontHost + " <- " + l.RemoteIp)
  65. bodyBytes, err := ioutil.ReadAll(r.Body)
  66. if err == nil { // Remove when production completes
  67. e.OkLog("Requested Json: " + string(bodyBytes))
  68. r.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes)) // This leads ReadCloser rewinded
  69. }
  70. // 미들웨어에서 receiver 받은 것을 echo.Context 로 넘겨줌.
  71. var abg abango.Controller
  72. // var err error
  73. if uri != "/gate-token-get" { // 미리 GateToken이 있는지 확인함.
  74. abg.GateToken = r.Header.Get("GateToken")
  75. if status, msg := abg.Init(); status != 200 {
  76. return c.String(status, msg)
  77. }
  78. if err := CheckAppPerm(&abg, uri); err != nil { //SsoSubId로 들어온 App 계정체크
  79. return c.String(508, "App Permission Denied: "+err.Error())
  80. }
  81. }
  82. if uri == "/gate-token-get" { // gate-token-test 는 아웃풋없는 test 이므로 여기 없음.
  83. var v controllers_func.GateTokenGetReq
  84. err = json.NewDecoder(r.Body).Decode(&v)
  85. c.Set("receiver", v)
  86. } else { //!!주의 긴 string이 먼저나오게 지정할 것
  87. if locals.HasPickActPage(uri, "dummydummy") {
  88. } else {
  89. return c.String(709, e.LogStr("ewgvdafewwa", "Request Function Not Found in Middleware for "+uri))
  90. }
  91. }
  92. if err != nil { // error 처리를 반복하지 않고 하나로 처리
  93. return c.String(800, e.JsonFormatErr("wevzxdfarfawe", uri)+" "+err.Error())
  94. }
  95. if uri[len(uri)-4:] == "-act" {
  96. var mapped map[string]interface{}
  97. // id 번호 구함.
  98. inxId := bytes.Index(bodyBytes, []byte("\"Id\""))
  99. inxIdStart := 1 + inxId + bytes.Index(bodyBytes[inxId:], []byte(":"))
  100. inxIdEnd := inxIdStart + bytes.Index(bodyBytes[inxIdStart:], []byte(","))
  101. if inxIdStart > inxIdEnd { //Delete 는 { "Id": -152 } 형태이므로 , 가 없다.
  102. inxIdEnd = inxIdStart + bytes.Index(bodyBytes[inxIdStart:], []byte("}"))
  103. }
  104. // fmt.Println("inxIdStart2:", inxIdStart)
  105. // fmt.Println("inxIdEnd2", inxIdEnd)
  106. idChar := string(bodyBytes[inxIdStart:inxIdEnd])
  107. if idChar[len(idChar)-1:] == "}" { //{"Id": -1541}, 다중 레코드의 경우도 처리
  108. idChar = idChar[:len(idChar)-1]
  109. }
  110. strId := strings.TrimSpace(idChar)
  111. intId, err := strconv.Atoi(strId)
  112. // fmt.Println("strId=", strId)
  113. if err != nil { // Id 가 정수가 아니면 치명적에러 이다. // ** 입력,수정,삭제를 복합적으로 한번에 request하는 것은 처리 안됨.
  114. return c.String(800, e.JsonFormatErr("0qjhoysaee", "Id value is not a integer "+err.Error()))
  115. }
  116. if intId > 0 { //id > 0 경우는 update인 경우만 처리한다.
  117. inxPage := bytes.Index(bodyBytes, []byte("\"Page\""))
  118. inxCoreStart := inxPage + bytes.Index(bodyBytes[inxPage:], []byte("{"))
  119. //Json 필드를 찾기 Skip 하기 위한 부분인데 만약 Json Field가 2개이상이면 For문으로 업그레이드 필요.
  120. // 현재는 setup-act 와 eyetest-act의 경우를 처리함.
  121. inxCoreJson := bytes.Index(bodyBytes, []byte("}\"")) //Json필드의 마지막은 이렇게 저장된다.
  122. inxCoreEnd := 0
  123. if inxCoreJson <= inxCoreStart { //Json 필드 없는 경우
  124. inxCoreEnd = 1 + bytes.Index(bodyBytes, []byte("}"))
  125. } else { //Json 필드 있는 경우
  126. inxCoreJson = 1 + inxCoreJson
  127. inxCoreEnd = 1 + inxCoreJson + bytes.Index(bodyBytes[inxCoreJson:], []byte("}"))
  128. }
  129. // fmt.Println("core:", string(bodyBytes[inxCoreStart:inxCoreEnd]))
  130. if err := json.Unmarshal(bodyBytes[inxCoreStart:inxCoreEnd], &mapped); err != nil {
  131. return c.String(800, e.JsonFormatErr("903uous09ur", uri))
  132. }
  133. update := ""
  134. for fld := range mapped {
  135. update += e.SnakeString(fld) + ","
  136. }
  137. abg.UpdateFieldList = update[0 : len(update)-1] //마지막 , 하니떼고
  138. // fmt.Println("UpdateFieldList:", abg.UpdateFieldList)
  139. }
  140. }
  141. // }
  142. c.Set("abango", abg) // 미들웨어에서 y.Db를 접속한 후 echo.Context 로 넘겨줌.
  143. //Log 기록 _setup반영요
  144. // if abg.GateToken != "" { //GateToken 이 있는 경우,유효 Request만 로그를 TargetDB에 쓸수가 있슴
  145. // l.MemberId = abg.Gtb.MemberId
  146. // l.UserId = abg.Gtb.UserId
  147. // if err := l.AddaRow(&abg); err != nil {
  148. // return c.String(603, err.Error())
  149. // }
  150. // }
  151. err1 := next(c)
  152. // fmt.Println("Pre 2번 미들웨어 종료")
  153. return err1
  154. }
  155. })
  156. c.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
  157. Format: "Finish >> method=${method}, uri=${uri}, status=${status} \n",
  158. }))
  159. for _, r := range routes {
  160. c.Match(r.Method, r.Path, r.Handler)
  161. }
  162. // c.Static("/.well-known/acme-challenge", "/home/.well-known/acme-challenge")
  163. xc := abango.XConfig
  164. if xc["SslMode"] == "Yes" {
  165. e.OkLog("SSL(HTTPS) Mode Started !!")
  166. //API Server 포트가 443: "https://api.dabory.com", 40443 포트: "https://api.dabory.com"
  167. c.Logger.Fatal(c.StartTLS(xc["SslConnect"], xc["SslFullChain"], xc["SslPrivate"]))
  168. } else {
  169. e.OkLog("PLAIN(HTTP) Mode Started !!")
  170. c.Logger.Fatal(c.Start(abango.XConfig["RestConnect"]))
  171. }
  172. }
  173. func CheckAppPerm(y *abango.Controller, uri string) error {
  174. if y.Gtb.SsoSubId == 0 {
  175. return nil
  176. }
  177. var qName string //이쿼리 queries 필드에거 가져와서 진행할 수 있도록 할 것.
  178. sql := `select
  179. mx.id
  180. from
  181. pro_member_app as mx
  182. inner join pro_app_perm as prm on mx.app_perm_id = prm.id
  183. inner join pro_app_perm_bd as bdy on prm.id = bdy.app_perm_id
  184. inner join pro_app_api as api on api.id = bdy.app_api_id
  185. where
  186. mx.sso_sub_id = ? and api.api_uri = ?`
  187. // qName = "kkk"
  188. //나중에 "QueryName 찾는 로직도 넣을 것
  189. if qName != "" {
  190. sql += " and api.query_name = '" + qName + "'"
  191. }
  192. // fmt.Println("y.Gtb.SsoSubId,", y.Gtb.SsoSubId)
  193. // fmt.Println("uri: ", uri)
  194. // fmt.Println("sql: ", sql)
  195. arr, err := y.Db.Query(sql, y.Gtb.SsoSubId, uri)
  196. if err != nil {
  197. return e.LogErr("3f0oijhbfre", "Query issue. ", err)
  198. }
  199. if len(arr) == 1 {
  200. return nil
  201. } else {
  202. return e.LogErr("34445ef34r", "App Api Record Issue ", errors.New(""))
  203. }
  204. }