scd_tools_go/service/db/db_sqlite3.go

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()
	}
}