139 lines
4.0 KiB
Go
139 lines
4.0 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"flag"
|
||
"fmt"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"path/filepath"
|
||
|
||
"res/dbop"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
// "github.com/robfig/cron/v3"
|
||
"github.com/shockliu/logger"
|
||
)
|
||
|
||
const uploadhtml = `<!DOCTYPE html>
|
||
<html lang="en">
|
||
<head>
|
||
<meta charset="UTF-8">
|
||
<title>upload file</title>
|
||
</head>
|
||
<body>
|
||
<form method="post" enctype="multipart/form-data">
|
||
<input type="file" name="file" multiple="multiple"/><br/>
|
||
<input type="submit" value="提交"/>
|
||
</form>
|
||
</body>`
|
||
|
||
/*
|
||
const uploadhtml = `<html>
|
||
|
||
<head></head>
|
||
<body>
|
||
<form method="post" enctype="multipart/form-data">
|
||
<input type="file" name="file" />
|
||
<input type="submit" />
|
||
</form>
|
||
</body>
|
||
|
||
</html>`
|
||
*/
|
||
func Mkdir(basePath string) string {
|
||
// 1.获取当前时间,并且格式化时间
|
||
folderName := time.Now().Format("2006-01/")
|
||
folderPath := filepath.Join(basePath, folderName)
|
||
//使用mkdirall会创建多层级目录
|
||
os.MkdirAll(folderPath, os.ModePerm)
|
||
return folderPath
|
||
}
|
||
|
||
func main() {
|
||
version := "1.0.0"
|
||
fmt.Printf("\n\n\n")
|
||
fmt.Printf(" **************************************************\n")
|
||
fmt.Printf(" * \033[1;34;42m豆曲咖资源|美团服务\033[0m *\n")
|
||
fmt.Printf(" * ========================================= *\n")
|
||
fmt.Printf(" * 版本: %v *\n", version)
|
||
fmt.Printf(" * 技术支持:\033[1;32;40m豆曲咖莱(杭州)科技有限公司\033[0m *\n")
|
||
fmt.Printf(" * Email: \033[1;32;40msales@daymele.com\033[0m *\n")
|
||
fmt.Printf(" **************************************************\n")
|
||
fmt.Printf("\n\n")
|
||
port := flag.Int("port", 9000, "端口")
|
||
conf := flag.String("conf", "/conf/daymele.conf", "配置文件")
|
||
flag.Parse()
|
||
logger.Debugf("端口参数%d\n", *port)
|
||
dbop.DBinit(*conf)
|
||
gin.SetMode(gin.ReleaseMode)
|
||
r := gin.Default()
|
||
gImg := r.Group("images")
|
||
gMp4 := r.Group("mp4")
|
||
gApk := r.Group("apk")
|
||
gQr := r.Group("QrCode")
|
||
gFile := r.Group("file")
|
||
|
||
gImg.StaticFS("/", http.Dir("/data/images"))
|
||
gMp4.StaticFS("/", http.Dir("/data/mp4"))
|
||
gApk.StaticFS("/", http.Dir("/data/apk"))
|
||
gQr.StaticFS("/", http.Dir("/data/QrCode"))
|
||
gFile.StaticFS("/", http.Dir("/data/file"))
|
||
//r.POST("/savefile",file.FileUpload)
|
||
// r.POST("/savefile", file.FileUpload)
|
||
// r.POST("/existfile", file.ExistFile)
|
||
r.GET("/savefile", func(c *gin.Context) {
|
||
c.Header("Content-Type", "text/html; charset=utf-8")
|
||
c.String(200, uploadhtml)
|
||
})
|
||
|
||
// 增加定时任务，晚上17点锁定馆所用户余额
|
||
// c := cron.New()
|
||
// c.AddFunc("0 17 * * *", func() {
|
||
// logger.Infof("锁定馆所员工余额\n")
|
||
// _, err := dbop.MDb.Exec("update dqk_user a,dqk_enterprise_user b set a.activity=JSON_SET(activity, '$.lock',true) where a.mobile=b.mobile and b.enterprise_id='100001' and a.funds>0;")
|
||
// if err != nil {
|
||
// logger.Errorf("数据库操作失败%s\n", err)
|
||
// }
|
||
// })
|
||
// c.Start()
|
||
|
||
// 美团接口
|
||
// g := r.Group("/client/trade")
|
||
// {
|
||
// g.POST("/queryBudget", meituan.QueryBudget)
|
||
// g.POST("/queryPayBudget", meituan.QueryPayBudget)
|
||
// g.POST("/payBudget", meituan.PayBudget)
|
||
// g.POST("/refundBudget", meituan.RefundBudget)
|
||
// g.POST("/queryRefundBudget", meituan.QueryRefundBudget)
|
||
// }
|
||
// logger.Debugf("7011\n")
|
||
// r.Run(":7011") // listen and serve on 0.0.0.0:8080
|
||
// 使用http启动服务
|
||
srv := &http.Server{
|
||
Addr: fmt.Sprintf(":%d", *port),
|
||
Handler: r,
|
||
}
|
||
go func() {
|
||
// 服务连接
|
||
fmt.Printf("豆曲咖[资源/美团]服务端口:%d\n", *port)
|
||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
logger.Fatalf("listen: %s\n", err)
|
||
}
|
||
}()
|
||
// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
|
||
quit := make(chan os.Signal)
|
||
signal.Notify(quit, os.Interrupt)
|
||
<-quit
|
||
logger.Info("Shutdown Server ...")
|
||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
defer cancel()
|
||
if err := srv.Shutdown(ctx); err != nil {
|
||
logger.Fatal("Server Shutdown:", err)
|
||
}
|
||
logger.Info("Server exiting")
|
||
|
||
}
|