🚀 go-xorm封装公共CURD类

go-xorm封装公共CURD类

$ 前言

  • go-xorm是一个简单而强大的Go语言ORM库. 通过它可以使数据库操作非常简便。

$ 配置目标

  • 封装一个公共的父级操作类,
  • 通过继承可让子类带有简单的操作CRUD分页查询数据库的能力
  • 可减少很多冗余,重复相似性很高的查询的代码编写
  • 提高代码整洁性,可维护性

$ 参考项目地址

https://www.jianshu.com/p/ed748eae96c9

app.ini配置

[mysql]
url=root:abc.123@tcp(localhost:3306)/foot?charset=utf8
maxIdle=10
maxConn=50

utils工具类源码(用于读取配置信息)

package utils

import (
    "gopkg.in/ini.v1"
    "tesou.io/platform/foot-parent/foot-api/common/base"
)

var (
    //配置信息
    iniFile *ini.File
)

func init() {
    file, e := ini.Load("conf/app.ini")
    if e != nil {
        base.Log.Info("Fail to load conf/app.ini" + e.Error())
        return
    }
    iniFile = file
}

func GetSection(sectionName string) *ini.Section {
    section, e := iniFile.GetSection(sectionName)
    if e != nil {
        base.Log.Info("未找到对应的配置信息:" + sectionName + e.Error())
        return nil
    }
    return section
}

func GetSectionMap(sectionName string) map[string]string {
    section, e := iniFile.GetSection(sectionName)
    if e != nil {
        base.Log.Info("未找到对应的配置信息:" + sectionName + e.Error())
        return nil
    }
    section_map := make(map[string]string, 0)
    for _, e := range section.Keys() {
        section_map[e.Name()] = e.Value()
    }
    return section_map
}

func GetVal(sectionName string, key string) string {
    var temp_val string
    section := GetSection(sectionName)
    if nil != section {
        temp_val = section.Key(key).Value()
    }
    return temp_val;
}

Page源码(分页结构体)

package pojo

type Page struct {
    //记录总数
    Counts int64
    //每页显示记录数
    PageSize int64
    //总页数
    TotalPage int64
    //当前页
    CurPage int64
    //页面显示开始记录数
    FirstResult int64
    //页面显示最后记录数
    LastResult int64
    //排序类型
    OrderType string
    //排序名称
    OrderName string
}

func (this *Page) Build(counts int64, pageSize int64) {
    this.Counts = counts
    this.PageSize = pageSize
    if (counts%pageSize == 0) {
        this.TotalPage = this.Counts / this.PageSize
    } else {
        this.TotalPage = this.Counts/this.PageSize + 1
    }
}

func (this *Page) GetCounts() int64 {
    return this.Counts
}

/**
 *  Counts
 *            the Counts to set
 */
func (this *Page) SetCounts(counts int64) {
    // 计算所有的页面数
    this.Counts = counts
    // this.TotalPage = (int)Math.ceil((this.Counts + this.perPageSize - 1)
    // / this.perPageSize)
    if (counts%this.PageSize == 0) {
        this.TotalPage = this.Counts / this.PageSize
    } else {
        this.TotalPage = this.Counts/this.PageSize + 1
    }
}

func (this *Page) GetPageSize() int64 {
    return this.PageSize
}

func (this *Page) SetPageSize(pageSize int64) {
    this.PageSize = pageSize
}

/**
 *  the TotalPage
 */
func (this *Page) GetTotalPage() int64 {
    if this.TotalPage < 1 {
        return 1
    }
    return this.TotalPage
}

/**
 *  TotalPage
 *            the TotalPage to set
 */
func (this *Page) SetTotalPage(totalPage int64) {
    this.TotalPage = totalPage
}

func (this *Page) GetCurPage() int64 {
    return this.CurPage
}

func (this *Page) SetCurPage(curPage int64) {
    this.CurPage = curPage
}

/**
 *  the FirstResult
 */
func (this *Page) GetFirstResult() int64 {
    temp := this.CurPage - 1
    if (temp <= 0) {
        return 0
    }
    this.FirstResult = (this.CurPage - 1) * this.PageSize
    return this.FirstResult
}

/**
 *  FirstResult
 *            the FirstResult to set
 */
func (this *Page) SetFirstResult(firstResult int64) {
    this.FirstResult = firstResult
}

/**
 *  the LastResult
 */
