main.go

package main

import (
	"flag"
	"io/ioutil"
	"log"
	"os"
	"os/signal"

	_ "github.com/GoAdminGroup/go-admin/adapter/gin"              // web framework adapter
	_ "github.com/GoAdminGroup/go-admin/modules/db/drivers/mysql" // sql driver
	_ "github.com/GoAdminGroup/themes/adminlte"                   // ui theme
	_ "github.com/GoAdminGroup/themes/adminlte/separation"

	"github.com/GoAdminGroup/go-admin/engine"
	"github.com/GoAdminGroup/go-admin/template"
	"github.com/GoAdminGroup/go-admin/template/chartjs"
	"github.com/gin-gonic/gin"

	"AssetInfo/models"
	"AssetInfo/pages"
	"AssetInfo/tables"
)

func main() {
	startServer()
}

func startServer() {
	gin.SetMode(gin.ReleaseMode)
	gin.DefaultWriter = ioutil.Discard

	r := gin.Default()

	template.AddComp(chartjs.NewChart())

	eng := engine.Default()

	if err := eng.AddConfigFromYAML("./config.yml").
		AddGenerators(tables.Generators).
		Use(r); err != nil {
		panic(err)
	}

	r.Static("/uploads", "./uploads")

	eng.HTML("GET", "/admin", pages.GetDashBoard)
	//eng.HTMLFile(
	//	"GET",
	//	"/admin/hello",
	//	"./html/hello.tmpl",
	//	map[string]interface{}{
	//		"msg": "Hello world",
	//})
	//eng.Data(
	//	"GET",
	//	"/admin/data/:id",
	//	handlers.AutoCheck,
	//)

	models.Init(eng.MysqlConnection())

	var serve string
	flag.StringVar(&serve, "serve", ":80", "服务地址及端口")
	flag.Parse()
	_ = r.Run(serve)

	quit := make(chan os.Signal, 1)
	signal.Notify(quit, os.Interrupt)
	<-quit
	log.Print("closing database connection")
	eng.MysqlConnection().Close()
}