vlambda博客
学习文章列表

Go语言使用gorm对MySQL进行性能测试

之前写过了,其中对gorm框架在操作MySQL的各种基础实践,下面分享一下如何使用gorm框架对MySQL直接进行性能测试的简单实践。

框架支持

这里我使用了一个原始的Go语言版本的FunTester测试框架,现在只有一个基本的方法,实在是因为Go语言特性太强了。框架设计的主要思路之一就是利用Go语言的闭包和方法参数特性,将一个func()当做性能测试的主题,通过不断运行这个func()来实现性能测试。当然还有另外一个思路就是运行一个多线程任务类,类似Java版本的com.funtester.base.constaint.ThreadBase抽象类,这样可以设置一些类的属性,绑定一些测试资源,适配更多的测试场景。

// ExecuteRoutineTimes
// @Description: FunTester性能测试执行框架
// @param fun 待执行方法
// @param times 次数
// @param thread 线程数
func ExecuteRoutineTimes(fun func()tr int) {
 c := make(chan int//确认所有线程都结束
 key := false        //用于控制所有线程一起结束
 start := futil.Milli()
 for i := 0; i < r; i++ {
  go func() {
   sum := 0
   for i := 0; i < t; i++ {
    if key {
     break
    }
    fun()
    sum++
   }
   key = true
   c <- sum
  }()
 }
 total := 0
 for i := 0; i < r; i++ {
  num := <-c
  total += num
 }
 end := futil.Milli()
 diff := end - start
 //total := thread * times
 log.Printf("总耗时: %f"float64(diff)/1000)

 log.Printf("请求总数: %d", total)
 log.Printf("QPS: %f"float64(total)/float64(diff)*1000.0)
}

select

下面演示select的性能测试,这里我用了随机ID查询的场景。

func TestSelectP(t *testing.T)  {
 execute.ExecuteRoutineTimes(func() {
  var f Funtester
  drive.Where("id = ?", futil.RangInt(3520000)).First(&f)
 },1000,100)
}

delete

这里我使用从35开始递增的ID进行删除。

func TestDeleteP(t *testing.T) {
 var index int32 = 35
 execute.ExecuteRoutineTimes(func() {
  id := atomic.AddInt32(&index, 1)
  drive.Where("id = ?", id).Delete(&Funtester{})
 },1000,100)
}

update

这里使用了select的用例部分,随机ID,然后更新name字段,随机10个长度的字符串。

func TestUpdateP(t *testing.T) {
 execute.ExecuteRoutineTimes(func() {
  drive.Where("id = ?",futil.RangInt(3520000)).Update("name",futil.RandomStr(10))
 },1000,100)
}

inset

这里用到了FunTester字段都是随机生成。

func TestInsertP(t *testing.T) {
 execute.ExecuteRoutineTimes(func() {
  drive.Create(&Funtester{Name: futil.RandomStr(10),Age: futil.RandomInt(100)})
 },1000,100)
}

FunTester构造方法

type Funtester struct {
 gorm.Model
 Name string
 Age  int
}

到这里可以看出,性能测试框架用到的都是gorm框架的基础API使用,这里MySQL连接池的管理工作完全交给了gorm框架完成,看资料说非常牛逼,我们只需要设置几个参数。这个使用体现很像HttpClient设置HTTP连接池类似,这里我们也可以看出这些优秀的框架使用起来都是非常简单的。

PS:关于gorm的基础使用的请参考上一期的文章。

Have Fun ~ Tester !