light-apiengine-feature/engine/system_init.go

package engine

import (
	"fmt"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/utils"
	"git.qianqiusoft.com/qianqiusoft/light-apiengine/logs"
	"github.com/xormplus/xorm"
	"regexp"
	"time"
)


func createDatabase(orm *xorm.Engine)  {
	if orm.DriverName() == "mysql" {
		dbname := orm.DataSourceName()
		reg := regexp.MustCompile("(.*)/(.*)(\\?.*)")
		//strings := reg.FindAllStringSubmatch(s, -1)
		strings := reg.FindStringSubmatch(dbname)
		fmt.Println(strings)
		if len(strings) > 3 {
			dbname = strings[2]
			db, _ := xorm.NewEngine(orm.DriverName(), strings[1]+"/")
			db.ShowSQL(true)
			_, err := db.Sql("CREATE DATABASE IF NOT EXISTS `" + dbname + "` default charset utf8 COLLATE utf8_general_ci").Execute()
			if err != nil {
				fmt.Println("create database error", err)
			} else {
				fmt.Println("create database success:", dbname)
			}
		}
	}
}

func initPlatformDbData(db *xorm.Engine) {

	orgs := []models.SysOrg{
		{
			Id:   "287c50a4-caad-4557-80c8-aca81af9f370",
			Name: "千秋软件",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
	}
	users := []models.SysUser{
		{
			Id:       "187c50a4-caad-4557-80c8-aca81af9f370",
			Name:     "管理员",
			LoginId:  "admin",
			OrgId:    "287c50a4-caad-4557-80c8-aca81af9f370",
			Password: "123456",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
	}

	roles := []models.SysRole{
		{
			Id:   "487c50a4-caad-4557-80c8-aca81af9f370",
			Name: "超级管理员",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
	}

	userRoles := []models.SysUserRole{
		{
			Id:     "587c50a4-caad-4557-80c8-aca81af9f370",
			RoleId: "387c50a4-caad-4557-80c8-aca81af9f370",
			UserId: "187c50a4-caad-4557-80c8-aca81af9f370",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
	}

	menus := []models.SysMenu{
		{
			Id:   "387c50a4-caad-4557-80c8-aca81af9f370",
			Name: "系统管理",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f371",
			Name:   "用户管理",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/user",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f372",
			Name:   "组织架构",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/org",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f373",
			Name:   "角色管理",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/role",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f374",
			Name:   "菜单管理",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/menu",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f375",
			Name:   "字典管理",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/dict",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
		{
			Id:     "387c50a4-caad-4557-80c8-aca81af9f376",
			Name:   "日志管理",
			Parent: "387c50a4-caad-4557-80c8-aca81af9f370",
			Url:    "/sys/log",
			CreateTime:time.Now(),
			LastUpdateDate: time.Now(),
		},
	}

	models.SyncDb(db)

	_, err := db.Insert(orgs)
	printError(err)
	_, err = db.Insert(users)
	printError(err)
	_, err = db.Insert(roles)
	printError(err)
	_, err = db.Insert(userRoles)
	printError(err)
	_, err = db.Insert(menus)
	printError(err)
}


func initOrmEngine(orm *xorm.Engine, driverName string) {
	path, _ := utils.GetCurrentPath()
	sqldir := path + "/sqlconfig"
	//注册SqlMap配置，xml格式
	err := orm.RegisterSqlMap(xorm.Xml(sqldir, ".xml"))
	printError(err)
	//注册SqlTemplate配置，使用Pongo2模板引擎
	err = orm.RegisterSqlTemplate(xorm.Pongo2(sqldir, "_"+driverName+".stpl"))
	printError(err)
	//注册SqlTemplate配置，使用Jet模板引擎
	err = orm.RegisterSqlTemplate(xorm.Jet(sqldir, "_"+driverName+".jet"))
	printError(err)
	//注册SqlTemplate配置，使用html/template模板引擎
	err = orm.RegisterSqlTemplate(xorm.Default(sqldir, "_"+driverName+".tpl"))
	printError(err)

	//设置时区
	orm.DatabaseTZ = time.Local // 必须
	orm.TZLocation = time.Local // 必须
	time.LoadLocation("Asia/Shanghai")

	orm.StartFSWatcher()

	orm.ShowSQL(true)
	/*for k,_:=range orm.SqlMap.Sql{
		fmt.Println(k)
	}*/
	//orm.ShowExecTime(true)
	createDatabase(orm)
}


func printError(err error) {
	if err != nil {
		logs.Error(err.Error())
	}
}