mirror of https://github.com/veypi/OneAuth.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
2.7 KiB
Go
136 lines
2.7 KiB
Go
package cfg
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/olivere/elastic/v7"
|
|
"github.com/olivere/elastic/v7/config"
|
|
"github.com/veypi/utils/cmd"
|
|
"github.com/veypi/utils/log"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
var Path = cmd.GetCfgPath("oa", "settings")
|
|
var Ctx, Cancel = context.WithCancel(context.Background())
|
|
|
|
var CFG = &struct {
|
|
AdminUser string
|
|
Host string
|
|
LoggerPath string
|
|
LoggerLevel string
|
|
APPUUID string
|
|
APPKey []byte
|
|
TimeFormat string
|
|
TimeZone string
|
|
Debug bool
|
|
FileUrlPrefix string
|
|
FireDir string
|
|
DB struct {
|
|
Type string
|
|
Addr string
|
|
User string
|
|
Pass string
|
|
DB string
|
|
}
|
|
ES *config.Config
|
|
}{
|
|
APPUUID: "jU5Jo5hM",
|
|
APPKey: []byte("cB43wF94MLTksyBK"),
|
|
AdminUser: "admin",
|
|
Host: "0.0.0.0:4001",
|
|
LoggerPath: "",
|
|
LoggerLevel: "debug",
|
|
TimeFormat: "2006/01/02 15:04:05",
|
|
TimeZone: "Asia/Shanghai",
|
|
Debug: true,
|
|
FileUrlPrefix: "/file",
|
|
FireDir: "/Users/light/test/media/",
|
|
DB: struct {
|
|
Type string
|
|
Addr string
|
|
User string
|
|
Pass string
|
|
DB string
|
|
}{
|
|
//Type: "sqlite",
|
|
Addr: "127.0.0.1:3306",
|
|
//Addr: "oa.db",
|
|
User: "root",
|
|
Pass: "123456",
|
|
DB: "one_auth",
|
|
},
|
|
ES: &config.Config{
|
|
URL: "http://127.0.0.1:9200",
|
|
Username: "elastic",
|
|
Password: "123456",
|
|
},
|
|
}
|
|
|
|
var (
|
|
db *gorm.DB
|
|
)
|
|
|
|
func DB() *gorm.DB {
|
|
if db == nil {
|
|
ConnectDB()
|
|
}
|
|
return db
|
|
}
|
|
|
|
var gormCfg = &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
SingularTable: false, // 使用单数表名，启用该选项后，`User` 表将是`user`
|
|
NoLowerCase: true,
|
|
},
|
|
}
|
|
|
|
func ConnectDB() *gorm.DB {
|
|
var err error
|
|
conn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=True&loc=Local", CFG.DB.User, CFG.DB.Pass, CFG.DB.Addr, CFG.DB.DB)
|
|
if CFG.DB.Type == "sqlite" {
|
|
conn = CFG.DB.Addr
|
|
db, err = gorm.Open(sqlite.Open(conn), gormCfg)
|
|
} else {
|
|
db, err = gorm.Open(mysql.Open(conn), gormCfg)
|
|
}
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return db
|
|
}
|
|
|
|
var es *elastic.Client
|
|
|
|
func ES() *elastic.Client {
|
|
if es == nil {
|
|
ConnectES()
|
|
}
|
|
return es
|
|
}
|
|
|
|
func ConnectES() *elastic.Client {
|
|
var err error
|
|
//es, err = elastic.NewClientFromConfig(CFG.ES)
|
|
es, err = elastic.NewClient(
|
|
elastic.SetURL(CFG.ES.URL),
|
|
elastic.SetBasicAuth(CFG.ES.Username, CFG.ES.Password),
|
|
elastic.SetSniff(false),
|
|
elastic.SetHealthcheck(false),
|
|
elastic.SetErrorLog(log.ConsoleLogger()),
|
|
)
|
|
if err != nil {
|
|
log.Warn().Msgf("connect es failed: %s", err)
|
|
return nil
|
|
}
|
|
_, _, err = es.Ping("http://127.0.0.1:9200").Do(context.Background())
|
|
if err != nil {
|
|
// Handle error
|
|
log.Warn().Msgf("connect es failed: %s", err)
|
|
}
|
|
return es
|
|
}
|