使用Go语言ORM库worm的SQL预处理功能

go lang

已于 2022-03-28 13:21:24 修改

阅读量785

点赞数

分类专栏： go orm web框架文章标签： golang 数据库 postgresql mysql

于 2022-03-27 21:57:03 首次发布

本文链接：https://blog.csdn.net/gz_hm/article/details/123782743

版权

go 同时被 3 个专栏收录

34 篇文章 1 订阅

订阅专栏

web框架

23 篇文章 1 订阅

订阅专栏

orm

13 篇文章 0 订阅

订阅专栏

worm支持SQL语句的预编译，使用SQL语句预编译可以提升数据库访问的效率。在worm中可以通过三种方式开启SQL语句预编译：全局开启、会话中开启、语句中开启。

开启全局SQL预编译

package main

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql"
	log "github.com/haming123/wego/dlog"
	"github.com/haming123/wego/worm"
)

func main() {
	dbcnn, err := initMySql("account:pwd@tcp(127.0.0.1:3306)/db?charset=utf8&parseTime=True")
	if err != nil {
		log.Error(err)
	}
	defer dbcnn.Close()
	err = worm.InitMysql(dbcnn)
	if err != nil {
		log.Error(err)
		return
	}
	worm.ShowSqlLog(true)
	//开启SQL语句预编译
	worm.UsePrepare(true)
	//设置预编译stmt缓存的最大数量
	worm.SetMaxStmtCacheNum(1000)
}

会话中开启SQL预编译

func demoPrepareSession() {
	session := worm.NewSession()
	session.UsePrepare(true)

	var user = User{Name:"name1", Age: 21, Created: time.Now()}
	id, err := worm.Model(&user).Insert()
	if err != nil{
		log.Error(err)
		return
	}

	_, err = worm.Model(&User{}).ID(id).Delete()
	if err != nil{
		log.Error(err)
		return
	}
}

语句中开启SQL预编译

func demoModelPrepare() {
	var user = User{Name:"name1", Age: 21, Created: time.Now()}
	_, err := worm.Model(&user).UsePrepare(true).Select("name", "age").ID(1).Update()
	if err != nil{
		log.Error(err)
		return
	}
}

SQL预编译的性能测试

worm.Find测试脚本

func BenchmarkModelFind(b *testing.B) {
	dbcnn, _ := initMySql(dsn_test)
	InitEngine(&dialectMysql{}, dbcnn)
	ShowSqlLog(false)
	b.StopTimer()

	b.StartTimer()
	for i := 0; i < b.N; i++ {
		var arr []User
		err := Model(&User{}).Where("id>? and name is not null", 0).Select("id", "name", "age").Find(&arr)
		if err != nil{
			b.Error(err)
			return
		}
	}
	b.StopTimer()
}

worm.Find测试脚本(开启预处理)

func BenchmarkModelFindWithCache(b *testing.B) {
	dbcnn, _ := initMySql(dsn_test)
	InitEngine(&dialectMysql{}, dbcnn)
	ShowSqlLog(false)
	UsePrepare(true)
	b.StopTimer()

	b.StartTimer()
	for i := 0; i < b.N; i++ {
		var arr []User
		err := Model(&User{}).Where("id>? and name is not null", 0).Select("id", "name", "age").Find(&arr)
		if err != nil{
			b.Error(err)
			return
		}
	}
	b.StopTimer()
}

测试结果

pkg: worm
BenchmarkModelFind-2   	    2562	    466247 ns/op	    5451 B/op	      79 allocs/op
pkg: worm(开启预处理)
BenchmarkFindWithCache-2    4662	    258889 ns/op	    5330 B/op	      76 allocs/op