Benchmark Foreign Key

benchmark_foreign_key.go

package main

import (
	"testing"

	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/mysql"
)

type Post struct {
	ID       int64
	AuthorID int64
}

type Author struct {
	ID int64
}

type ForeignKeyPost struct {
	ID       int64
	AuthorID int64
}

var conn *gorm.DB

func init() {
	var err error
	conn, err = gorm.Open("mysql", "root:@tcp(localhost:3306)/fk_dev?parseTime=true")
	if err != nil {
		panic(err)
	}

	conn.AutoMigrate(&Post{}).
		AutoMigrate(&Author{}).
		AutoMigrate(&ForeignKeyPost{})

	conn.Model(&ForeignKeyPost{}).AddForeignKey("author_id", "authors(id)", "RESTRICT", "RESTRICT")

	conn.Create(&Author{ID: 1})
}

func BenchmarkBaseline(b *testing.B) {
	for i := 0; i < b.N; i++ {
		if err := conn.Create(&Post{AuthorID: 1}).Error; err != nil {
			panic(err)
		}
	}
}

func BenchmarkForeignKey(b *testing.B) {
	for i := 0; i < b.N; i++ {
		if err := conn.Create(&ForeignKeyPost{AuthorID: 1}).Error; err != nil {
			panic(err)
		}
	}
}

BenchmarkBaseline-8     	    3770	    309503 ns/op
BenchmarkForeignKey-8   	    3331	    317162 ns/op

BenchmarkBaseline-8     	    3192	    315506 ns/op
BenchmarkForeignKey-8   	    3381	    315577 ns/op

BenchmarkBaseline-8     	    3298	    312761 ns/op
BenchmarkForeignKey-8   	    3829	    345342 ns/op

BenchmarkBaseline-8     	    3753	    291642 ns/op
BenchmarkForeignKey-8   	    3948	    325239 ns/op