func (this *Page) GetLastResult() int64 {
    this.LastResult = this.FirstResult + this.PageSize
    return this.LastResult
}

/**
 *  LastResult
 *            the LastResult to set
 */
func (this *Page) SetLastResult(lastResult int64) {
    this.LastResult = lastResult
}

/**
 *  the OrderName
 */
func (this *Page) GetOrderName() string {
    return this.OrderName
}

/**
 *  OrderName
 *            the OrderName to set
 */
func (this *Page) SetOrderName(orderName string) {
    this.OrderName = orderName
}

/**
 *  the orderBy
 */
func (this *Page) getOrderType() string {
    return this.OrderType
}

/**
 *  orderBy
 *            the orderBy to set
 */
func (this *Page) SetOrderType(orderType string) {
    this.OrderType = orderType
}

/**
 *  the orderBy
 */
func (this *Page) GetOrderBy() string {
    if len(this.GetOrderName()) <= 0 {
        return ""
    }
    orderBy := " order by " + this.GetOrderName() + " " + this.getOrderType()
    return orderBy
}

BaseService源码(公共的CRUD类)

package mysql

import (
    "container/list"
    _ "github.com/go-sql-driver/mysql"
    "github.com/go-xorm/core"
    "github.com/go-xorm/xorm"
    "gopkg.in/mgo.v2/bson"
    "reflect"
    "strconv"
    "tesou.io/platform/foot-parent/foot-api/common/base"
    "tesou.io/platform/foot-parent/foot-api/common/base/pojo"
    "tesou.io/platform/foot-parent/foot-core/common/utils"
)

type BaseService struct {
}

var (
    engine *xorm.Engine
)

func GetEngine() *xorm.Engine {
    if nil == engine {
        setEngine()
    }
    return engine
}

func ShowSQL(show bool) {
    engine := GetEngine()
    engine.ShowSQL(show)
    engine.ShowExecTime(show)

}

func setEngine() *xorm.Engine {
    url := utils.GetVal("mysql", "url")
    maxIdle, _ := strconv.Atoi(utils.GetVal("mysql", "maxIdle"))
    maxConn, _ := strconv.Atoi(utils.GetVal("mysql", "maxConn"))

    var err error
    engine, err = xorm.NewEngine("mysql", url)
    if nil != err {
        base.Log.Error("init" + err.Error())
    }

    //engine.ShowExecTime(true)
    //则会在控制台打印出生成的SQL语句
    //则会在控制台打印调试及以上的信息
    engine.ShowSQL(true)
    //engine.Logger().SetLevel(core.LOG_DEBUG)
    engine.SetMaxIdleConns(maxIdle)
    engine.SetMaxOpenConns(maxConn)
    tbMapper := core.NewPrefixMapper(core.SnakeMapper{}, "t_")
    engine.SetTableMapper(tbMapper)
    engine.SetColumnMapper(core.SameMapper{})
    /**
    当使用了Distinct,Having,GroupBy方法将不会使用缓存
    在Get或者Find时使用了Cols,Omit方法，则在开启缓存后此方法无效，系统仍旧会取出这个表中的所有字段。
    在使用Exec方法执行了方法之后，可能会导致缓存与数据库不一致的地方。因此如果启用缓存，尽量避免使用Exec。
    如果必须使用，则需要在使用了Exec之后调用ClearCache手动做缓存清除的工作。比如：
    engine.Exec("update user set name = ? where id = ?", "xlw", 1)
    engine.ClearCache(new(User))
    */
    //cacher := xorm.NewLRUCacher(xorm.NewMemoryStore(), 999)
    //engine.SetDefaultCacher(cacher)

    return engine
}

func init() {
    //设置初始化数据库引擎
    setEngine()
}

func beforeModify(entity interface{}) {
    //当前时间
    //current_date := time.Now().Format("2006-01-02 15:04:05")
    //默认更新者
    default_user := "100000"
    //对象操作
    entity_value := reflect.ValueOf(entity).Elem()

    /*//设置更新时间
    field_ModifyDate := entity_value.FieldByName("ModifyTime")
    if field_ModifyDate.String() == "" {
        field_ModifyDate.SetString(current_date)
    }*/
    //设置更新者
    field_ModifyUser := entity_value.FieldByName("ModifyUser")
    if field_ModifyUser.String() == "" {
        field_ModifyUser.SetString(default_user)
    }

}

