udeserver.go 936 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. // Code scaffolded by goctl. Safe to edit.
  2. // goctl 1.10.2
  3. package main
  4. import (
  5. "flag"
  6. "fmt"
  7. "udeServer/internal/apierr"
  8. "udeServer/internal/config"
  9. "udeServer/internal/handler"
  10. "udeServer/internal/svc"
  11. "github.com/zeromicro/go-zero/core/conf"
  12. "github.com/zeromicro/go-zero/rest"
  13. "github.com/zeromicro/go-zero/rest/httpx"
  14. )
  15. var configFile = flag.String("f", "etc/udeserver-api.yaml", "the config file")
  16. func main() {
  17. flag.Parse()
  18. var c config.Config
  19. conf.MustLoad(*configFile, &c)
  20. httpx.SetErrorHandler(func(err error) (int, any) {
  21. status := apierr.Status(err)
  22. return status, map[string]any{
  23. "code": status,
  24. "msg": err.Error(),
  25. }
  26. })
  27. // rest.WithCors() 解决跨域
  28. server := rest.MustNewServer(c.RestConf, rest.WithCors())
  29. defer server.Stop()
  30. ctx := svc.NewServiceContext(c)
  31. handler.RegisterHandlers(server, ctx)
  32. fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
  33. server.Start()
  34. }