package main

import (
	"flag"

	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/viper"
	"myschools.me/wyh/gindownloadfile.git/handler"
)

func main() {
	cf := flag.String("config", "config.yaml", "file of config")
	flag.Parse()
	viper.SetConfigFile(*cf)
	if err := viper.ReadInConfig(); err != nil {
		log.WithFields(log.Fields{
			"func": "main",
		}).Warnf("%s", err)
	}

	r := gin.Default()
	r.GET(`/downloadfile`, handler.DownLoadFile)

	r.Run(viper.GetString("app.host"))
}