golang gorm通过泛型实现通用单表增删改

news/2024/10/28 22:35:53/

golang gorm通过泛型实现通用单表增删改

无废话，直接上代码

想实现通用，首先得实现查询的通用，可以用传递map实现

func Where(where map[string]interface{}) func(db *gorm.DB) *gorm.DB {return func(db *gorm.DB) *gorm.DB {dbTmp := dbfor ko, value := range where {koSlice := strings.Split(ko, "#")var key, op stringif len(koSlice) == 1 {key = koif php2go.InArray(key, []string{"select", "order", "offset", "limit"}) {op = ko} else {op = "eq"}} else {key = koSlice[0]op = koSlice[1]}switch op {case "eq":dbTmp.Where(key+" = ?", value)case "neq":dbTmp.Where(key+" != ?", value)case "gt":dbTmp.Where(key+" > ?", value)case "gte":dbTmp.Where(key+" >= ?", value)case "lt":dbTmp.Where(key+" < ?", value)case "lte":dbTmp.Where(key+" <= ?", value)case "like":dbTmp.Where(key+" like ?", "%"+value.(string)+"%")case "in":dbTmp.Where(key+" in ?", value)case "notin":dbTmp.Where(key+" not in ?", value)case "between":dbTmp.Where(key+" between ? and ?", value.([]int)[0], value.([]int)[1])case "betweens":dbTmp.Where(key+" between ? and ?", value.([]string)[0], value.([]string)[1])case "select":dbTmp.Select(value.(string))case "order":dbTmp.Order(value.(string))case "offset":dbTmp.Offset(value.(int))case "limit":dbTmp.Limit(value.(int))}}return dbTmp}
}

调用方式

examleWhere := map[string]interface{}{"name#like":           data.ToUserName,"id":        		   data.UserID,"create_time#between": []int{17000000, 18000000},"grade#in":            []int{3,4},"order":               "id desc",}
Gorm().Table("xxx").Scopes(Where(examleWhere)).xxx打印 sql:
select * from xxx where name like "%xx%" and id = xxx and create_time between 17xxx and 18xxx and grade in (3,4) order by id desc

通用查询

type FieldType interface {uint | uint8 | uint16 | uint32 | uint64 | int | int8 | int16 | int32 | int64 | float32 | float64 | string
}
func Get[T any](where map[string]interface{}, field ...string) (ret T, err error) {db := sys.Gorm().Scopes(Where(where)).Where("delete_time = 0")if len(field) > 0 {db.Select(field)}err = db.Find(&ret).Errorreturn
}func GetVal[T FieldType](tableName string, where map[string]interface{}, field string, includeDel ...bool) (ret T, err error) {db := sys.Gorm().Table(tableName).Scopes(Where(where)).Select(field)if len(includeDel) > 0 && includeDel[0] == true {db.Unscoped()} else {db.Where("delete_time = 0")}err = db.Find(&ret).Errorreturn
}func Gets[T any](where map[string]interface{}, field ...string) (ret []T, err error) {db := sys.Gorm().Scopes(Where(where)).Where("delete_time = 0")if len(field) > 0 {db.Select(field)}err = db.Find(&ret).Errorreturn
}func GetSlice[T FieldType](tableName string, where map[string]interface{}, field string, includeDel ...bool) (ret []T, err error) {ret = make([]T, 0)db := sys.Gorm().Table(tableName).Scopes(Where(where)).Select(field)if len(includeDel) > 0 && includeDel[0] == true {db.Unscoped()} else {db.Where("delete_time = 0")}err = db.Find(&ret).Errorreturn
}func Update[T any](where map[string]interface{}, update T) (err error) {db := sys.Gorm()err = db.Scopes(Where(where)).Updates(update).Errorreturn
}func UpdateByMap(table string, where, update map[string]interface{}) (err error) {db := sys.Gorm().Table(table)err = db.Scopes(Where(where)).Updates(update).Errorreturn
}func UpdateByMapTx(tx *gorm.DB, table string, where, update map[string]interface{}) (err error) {db := tx.Table(table)err = db.Scopes(Where(where)).Updates(update).Errorreturn
}// Create
// doUpdate: on duplicate key update : all, []string{}...
func Create[T any](data *[]T, doUpdate ...string) (err error) {var db *gorm.DBif len(doUpdate) > 0 {if doUpdate[0] == "all" {db = sys.Gorm().Clauses(clause.OnConflict{UpdateAll: true,}).Scopes()} else {db = sys.Gorm().Clauses(clause.OnConflict{DoUpdates: clause.AssignmentColumns(doUpdate),}).Scopes()}} else {db = sys.Gorm().Scopes()}err = db.Create(&data).Errorreturn
}func CreateTx[T any](tx *gorm.DB, data *T, doUpdate ...string) (err error) {if len(doUpdate) > 0 {if doUpdate[0] == "all" {tx.Clauses(clause.OnConflict{UpdateAll: true,}).Scopes()} else {tx.Clauses(clause.OnConflict{DoUpdates: clause.AssignmentColumns(doUpdate),}).Scopes()}}err = tx.Create(&data).Errorreturn
}func CreatesTx[T any](tx *gorm.DB, data *[]T, doUpdate ...string) (err error) {if len(doUpdate) > 0 {if doUpdate[0] == "all" {tx.Clauses(clause.OnConflict{UpdateAll: true,}).Scopes()} else {tx.Clauses(clause.OnConflict{DoUpdates: clause.AssignmentColumns(doUpdate),}).Scopes()}}err = tx.Create(&data).Errorreturn
}

