RawQuerier 原生查询

于 2023-04-03 22:30:35 发布
阅读量344 收藏
点赞数
分类专栏： go语言基础 go 实战开发 文章标签： mysql 数据库 sql
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_48943299/article/details/129941527
版权

RawQuerier 原生查询

语法分析

就 MySQL SELECT 语句来说，ORM 框架能支持全部 语法吗？ 显然不能，也不愿意支持全部 ， 也不仅仅是 ORM 框架，大多数框架设计的时候，都要考 虑提供兜底的措施，或者提供绕开你的框架的机制。 在 ORM 这里，就是要允许用户手写 SQL，直接绕开 ORM 的各种机制。
image.png

开源实例

Beego ORM

image.png
image.png
Beego ORM 的原生查询接口是，直接让用户传入 sql 语句与对应的参数，然后返回一个原生查询的抽象 RawSeter，该抽象提供了众多的方法支持。

GORM

image.png
image.png
image.png
image.png
GORM 也是依赖于用户传 sql 语句与参数，但是返回给用户的是 DB，用户负责调用目标获取结果的接口，例如: Row()、Rows()、Scan() 等方法。

type Result struct {
  ID   int
  Name string
  Age  int
}

var result Result
db.Raw("SELECT id, name, age FROM users WHERE name = ?", 3).Scan(&result)

db.Raw("SELECT id, name, age FROM users WHERE name = ?", 3).Scan(&result)

var age int
db.Raw("SELECT SUM(age) FROM users WHERE role = ?", "admin").Scan(&age)

var users []User
db.Raw("UPDATE users SET name = ? WHERE age = ? RETURNING id, name", "jinzhu", 20).Scan(&users)

// 使用原生 SQL
row := db.Raw("select name, age, email from users where name = ?", "jinzhu").Row()
row.Scan(&name, &age, &email)

db.Exec("DROP TABLE users")
db.Exec("UPDATE orders SET shipped_at = ? WHERE id IN ?", time.Now(), []int64{1, 2, 3})

// Exec with SQL Expression
db.Exec("UPDATE users SET money = ? WHERE name = ?", gorm.Expr("money * ? + ?", 10000, 1), "jinzhu")

API 设计

本文主要支持中间这种。 第三种用户可以直接使用 sql.DB，都用不着 ORM 框架
image.png

var _ Querier[any] = &RawQuerier[any]{}

// RawQuerier 原生查询器
type RawQuerier[T any] struct {
	core
	sess session
	sql string
	args []any
}

func (r *RawQuerier[T]) Exec(ctx context.Context) Result {
	// TODO implement me
	panic("implement me")
}

func (r *RawQuerier[T]) Get(ctx context.Context) (*T, error) {
	// TODO implement me
	panic("implement me")
}

func (r *RawQuerier[T]) GetMulti(ctx context.Context) ([]*T, error) {
	// TODO implement me
	panic("implement me")
}

具体实现

其中exec 和 get 两个方法是从我们原本的 Selector 与 Updateor 等实现里面抽取出来的。 目的是为了实现
Executor、Querier 等接口，另外为了兼容 Seletor 等构造模式的构造方式，也必须实现 QueryBuilder 接口。

var _ Querier[any] = &RawQuerier[any]{}

// RawQuerier 原生查询器
type RawQuerier[T any] struct {
	core
	sess session
	sql  string
	args []any
}

// RawQuery 创建一个 RawQuerier 实例
// 泛型参数 T 是目标类型。
// 例如，如果查询 User 的数据，那么 T 就是 User
func RawQuery[T any](sess session, sql string, args ...any) *RawQuerier[T] {
	return &RawQuerier[T]{
		core: sess.getCore(),
		sess: sess,
		sql:  sql,
		args: args,
	}
}

func (r *RawQuerier[T]) Build() (*Query, error) {
	return &Query{
		SQL:  r.sql,
		Args: r.args,
	}, nil
}

func (r *RawQuerier[T]) Get(ctx context.Context) (*T, error) {
	//  当通过 RawQuery 方法调用 Get ,如果 T 是 time.Time, sql.Scanner 的实现，
	//  内置类型或者基本类型时， 在这里都会报错，但是这种情况我们认为是可以接受的
	//  所以在此将报错忽略，因为基本类型取值用不到 meta 里的数据
	model, _ := r.r.Get(new(T))
	res := get[T](ctx, r.core, r.sess, &QueryContext{
		Builder: r,
		Type:    "RAW",
		Meta:    model,
	})
	if res.Err != nil {
		return nil, res.Err
	}
	return res.Result.(*T), nil
}