func beforeDelete(entity interface{}) {
    //当前时间
    //current_date := time.Now().Format("2006-01-02 15:04:05")
    //默认删除者
    default_user := "100000"
    //对象操作
    entity_value := reflect.ValueOf(entity).Elem()

    /*//设置更新时间
    field_ModifyDate := entity_value.FieldByName("ModifyTime")
    if field_ModifyDate.String() == "" {
        field_ModifyDate.SetString(current_date)
    }*/
    //设置删除者
    field_DeleteUser := entity_value.FieldByName("DeleteUser")
    if field_DeleteUser.String() == "" {
        field_DeleteUser.SetString(default_user)
    }

}

func beforeSave(entity interface{}) interface{} {
    //当前时间
    //current_date := time.Now().Format("2006-01-02 15:04:05")
    //默认创建者
    default_user := "100000"
    //对象操作
    entity_value := reflect.ValueOf(entity).Elem()

    /*  //设置创建时间  通过`xorm:"created"`配置
        field_CreateDate := entity_value.FieldByName("CreateTime")
        if field_CreateDate.String() == "" {
            field_CreateDate.SetString(current_date)
        }*/
    //设置创建者
    field_CreateUser := entity_value.FieldByName("CreateUser")
    if field_CreateUser.String() == "" {
        field_CreateUser.SetString(default_user)
    }

    beforeModify(entity)

    var id interface{}
    //设置id
    field_Id := entity_value.FieldByName("Id")
    if field_Id.String() == "" {
        //使用bson.NewObject作为主键
        id = bson.NewObjectId().Hex()
        field_Id.Set(reflect.ValueOf(id))
    }
    return id
}

func (this *BaseService) SaveOrModify(entity interface{}) {
    b, err := engine.Exist(entity)
    if nil != err {
        base.Log.Info("SaveOrModify:" + err.Error())
    }
    if b {
        this.Modify(entity)
    } else {
        this.Save(entity)
    }
}
func (this *BaseService) Save(entity interface{}) interface{} {
    id := beforeSave(entity)
    _, err := engine.InsertOne(entity)
    if nil != err {
        base.Log.Info("Save:" + err.Error())
    }
    return id
}

func (this *BaseService) SaveList(entitys []interface{}) *list.List {
    if len(entitys) <= 0 {
        return nil
    }
    list_ids := list.New()
    for _, v := range entitys {
        id := beforeSave(v)
        list_ids.PushBack(id)
    }

    _, err := engine.Insert(entitys...)
    if nil != err {
        base.Log.Info("SaveList:" + err.Error())
    }
    return list_ids
}

func (this *BaseService) Del(entity interface{}) int64 {
    beforeDelete(entity)
    entity_value := reflect.ValueOf(entity).Elem()
    id_field := entity_value.FieldByName("Id")
    i, err := engine.Id(id_field.Interface()).Delete(entity)
    if err != nil {
        base.Log.Info("Del:", err)
    }
    return i
}

func (this *BaseService) Modify(entity interface{}) int64 {
    beforeModify(entity)

    entity_value := reflect.ValueOf(entity).Elem()
    id_field := entity_value.FieldByName("Id")
    i, err := engine.Id(id_field.Interface()).AllCols().Update(entity)
    if err != nil {
        base.Log.Info("Modify:", err)
    }
    return i
}

func (this *BaseService) ModifyList(entitys []interface{}) int64 {
    if len(entitys) <= 0 {
        return 0
    }
    //i, err := engine.In("id",ids).Update(entitys)
    for _, v := range entitys {
        //entity_value := reflect.ValueOf(v).Elem()
        //id_field := entity_value.FieldByName("Id")
        this.Modify(v)
    }
    return 1
}

func (this *BaseService) Exist(entity interface{}) bool {
    //对象操作
    exist, err := engine.Exist(entity)
    if nil != err {
        base.Log.Info("ExistByName:" + err.Error())
    }
    return exist
}

func (this *BaseService) Find(entity interface{}) {
    engine.Find(entity)
}

func (this *BaseService) FindBySQL(sql string, entity interface{}) {
    engine.SQL(sql).Find(entity)
}

func (this *BaseService) FindAll(entity interface{}) {
    err := engine.Find(entity)
    if nil != err {
        base.Log.Info("FindAll: " + err.Error())
    }
}

