bxbuyer.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. package main
  2. import (
  3. MC "app.yhyue.com/moapp/jybase/common"
  4. "app.yhyue.com/moapp/jybase/endless"
  5. "bp.jydev.jianyu360.cn/BaseService/gateway/core/node"
  6. "fmt"
  7. IC "jyBXBuyer/api/init"
  8. "log"
  9. "net/http"
  10. "os"
  11. "os/signal"
  12. "syscall"
  13. "jyBXBuyer/api/internal/handler"
  14. "jyBXBuyer/api/internal/svc"
  15. "github.com/zeromicro/go-zero/rest"
  16. )
  17. func main() {
  18. //注册代理服务
  19. closeNotify, err := node.NewNode(IC.C.Gateway.Etcd...).Register(IC.C.Gateway.ServerCode, MC.InterfaceToStr(IC.C.Port))
  20. if err != nil {
  21. panic(err)
  22. }
  23. //
  24. go func() {
  25. err := endless.ListenAndServe(":"+MC.InterfaceToStr(IC.C.Webrpcport), nil, func() {})
  26. if err != nil {
  27. log.Println("ListenAndServe: ", err)
  28. }
  29. }()
  30. ctx := svc.NewServiceContext(IC.C)
  31. server := rest.MustNewServer(IC.C.RestConf)
  32. defer server.Stop()
  33. //全局中间件
  34. server.Use(func(next http.HandlerFunc) http.HandlerFunc {
  35. return func(w http.ResponseWriter, r *http.Request) {
  36. defer func() {
  37. MC.AddMgoLogs(IC.MgoLog, r, IC.C.MgoLogsName, IC.ExcludeUrl, IC.C.MgoLogsCount)
  38. }()
  39. next.ServeHTTP(w, r)
  40. }
  41. })
  42. handler.RegisterHandlers(server, ctx)
  43. fmt.Printf("Starting server at %s:%d...\n", IC.C.Host, IC.C.Port)
  44. server.Start()
  45. quit := make(chan os.Signal, 1)
  46. signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
  47. <-quit
  48. closeNotify()
  49. }