func (r *RawQuerier[T]) GetMulti(ctx context.Context) ([]*T, error) {
	//  当通过 RawQuery 方法调用 Get ,如果 T 是 time.Time, sql.Scanner 的实现，
	//  内置类型或者基本类型时， 在这里都会报错，但是这种情况我们认为是可以接受的
	//  所以在此将报错忽略，因为基本类型取值用不到 meta 里的数据
	model, _ := r.r.Get(new(T))
	res := getMulti[T](ctx, r.core, r.sess, &QueryContext{
		Builder: r,
		Type:    "RAW",
		Meta:    model,
	})
	if res.Err != nil {
		return nil, res.Err
	}
	return res.Result.([]*T), nil
}

func (r *RawQuerier[T]) Exec(ctx context.Context) Result {
	return exec[T](ctx, r.core, r.sess, &QueryContext{
		Type:    "RAW",
		Builder: r,
	})
}

单元测试

func TestRawQuerier_Get(t *testing.T) {
	//mockDB, mock, err := sqlmock.New()
	mockDB, mock, err := sqlmock.New(
		sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
	if err != nil {
		t.Fatal(err)
	}
	defer func() { _ = mockDB.Close() }()
	db, err := OpenDB("mysql", mockDB)
	if err != nil {
		t.Fatal(err)
	}

	testCases := []struct {
		name      string
		queryRes  func(t *testing.T) any
		mockErr   error
		mockOrder func(mock sqlmock.Sqlmock)
		wantErr   error
		wantVal   any
	}{
		//返回原生基本类型
		{
			name: "res RawQuery int",
			queryRes: func(t *testing.T) any {
				queryer := RawQuery[int](db, "SELECT `age` FROM `test_model` LIMIT ?;", 1)
				result, err := queryer.Get(context.Background())
				require.NoError(t, err)
				return result
			},
			mockOrder: func(mock sqlmock.Sqlmock) {
				rows := mock.NewRows([]string{"age"}).AddRow(10)
				mock.ExpectQuery("SELECT `age` FROM `test_model` LIMIT ?;").
					WithArgs(1).
					WillReturnRows(rows)
			},
			wantVal: func() *int {
				val := 10
				return &val
			}(),
		},
		{
			name: "res RawQuery bytes",
			queryRes: func(t *testing.T) any {
				queryer := RawQuery[[]byte](db, "SELECT `first_name` FROM `test_model` WHERE `id`=? LIMIT ?;", 1, 1)
				result, err := queryer.Get(context.Background())
				require.NoError(t, err)
				return result
			},
			mockOrder: func(mock sqlmock.Sqlmock) {
				rows := mock.NewRows([]string{"first_name"}).AddRow([]byte("Li"))
				mock.ExpectQuery("SELECT `first_name` FROM `test_model` WHERE `id`=? LIMIT ?;").
					WithArgs(1, 1).
					WillReturnRows(rows)
			},
			wantVal: func() *[]byte {
				val := []byte("Li")
				return &val
			}(),
		},
		{
			name: "res RawQuery string",
			queryRes: func(t *testing.T) any {
				queryer := RawQuery[string](db, "SELECT `first_name` FROM `test_model` WHERE `id`=? LIMIT ?;", 1, 1)
				result, err := queryer.Get(context.Background())
				require.NoError(t, err)
				return result
			},
			mockOrder: func(mock sqlmock.Sqlmock) {
				rows := mock.NewRows([]string{"first_name"}).AddRow("Da")
				mock.ExpectQuery("SELECT `first_name` FROM `test_model` WHERE `id`=? LIMIT ?;").
					WithArgs(1, 1).
					WillReturnRows(rows)
			},
			wantVal: func() *string {
				val := "Da"
				return &val
			}(),
		},
		{
			name: "res RawQuery struct ptr",
			queryRes: func(t *testing.T) any {
				queryer := RawQuery[TestModel](db, "SELECT `first_name`,`age` FROM `test_model` WHERE `id`=? LIMIT ?;", 1, 1)
				result, err := queryer.Get(context.Background())
				require.NoError(t, err)
				return result
			},
			mockOrder: func(mock sqlmock.Sqlmock) {
				rows := mock.NewRows([]string{"first_name", "age"}).AddRow("Da", 18)
				mock.ExpectQuery("SELECT `first_name`,`age` FROM `test_model` WHERE `id`=? LIMIT ?;").
					WithArgs(1, 1).
					WillReturnRows(rows)
			},
			wantVal: func() *TestModel {
				return &TestModel{
					FirstName: "Da",
					Age:       18,
				}
			}(),
		},
		{
			name: "res RawQuery sql.NullString",
			queryRes: func(t *testing.T) any {
				queryer := RawQuery[sql.NullString](db, "SELECT `last_name` FROM `test_model` WHERE `id`=? LIMIT ?;", 1, 1)
				result, err := queryer.Get(context.Background())
				require.NoError(t, err)
				return result
			},
			mockOrder: func(mock sqlmock.Sqlmock) {
				rows := mock.NewRows([]string{"last_name"}).AddRow([]byte("ming"))
				mock.ExpectQuery("SELECT `last_name` FROM `test_model` WHERE `id`=? LIMIT ?;").
					WithArgs(1, 1).
					WillReturnRows(rows)
			},
			wantVal: func() *sql.NullString {
				return &sql.NullString{String: "ming", Valid: true}
			}(),
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			tc.mockOrder(mock)
			res := tc.queryRes(t)
			assert.Equal(t, tc.wantVal, res)
		})
	}
}
vicent_uuid4
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Outlier Detection for Improved Data Quality and Diversity in Dialog Systems-学习笔记
ChanYeol666的博客
10-12 436
Outlier Detection for Improved Data Quality and Diversity in Dialog Systems 论文按如下方式检测数据集中的异常值： 1.生成每个实例的矢量表示。 2.平均向量以获得均值表示。 3.计算每个实例与平均值的距离。 4.按距离升序排列。 5.（删除列表，仅将前k％作为离群值。） 最后一步用括号括起来，因为在实践中使用动态阈值方法，允许...
PHP mysqli 原生查询
02-22
两种方式 写法
Android之采用execSQL与rawQuery方法完成数据的添删改查操作详解
01-05
使用 SQLiteDatabase 操作 SQLite 数据库 代码如下:/* Android提供了一个名为SQLiteDatabase的类，该类封装了一些操作数据库的API，使用该类可以完成对数据进行添加(Create)、查询(Retrieve)、更新(Update)和删除(Delete)操作（这些操作简称为CRUD）。对SQLiteDatabase的学习，我们应该重点掌握execSQL()和rawQuery()方法。 execSQL()方法可以执行insert、delete、update和CREATE TABLE之类有更改行为的SQL语句； rawQuery()方法用于执行select语句
rawQuery()方法的使用
上帝派来的逗比
09-02 6130
SQLiteDatabase的rawQuery()用于执行select语句，使用例子如下： SQLiteDatabase db= ....; Cursor cursor = db.rawQuery("select * from person",null); ... cursor.close(); db.close(); rawQuery()方法的第一个参数为select语句；第二个参数为s
query和rawQuery的区别
today_work的博客
01-08 5528
12px;”>Cursor cursor = db.rawQuery(“select name from *** where id=?”, new String[]{“1”}); Cursor cursor = db.query(”***”, new String[]{“name”}, “id=?”, new String[]{“1”}, null, null, null); 上面是两个
数据库的增删查改--Cursor结果集和execSQL和rawQuery方法的使用--SQLiteDatabase类的getWritableDatabase和getReadableDatabase方法
qq_45491681的博客
02-02 2946
数据库的增删查改--Cursor结果集的execSQL和execSQL方法的使用--SQLiteDatabase类的getWritableDatabase和getReadableDatabase方法
Laravel 使用查询构造器配合原生sql语句查询的例子
10-16
然而，在处理某些复杂查询场景时，如多表联合查询、子查询或自定义SQL逻辑，可能需要结合使用查询构造器与原生SQL语句。下面我们将详细探讨如何在Laravel中结合两者进行查询。 首先，让我们了解一下Laravel查询构造...
原生js实现查询天气小应用
10-20
主要为大家详细介绍了原生js实现查询天气的小应用,具有一定的参考价值，感兴趣的小伙伴们可以参考一下
云原生docker实战
最新发布
07-17
云原生docker实战
云原生架构概述
02-24
在讲云原生之前，我们先了解一下CNCF，即云原生计算基金会，2015年由谷歌牵头成立，基金会成员目前已有一百多企业与机构，包括亚马逊、微软。思科等巨头。目前CNCF所托管的应用已达14个，下图为其公布的...
rawQuery的使用
热门推荐
Dijkstra的专栏
08-25 1万+
File databaseFile = getDatabasePath(Contents.EnMicroMsg_sql_test); Log.d("wwwww", databaseFile.toString()); SQLiteDatabase db = SQLiteDatabase.openOrCreateDatabase(databaseFile, uin, null, h
Android心得4.3--SQLite数据库--execSQL()和rawQuery()方法
追梦_1394270543的博客
06-01 745
1.   除了可以使用文件或SharedPreferences存储数据，还可以选择使用SQLite数据库存储数据。 在Android平台上，集成了一个嵌入式关系型数据库—SQLite，SQLite3支持 NULL、INTEGER、REAL（浮点数字）、TEXT(字符串文本)和BL...
androidrawquery_Android之采用execSQL与rawQuery方法完成数据的添删改查操作详解
weixin_39862794的博客
12-23 813
使用 SQLiteDatabase 操作 SQLite 数据库/*Android提供了一个名为SQLiteDatabase的类，该类封装了一些操作数据库的API，使用该类可以完成对数据进行添加(Create)、查询(Retrieve)、更新(Update)和删除(Delete)操作(这些操作简称为CRUD)。对SQLiteDatabase的学习，我们应该重点掌握execSQL()和rawQuery...
sqlite之query、rawQuery区别；moveToNext，moveToFirst区别
火星男孩的分享空间
03-01 4563
query public Cursor query(boolean distinct, String table, String[] columns, String selection, String[] selectionArgs, String groupBy, String having, String orderBy, String l...
rawquery 没扎到返回什么_php – Laravel SQL查询返回错误的结果
weixin_31533357的博客
02-21 96
我有这个带有Laravel Query构建器的SQL查询$this->candidate->select('candidates.*', 'candidates.id as candidate_id', 'mj.interviewScheduledDate')->join('candidates_positions', 'candidates_positions.candidat...
For @RawQuery, you should specify the list of tables to be observed via the observedEntities field.
Cherish
02-09 358
Observable query return type (LiveData, Flowable, DataSource, DataSourceFactory etc) can only be used with SELECT queries that directly or indirectly (via @Relation, for example) access at least one table. For @RawQuery, you should specify the list of tab.
Golang gorm 快速开始、保存、查询单条数据、查询全部数据
wgchen
01-08 2399
Golang gorm 快速开始、保存、查询单条数据、查询全部数据
Room 数据库快速入门
m0_60396106的博客
07-22 719
被无穷无尽的 SQLite 样板代码折磨后，发现 Google 的 Room 封装似乎是个不错的选择，不过大量运用注释，使得不熟悉 SQLite 的开发者如果一开始就直接上手 Room，会不知其所以然，这里记录下我的学习笔记。 1. 回顾直接操作 SQLite 我本人在操作数据库数据库时有个习惯，就是不使用封装的api直接写SQL。因为大多数操作数据库的api实在是过于繁琐。如果直接写SQL来写入数据库，只需要使用到 SQLiteOpenHelper， SQLiteDatabase，Cursor 三个类和
使用MYSQL原生语言查询
05-31
好的，以下是使用 MySQL 原生语言查询的示例： 假设您的积分变动历史表名为 `points_history`，包含字段有 `user_id`、`change_value`、`change_time` 和 `change_type`。 查询某个用户的剩余积分可以使用以下 SQL 语句： ``` SELECT SUM(CASE WHEN change_type = '增加' THEN change_value ELSE 0 END) - SUM(CASE WHEN change_type = '减少' THEN change_value ELSE 0 END) AS remaining_points FROM points_history WHERE user_id = '用户ID'; ``` 其中，`CASE WHEN` 语句用于根据变动类型对变动值进行分类求和，`SUM` 函数用于对分类求和的结果进行求和，`AS` 语句用于给结果取别名。 您需要将 `用户ID` 替换为实际的用户ID值。如果有多个用户需要查询，可以在 `WHERE` 子句中使用 `IN` 操作符指定多个用户ID值。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值