/**
分页查询
*/
func (this *BaseService) Page(v interface{}, page *pojo.Page, dataList interface{}) error {
    tableName := engine.TableName(v)
    sql := "select t.* from " + tableName + " t where 1=1 "

    return this.PageSql(sql, page, dataList)
}

/**
分页查询
*/
func (this *BaseService) PageSql(sql string, page *pojo.Page, dataList interface{}) error {
    //声明结果变量
    var err error
    var counts int64
    //获取总记录数处理
    countSql := " select count(1) from (" + sql + ") t"
    counts, err = engine.SQL(countSql).Count()
    if nil != err {
        return err
    } else {
        page.SetCounts(counts)
    }
    //排序处理
    orderBy := page.GetOrderBy()
    if len(orderBy) > 0 {
        sql += orderBy
    }
    sql += " limit " + strconv.FormatInt(page.GetFirstResult(), 10) + "," + strconv.FormatInt(page.GetPageSize(), 10)
    err = engine.SQL(sql).Find(dataList)
    return err
}

使用示例

package service

import (
    "tesou.io/platform/foot-parent/foot-api/common/base"
    "tesou.io/platform/foot-parent/foot-api/module/match/pojo"
    "tesou.io/platform/foot-parent/foot-core/common/base/service/mysql"
)

type MatchHisService struct {
    //引入公共类
    mysql.BaseService
}

func (this *MatchHisService) Exist(v *pojo.MatchHis) bool {
    has, err := mysql.GetEngine().Table("`t_match_his`").Where(" `Id` = ?  ", v.Id).Exist()
    if err != nil {
        base.Log.Error("Exist", err)
    }
    return has
}

func (this *MatchHisService) FindAll() []*pojo.MatchHis {
    dataList := make([]*pojo.MatchHis, 0)
    mysql.GetEngine().OrderBy("MatchDate").Find(&dataList)
    return dataList
}

func (this *MatchHisService) FindById(matchId string) *pojo.MatchHis {
    data := new(pojo.MatchHis)
    data.Id = matchId
    _, err := mysql.GetEngine().Get(data)
    if err != nil {
        base.Log.Error("FindById:", err)
    }
    return data
}

func (this *MatchHisService) FindBySeason(season string) []*pojo.MatchLast {
    sql_build := `
SELECT 
  la.* 
FROM
  foot.t_match_his la
WHERE 1=1
    `
    sql_build = sql_build + " AND la.MatchDate >= '" + season + "-01-01 00:00:00' AND la.MatchDate <= '" + season + "-12-31 23:59:59'"

    //结果值
    dataList := make([]*pojo.MatchLast, 0)
    //执行查询
    this.FindBySQL(sql_build, &dataList)
    return dataList
}
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 204,445评论 6赞 478
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 85,889评论 2赞 381
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 151,047评论 0赞 337
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,760评论 1赞 276
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,745评论 5赞 367
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,638评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 38,011评论 3赞 398
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,669评论 0赞 258
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,923评论 1赞 299
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2赞 321
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1赞 330
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,406评论 4赞 320
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,995评论 3赞 307
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1赞 260
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2赞 350
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2赞 342

推荐阅读更多精彩内容

  • Java面试
    一. Java基础部分.................................................
    wy_sure阅读 3,788评论 0赞 11
  • Thinkphp 第一章:数据库架构基础
    本章我们首先从ThinkPHP5.0的数据库访问层架构设计原理开始，然后熟悉下数据库的配置，并掌握如何进行基础的查...
    寒冬夜行人_51a4阅读 2,016评论 11赞 6
  • 我的朋友苏明海
    我喜欢有个性的人，苏明海就是其中一个。 苏明海有过三个女人，两个是在小学谈的，一个是现在谈的老婆，我到现在还记得...
    椽罔阅读 756评论 1赞 2
  • 生活很苦，生活很累，但我们必须要坚持！
    昨晚奶奶给我打来电话，问我什么时候回去过年。我说我们还有十多天才放假，电话那头沉默了一会儿，我知道奶奶想急切见到我...
    罗志远阅读 1,974评论 6赞 9
  • 2019-11-09
    做生意难是普遍现象 要求房东降房租，房东要求房地产，降价，房地产要求钢筋，水泥，人工，地皮降价，最后发现是债务成本...
    1ce94d33416d阅读 191评论 0赞 2