package db import ( "scd_check_tools/conf" "scd_check_tools/global" "scd_check_tools/models/bo" "scd_check_tools/tools" "database/sql" "encoding/json" "fmt" "io/ioutil" "os" "strings" "github.com/astaxie/beego/orm" _ "github.com/mattn/go-sqlite3" ) func ConnSqlite(cnfpath string) (*sql.DB, error) { fmt.Println(tools.NowTime()+" 加载数据库配置文件", cnfpath) fileHanlder, err := os.Open(cnfpath) if err != nil { fmt.Printf(tools.NowTime()+" 配置文件%s加载失败:%s\r", cnfpath, err) os.Exit(1) } txt, _ := ioutil.ReadAll(fileHanlder) fileHanlder.Close() txtStr := string(txt) if txtStr == "" { fmt.Println(tools.NowTime() + " 配置文件格式不正确,请正确配置") os.Exit(1) } cfgdata := global.SqliteConfig{} err = json.Unmarshal(txt, &cfgdata) if err != nil { fmt.Println(tools.NowTime()+" 配置文件解析失败:", err) os.Exit(1) } url := cfgdata.Path conn, err := sql.Open("sqlite3", url) if err != nil { return conn, err } fmt.Println(tools.NowTime() + " Sqlite3数据库" + url + "连接成功") global.SqliteConn = conn orm.RegisterDriver("sqlite", orm.DRSqlite) orm.RegisterDataBase("default", "sqlite3", url, 30) if conf.GlobalConfig["loglevel"] == "4" { orm.Debug = true } //初始化数据库 sql := "select count(1) cnt from sqlite_master where type = 'table'" o := orm.NewOrm() datas := []orm.Params{} _, dberr := o.Raw(sql).Values(&datas) if dberr == nil { iscreatedb := "0" for _, row := range datas { iscreatedb = row["cnt"].(string) } if iscreatedb == "0" { fileHanlder, err = os.Open("db/db_init.sql") if err != nil { tools.Error(" 数据库初始化脚本文件db/db_init.sql加载失败:" + err.Error()) return nil, err } txt, _ = ioutil.ReadAll(fileHanlder) fileHanlder.Close() createdb_sql := string(txt) sqllist := strings.Split(createdb_sql, ";") for _, s := range sqllist { ts := strings.Trim(s, " ") if ts == "" || strings.HasPrefix(ts, "--") { continue } //tools.Log(ts) _, dberr = o.Raw(ts).Exec() tools.CheckErr(dberr) } //初始化超级管理员 //bo.SaveSysParam(bo.Global_sys_param{Param_name: "admin_account", Param_value: "admin", Param_memo: "超级管理员"}) //bo.SaveSysParam(bo.Global_sys_param{Param_name: "admin_pwd", Param_value: "{crypto}04444C66AD91C86262BE738C2F4848DE39B13327CDB8AA2E58CA3F4754D29354507307FF53D6839BACAFBC3CFBDBD639C7869A1B296FF0D3A43D9081B8012C51C8527647C59C9EA6863B29C5BD011A3B1B246838A74CC4225BAC3E027D99295759CD553414C6D4B4A8C0E340B50B5B9D9FFDD96B221EB6D0DA28AD1EDD641739A88DFBB65EACEDF8E4FE7FC8829B235A43925392BAEC3E6690F98094E21A63C2BA4BA6C63BADD8472B5DEC40877473C9D0326700005274646c31323334", Param_memo: "超级管理员密码(Rtdl1234)"}) //初始化功能菜单 bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "0", Code: "userpriv", Name: "功能菜单"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_device", Name: "终端管理"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_device_group", Name: "终端分组管理"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_alertpolicy", Name: "告警策略配置"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_devicedata", Name: "数据管理"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_sys", Name: "用户管理"}) bo.SaveGlobalCode(bo.Global_const_code{Parentcode: "userpriv", Code: "pl_log", Name: "日志管理"}) } else { } } else { tools.CheckErr(dberr) } return conn, nil } func CloseSqlite() { if global.SqliteConn != nil { global.SqliteConn.Close() } }