|
|
package main
|
|
|
|
|
|
import (
|
|
|
"WechatGateWay/api"
|
|
|
"WechatGateWay/global"
|
|
|
"WechatGateWay/handle"
|
|
|
"WechatGateWay/third_part"
|
|
|
"fmt"
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
|
"github.com/jinzhu/gorm"
|
|
|
"github.com/robfig/cron"
|
|
|
"github.com/spf13/viper"
|
|
|
"log"
|
|
|
"math/rand"
|
|
|
"net/http"
|
|
|
"net/url"
|
|
|
"os"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
func InitDB() *gorm.DB {
|
|
|
driverName := "mysql"
|
|
|
host := viper.GetString("db.MySQLHOST")
|
|
|
port := viper.GetString("db.MySQLPORT")
|
|
|
database := viper.GetString("db.MySQLDB")
|
|
|
username := viper.GetString("db.MySQLUSER")
|
|
|
password := viper.GetString("db.MySQLPASSWORD")
|
|
|
charset := viper.GetString("db.MySQLCharset")
|
|
|
loc := viper.GetString("db.MySQLLoc")
|
|
|
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&parseTime=true&loc=%s",
|
|
|
username,
|
|
|
password,
|
|
|
host,
|
|
|
port,
|
|
|
database,
|
|
|
charset,
|
|
|
url.QueryEscape(loc))
|
|
|
|
|
|
//dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
|
|
|
db, err := gorm.Open(driverName, dsn)
|
|
|
//db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
|
|
|
|
if err == nil {
|
|
|
log.Println("Connected to database")
|
|
|
|
|
|
//db.DB()实现一个连接池
|
|
|
//设置最大连接数和最大闲置数
|
|
|
//默认0表示不限制
|
|
|
db.DB().SetMaxIdleConns(0)
|
|
|
db.DB().SetMaxOpenConns(0)
|
|
|
|
|
|
//给默认表名加前缀
|
|
|
gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
|
|
|
//return "admin_" + defaultTableName + "_test_env"
|
|
|
return "" + defaultTableName + ""
|
|
|
}
|
|
|
|
|
|
// 全局禁用表名复数
|
|
|
// 如果设置为true,`User`的默认表名为`user`,使用`TableName`设置的表名不受影响
|
|
|
// 如果设置为false,`User`的默认表名为`users`
|
|
|
db.SingularTable(false)
|
|
|
|
|
|
//打开sql日志
|
|
|
db.LogMode(viper.GetBool("db.GORMLog"))
|
|
|
|
|
|
//创建表时，添加后缀
|
|
|
db.Set("grom:table_options", "ENGINE=InnoDB")
|
|
|
|
|
|
//自动建表
|
|
|
db.AutoMigrate(&handle.WechatLog{})
|
|
|
db.AutoMigrate(&handle.WechatBlockWords{})
|
|
|
db.AutoMigrate(&api.BinancePrice{})
|
|
|
|
|
|
global.DB = db
|
|
|
return db
|
|
|
|
|
|
} else {
|
|
|
log.Println("Connected to database failed")
|
|
|
//log.Println("failed to connect database, err: " + err.Error())
|
|
|
panic("failed to connect database, err: " + err.Error())
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
func init() {
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
// 读取配置文件
|
|
|
workDir, _ := os.Getwd()
|
|
|
viper.SetConfigName("application")
|
|
|
viper.SetConfigType("yml")
|
|
|
viper.AddConfigPath(workDir)
|
|
|
err := viper.ReadInConfig()
|
|
|
if err != nil {
|
|
|
fmt.Println("config file not found")
|
|
|
os.Exit(1)
|
|
|
}
|
|
|
global.WechatCorpId = viper.GetString("wechat.CorpId")
|
|
|
global.WechatToken = viper.GetString("wechat.AppToken")
|
|
|
global.WechatEncodingAesKey = viper.GetString("wechat.EncodingAesKey")
|
|
|
global.WechatSendSecret = viper.GetString("wechat.SendSecret")
|
|
|
global.WechatSendAid = viper.GetString("wechat.SendAid")
|
|
|
global.WechatImagePath = viper.GetString("wechat.ImagePath")
|
|
|
// receive_id 企业应用的回调，表示corpid
|
|
|
global.WxCrypt = global.NewWXBizMsgCrypt(global.WechatToken, global.WechatEncodingAesKey, global.WechatCorpId, global.XmlType)
|
|
|
third_part.GetRemoteToken()
|
|
|
//初始化数据库
|
|
|
global.DB = InitDB()
|
|
|
|
|
|
log.Println("数据库初始化成功")
|
|
|
|
|
|
log.Println("server init success")
|
|
|
}
|
|
|
func main() {
|
|
|
c := cron.New()
|
|
|
//每30分钟执行一次
|
|
|
c.AddFunc("0 */30 * * * *", func() { go third_part.GetRemoteToken() })
|
|
|
c.Start()
|
|
|
log.Println("任务注册成功")
|
|
|
|
|
|
// 开启一个http服务器，接收来自企业微信的消息
|
|
|
http.HandleFunc("/", handle.HandleTencent)
|
|
|
port := viper.GetString("server.port")
|
|
|
if port == "" {
|
|
|
port = "8080"
|
|
|
}
|
|
|
log.Println("server start at port:", port)
|
|
|
log.Fatalln(http.ListenAndServe(":"+port, nil))
|
|
|
defer global.DB.Close()
|
|
|
}
|
