init

main.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net/http"
+	"os"
+	"tinyurl/pkg/config"
+	"tinyurl/pkg/db"
+	"tinyurl/pkg/log"
+	"tinyurl/route"
+
+	"github.com/flamego/flamego"
+	"github.com/ory/graceful"
+	// "github.com/justinas/alice"
+)
+
+// go build -ldflags "-X main.Version=x.y.z"
+var (
+	// Name is the name of the compiled software.
+	Name string
+	// Version is the version of the compiled software.
+	Version string
+	// flagconf is the config flag.
+	flagconf string
+
+	id, _ = os.Hostname()
+)
+
+func init() {
+	flag.StringVar(&flagconf, "conf", "./configs/config.yml", "config path, eg: -conf config.yaml")
+}
+
+func main() {
+	cfg, err := config.Parse(flagconf)
+	if err != nil {
+		panic(err)
+	}
+	logger, err := log.New(cfg.Log)
+	if err != nil {
+		panic(err)
+	}
+	f := flamego.Classic()
+	f.Map(cfg, logger, db.DB)
+	// f.Use(auth.Basic("admin", "1111"))
+	route.Route(f)
+	server := graceful.WithDefaults(
+		&http.Server{
+			Addr:    "0.0.0.0:2830",
+			Handler: f,
+		},
+	)
+
+	if err := graceful.Graceful(server.ListenAndServe, server.Shutdown); err != nil {
+		fmt.Println("main: Failed to gracefully shutdown")
+	}
+	fmt.Println("main: Server was shutdown gracefully")
+}