32 lines
442 B
Go
32 lines
442 B
Go
package init_db_data
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
type InitDBFunc interface {
|
|
Init() (err error)
|
|
}
|
|
|
|
// 初始化数据库
|
|
func initDB(InitDBFunctions ...InitDBFunc) (err error) {
|
|
for _, v := range InitDBFunctions {
|
|
err = v.Init()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func InitDbData() {
|
|
err := initDB(initPeakValleyTimeBlockData)
|
|
if err != nil {
|
|
fmt.Println("[-]初始化基础数据失败:", err)
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
|
|
//
|