main.go 864 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  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. app.AllowMethods(iris.MethodOptions)
  31. registerErrors(app)
  32. db, err := getDB()
  33. if err != nil {
  34. fmt.Print(err)
  35. }
  36. models := new(model.All).Init(db)
  37. gen, err := generator.NewKeyGenerator(*confAESKey)
  38. if err != nil {
  39. fmt.Print(err)
  40. }
  41. //注册路由
  42. registerRouters(app, models, gen)
  43. app.Build()
  44. return app
  45. }