给出调用方式案例

model.Get[model.WwStaff](map[string]interface{}{"corp_id": req.CorpId, "userid": req.StaffUserid})model.GetVal[int]("table_name", map[string]interface{}{"corp_id":     corpId,"dept_id":     deptId,"delete_time": 0,}, "parentid")model.GetSlice[int64]("table_name",map[string]interface{}{"staff_id": req.Id}, "id")// insert ...
model.Create(&tableModel)
// insert ... ON DUPLICATE KEY UPDATE id = VALUES(id) ......
model.Create(&tableModel, "id", "update_time")model.UpdateByMap("table_name",map[string]interface{}{"id#in": roomUnionIds},map[string]interface{}{"upload": 2})model.UpdateByMapTx(tx, "table_name",map[string]interface{}{"id#in": roomUnionIds},map[string]interface{}{"upload": 2})

可以依照代码，写出其它灵活的使用方式
结束


http://www.ppmy.cn/news/1201509.html

相关文章

【鸿蒙软件开发】ArkUI容器组件之Grid(网格布局)

【鸿蒙软件开发】ArkUI容器组件之Grid(网格布局)

文章目录 前言一、Grid1.1 子组件GridItem是什么子组件接口属性事件示例代码 1.2 接口参数 1.3 属性1.4 Grid的几种布局模式1.5 GridDirection枚举说明1.6事件ItemDragInfo对象说明 1.7 示例代码 总结 前言 Grid容器组件&#xff1a;网格容器&#xff0c;由“行”和“列”分割…
阅读更多...
OpenCV入门3：像素操作

OpenCV入门3：像素操作

在OpenCV中&#xff0c;图像的像素值是以一个多维数组的形式表示的。上一篇已经介绍了cv::Mat类。对于图像中的每一个像素&#xff0c;可以通过Mat对象中的at<type>(i,j)函数&#xff08;type可以是uchar、int等&#xff09;获得Mat对象的像素值。 访问像素值&#xff1…
阅读更多...
汽车网络安全渗透测试概述

汽车网络安全渗透测试概述

目录 1.汽车网络安全法规概述 1.1 国外标准 1.2 国内标准 2.汽车网络安全威胁分析 2.1 汽车网络安全资产定义 2.2 汽车网络安全影响场景及评级示例 3.汽车网络安全渗透测试描述 3.1 参考法规 3.2 渗透测试内容 4.小结 1.汽车网络安全法规概述 近年来&#xff0c;汽车…
阅读更多...
STM32_project:led_beep

STM32_project:led_beep

代码&#xff1a; 主要部分&#xff1a; #include "stm32f10x.h" // Device header #include "delay.h"// 给蜂鸣器IO口输出低电平&#xff0c;响&#xff0c;高&#xff0c;不向。 //int main (void) //{ // // 开启时钟 // RC…
阅读更多...
存在重复元素 II[简单]

存在重复元素 II[简单]

优质博文&#xff1a;IT-BLOG-CN 一、题目 给你一个整数数组nums和一个整数k&#xff0c;判断数组中是否存在两个不同的索引i和j&#xff0c;满足nums[i] nums[j]且abs(i - j) < k。如果存在&#xff0c;返回true&#xff1b;否则&#xff0c;返回false。 示例 1&#x…
阅读更多...
了解web3，什么是web3

了解web3，什么是web3

Web3是指下一代互联网&#xff0c;它基于区块链技术&#xff0c;将各种在线活动更加安全、透明和去中心化。Web3是一个广义的概念&#xff0c;它包括了很多方面&#xff0c;如数字货币、去中心化应用、智能合约等等。听不懂且大多数人听到这个东西&#xff0c;直觉感觉就像骗子…
阅读更多...
React实现一个拖拽排序组件 - 支持多行多列、支持TypeScript、支持Flip动画、可自定义拖拽区域

React实现一个拖拽排序组件 - 支持多行多列、支持TypeScript、支持Flip动画、可自定义拖拽区域

一、效果展示 排序&#xff1a; 丝滑的Flip动画 自定义列数 &#xff08;并且宽度会随着屏幕宽度自适应&#xff09; 自定义拖拽区域&#xff1a;&#xff08;扩展性高&#xff0c;可以全部可拖拽、自定义拖拽图标&#xff09; 二、主要思路 Tip&#xff1a; 本代码的CSS使用…
阅读更多...
ceval 数据集明文位置编码嵌入

ceval 数据集明文位置编码嵌入

明文位置编码嵌入 数据集地址嵌入代码解释说明数据集地址 ceval 数据集 嵌入代码 import pandas as pd from glob import glob from tqdm import tqdm# 训练集数据处理 voc = set() one_data_list = []
阅读更多...
最新文章

Copyright @ 2022~2023