Refactor setting module and add tests (#4)

This commit is contained in:
2021-11-29 18:16:27 +01:00
committed by GitHub
parent 2571e22843
commit 8b5cccd744
3 changed files with 250 additions and 44 deletions

View File

@ -3,6 +3,8 @@ package main
import (
"context"
"errors"
"flag"
"fmt"
"log"
"net/http"
"os"
@ -24,7 +26,15 @@ var (
)
func main() {
setting.Setup()
o, err := setting.Setup(os.Args[1:])
if err == flag.ErrHelp || err == setting.ErrVersion {
fmt.Print(o)
os.Exit(0)
} else if err != nil {
fmt.Print(err)
os.Exit(1)
}
models.Setup(setting.App.GeodbPath.City, setting.App.GeodbPath.ASN)
setupEngine()
router.SetupTemplate(engine)