main.go 909 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package main
  2. import (
  3. "fmt"
  4. "sparrow/pkg/generator"
  5. "sparrow/pkg/server"
  6. "sparrow/services/knowoapi/model"
  7. "github.com/kataras/iris/v12"
  8. )
  9. func main() {
  10. err := server.Init("knowoapi")
  11. if err != nil {
  12. server.Log.Fatal(err)
  13. return
  14. }
  15. app := newApp()
  16. // register a http handler
  17. err = server.RegisterHTTPHandler(app)
  18. if err != nil {
  19. server.Log.Errorf("RegisterHTTPHandler Error: %s", err)
  20. return
  21. }
  22. // go
  23. err = server.Run()
  24. if err != nil {
  25. server.Log.Fatal(err)
  26. }
  27. }
  28. func newApp() *iris.Application {
  29. app := iris.New()
  30. iris.WithoutBodyConsumptionOnUnmarshal(app)
  31. app.AllowMethods(iris.MethodOptions)
  32. registerErrors(app)
  33. db, err := getDB()
  34. if err != nil {
  35. fmt.Print(err)
  36. }
  37. models := new(model.All).Init(db)
  38. gen, err := generator.NewKeyGenerator(*confAESKey)
  39. if err != nil {
  40. fmt.Print(err)
  41. }
  42. //注册路由
  43. registerRouters(app, models, gen)
  44. app.Build()
  45. return app
  46. }