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

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