xormplus__xorm/engineplus.go

package xorm

import (
	"encoding/json"
	"reflect"

	"gopkg.in/flosch/pongo2.v3"
)

func (engine *Engine) SetSqlMapRootDir(sqlMapRootDir string) *Engine {
	engine.sqlMap.SqlMapRootDir = sqlMapRootDir
	return engine
}

func (engine *Engine) SetSqlTemplateRootDir(sqlTemplateRootDir string) *Engine {
	engine.sqlTemplate.SqlTemplateRootDir = sqlTemplateRootDir
	return engine
}

func (engine *Engine) SqlMapClient(sqlTagName string, args ...interface{}) *Session {
	session := engine.NewSession()
	session.IsAutoClose = true
	session.IsSqlFuc = true
	return session.Sql(engine.sqlMap.Sql[sqlTagName], args...)
}

func (engine *Engine) SqlTemplateClient(sqlTagName string, args ...interface{}) *Session {
	session := engine.NewSession()
	session.IsAutoClose = true
	session.IsSqlFuc = true

	if engine.sqlTemplate.Template[sqlTagName] == nil {
		if len(args) == 0 {
			return session.Sql("")
		} else {
			map1 := args[0].(*map[string]interface{})
			return session.Sql("", map1)
		}
	}

	if len(args) == 0 {
		parmap := &pongo2.Context{"1": 1}
		sql, err := engine.sqlTemplate.Template[sqlTagName].Execute(*parmap)
		if err != nil {
			engine.logger.Error(err)

		}
		return session.Sql(sql)
	} else {
		map1 := args[0].(*map[string]interface{})
		sql, err := engine.sqlTemplate.Template[sqlTagName].Execute(*map1)
		if err != nil {
			engine.logger.Error(err)

		}
		return session.Sql(sql, map1)
	}

}

func (engine *Engine) Search(beans interface{}, condiBeans ...interface{}) ResultStructs {
	session := engine.NewSession()
	defer session.Close()
	return session.Search(beans, condiBeans...)
}

// Get retrieve one record from table, bean's non-empty fields
// are conditions
func (engine *Engine) GetFirst(bean interface{}) ResultBean {
	session := engine.NewSession()
	defer session.Close()
	return session.GetFirst(bean)
}

func JSONString(v interface{}, IndentJSON bool) (string, error) {
	var result []byte
	var err error
	if IndentJSON {
		result, err = json.MarshalIndent(v, "", "  ")
	} else {
		result, err = json.Marshal(v)
	}
	if err != nil {
		return "", err
	}

	if string(result) == "null" {
		return "", nil
	}
	return string(result), nil
}

func (engine *Engine) Sqls(sqls interface{}, parmas ...interface{}) *SqlsExecutor {
	session := engine.NewSession()
	session.IsSqlFuc = true
	return session.Sqls(sqls, parmas...)
}

func (engine *Engine) SqlMapsClient(sqls interface{}, parmas ...interface{}) *SqlMapsExecutor {
	session := engine.NewSession()
	session.IsSqlFuc = true
	return session.SqlMapsClient(sqls, parmas...)
}

func (engine *Engine) SqlTemplatesClient(sqls interface{}, parmas ...interface{}) *SqlsExecutor {
	session := engine.NewSession()
	session.IsSqlFuc = true
	return session.Sqls(sqls, parmas...)
}

func (engine *Engine) BatchSql(sqls interface{}) *Session {
	session := engine.NewSession()
	types := reflect.TypeOf(sqls)
	if types.Kind() == reflect.Map {
		engine.logger.Info("sqls is Map")
		engine.logger.Info(types.Elem())
		engine.logger.Info(types.Elem().Kind())
	}

	if types.Kind() == reflect.Slice {
		engine.logger.Info("sqls is Slice")
		engine.logger.Info(types.Elem())
		engine.logger.Info(types.Elem().Kind())
	}

	switch sqls.(type) {
	case []string:
		engine.logger.Info("sqls is []string")
	case map[string]string:
		engine.logger.Info("sqls is map[string]string")
	}
	return session.BatchSql()
}