2023-02-22 16:22:50 +04:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
2023-08-02 15:54:14 +04:00
|
|
|
"fmt"
|
2023-03-27 22:05:17 +04:00
|
|
|
"log"
|
2023-02-22 16:22:50 +04:00
|
|
|
"math/rand"
|
2023-03-27 22:05:17 +04:00
|
|
|
"os"
|
2023-08-02 15:54:14 +04:00
|
|
|
"time"
|
2023-02-22 16:22:50 +04:00
|
|
|
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
|
|
"xorm.io/xorm"
|
2023-03-27 22:05:17 +04:00
|
|
|
xlog "xorm.io/xorm/log"
|
2023-02-22 16:22:50 +04:00
|
|
|
"xorm.io/xorm/names"
|
|
|
|
)
|
|
|
|
|
2023-08-02 22:22:36 +04:00
|
|
|
type DB struct {
|
|
|
|
User string
|
|
|
|
Pass string
|
|
|
|
Schema string
|
|
|
|
}
|
|
|
|
|
|
|
|
func Connect(db DB) (*xorm.Engine, error) {
|
|
|
|
engine, err := xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(localhost:3306)/%s?charset=utf8", db.User, db.Pass, db.Schema))
|
2023-02-22 16:22:50 +04:00
|
|
|
if err != nil {
|
2023-03-07 13:38:50 +04:00
|
|
|
return nil, err
|
2023-02-22 16:22:50 +04:00
|
|
|
}
|
2023-03-27 22:05:17 +04:00
|
|
|
sqlLogger := xlog.NewSimpleLogger(CreateLog("sql"))
|
|
|
|
engine.SetLogger(sqlLogger)
|
2023-02-22 16:22:50 +04:00
|
|
|
engine.ShowSQL(true)
|
|
|
|
engine.SetMapper(names.SameMapper{})
|
|
|
|
|
2023-03-08 14:03:27 +04:00
|
|
|
err = engine.Sync(&User{}, &TgUser{}, &Group{}, &Lesson{}, &Teacher{}, &ShedulesInUser{})
|
2023-02-22 16:22:50 +04:00
|
|
|
if err != nil {
|
2023-03-07 13:38:50 +04:00
|
|
|
return nil, err
|
2023-02-22 16:22:50 +04:00
|
|
|
}
|
2023-03-07 13:38:50 +04:00
|
|
|
return engine, nil
|
2023-02-22 16:22:50 +04:00
|
|
|
}
|
|
|
|
|
2023-03-07 13:38:50 +04:00
|
|
|
func GenerateID(engine *xorm.Engine) (int64, error) {
|
2023-02-22 16:22:50 +04:00
|
|
|
id := rand.Int63n(899999999) + 100000000
|
|
|
|
|
|
|
|
exists, err := engine.ID(id).Exist(&User{})
|
|
|
|
if err != nil {
|
2023-03-07 13:38:50 +04:00
|
|
|
return 0, err
|
2023-02-22 16:22:50 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if exists {
|
|
|
|
return GenerateID(engine)
|
|
|
|
} else {
|
2023-03-07 13:38:50 +04:00
|
|
|
return id, nil
|
2023-02-22 16:22:50 +04:00
|
|
|
}
|
|
|
|
}
|
2023-03-27 22:05:17 +04:00
|
|
|
|
|
|
|
func CreateLog(name string) *os.File {
|
|
|
|
if _, err := os.Stat("logs"); os.IsNotExist(err) {
|
|
|
|
err = os.Mkdir("logs", os.ModePerm)
|
|
|
|
if err != nil {
|
2023-08-02 15:54:14 +04:00
|
|
|
log.Fatal("failed create log folder")
|
2023-03-27 22:05:17 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
fileName := "./logs/" + name + ".log"
|
2023-08-02 15:54:14 +04:00
|
|
|
if _, err := os.Stat(fileName); !os.IsNotExist(err) {
|
|
|
|
new := fmt.Sprintf("./logs/%s.before.%s.log", name, time.Now().Format("06-02-01-15-04-05"))
|
|
|
|
err := os.Rename(fileName, new)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("failed rename %s file to %s", fileName, new)
|
|
|
|
}
|
|
|
|
}
|
2023-03-27 22:05:17 +04:00
|
|
|
logFile, err := os.Create(fileName)
|
|
|
|
if err != nil {
|
2023-08-02 15:54:14 +04:00
|
|
|
log.Fatalf("failed open %s.log file", name)
|
2023-03-27 22:05:17 +04:00
|
|
|
}
|
2023-08-02 15:54:14 +04:00
|
|
|
defer logFile.Close()
|
2023-03-27 22:05:17 +04:00
|
|
|
return logFile
|
|
|
|
}
|