Go言語のgormでORMしてみた（単純CUD編）

前提

• Go言語そのものに慣れていない
• MySQLはVagrantで適当にCentOS6.6に構築
• 単純検索ぐらいはできるようになった

今回のTODO

• 登録・更新・削除できるようになる
• トランザクション制御ができるようになる
• 登録時のPKを取得する

いざ実践！

• 概略は理解した（つもり）なので、サンプルを見ながら実装
  • Gorm#Transactions
  • Gorm#Create
  • Gorm#Update

問題点

• 登録時のPKを取得数方法がわからない
  • tx.Create の戻り値の中に Value があるが、interface{} でリフレクションしたくない

解決策

• 検索結果をstructに変換するメソッドで解決できた
  • Gorm#Scan

db_connection2.go

// DBアクセスsample2.go
package main

import (
	"fmt"

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

type User struct {
	ID    uint64 `gorm:"primary_key"`
	Email string `gorm:"type:varchar(255);"`
	Name  string `gorm:"type:varchar(45);"`
}

func main() {
	db, err := gorm.Open("mysql", "testuser:testuser@/go_sample?charset=utf8&parseTime=True&loc=Local")
	if err != nil {
		panic("failed to connect database")
	}
	// テーブル名が複数系でない場合、これを指定すること
	db.SingularTable(true)

	// 実行ログ出力
	//	db.LogMode(true)

	// 最新のIDを取得してインサートすると見せかけてロールバックする
	var user User
	var id uint64 = lastPrint(db, user)

	tx := db.Begin()

	var sid = fmt.Sprint(id)
	user = User{Email: "sample@" + sid + "test.com", Name: "Name" + sid}
	savePrint(tx, user)

	tx.Rollback()

	// 更新されていないことを確認
	lastPrint(db, user)

	tx = db.Begin()
	id, err = savePrint(tx, user)
	// 本来はエラーならロールバック
	tx.Commit()

	// 更新されていることを確認
	lastPrint(db, user)

	// 更新メソッドが実行されることを確認
	tx = db.Begin()
	user.ID = id
	savePrint(tx, user)
	tx.Commit()

	// レコードが追加されていないことを確認
	lastPrint(db, user)

	// 削除されることを確認
	tx = db.Begin()
	tx.Delete(user)
	tx.Commit()

	// 削除されていることを確認
	user.ID = uint64(0)
	lastPrint(db, user)
}

func lastPrint(db *gorm.DB, user User) uint64 {
	var count int
	db.Last(&user).Count(&count)
	if count == 0 {
		fmt.Println("該当レコードなし")
	} else {
		fmt.Println("id:" + fmt.Sprint(user.ID))
	}
	return user.ID
}

// 本当は Save を使えば事足りるはず
func savePrint(tx *gorm.DB, user User) (uint64, error) {
	// 戻り値用の変数定義
	var rId uint64 = 0

	// PKを設定しないと true を返すらしい
	if tx.NewRecord(user) {
		fmt.Println("\t新規レコードです。")
		result := tx.Create(&user)
		if result.Error != nil {
			return rId, result.Error
		}

		// 追加したIDが欲しい場合、Scanで元の形に変換する
		result.Scan(&user)
		sid := fmt.Sprint(user.ID)
		fmt.Println("\t新規追加したID:" + sid)
		rId = user.ID
	} else {
		fmt.Println("\t更新レコードです。")
		rId = user.ID
		result := tx.Model(&user).Update(&user)
		if result.Error != nil {
			return rId, result.Error
		}
		sid := fmt.Sprint(user.ID)
		fmt.Println("\t更新したID:" + sid)
	}
	return rId, nil
}

残課題

• 件数をわざわざ取得しないで存在しないチェックを行いたい
  • ID == 0 でも良いけれど、美しくない
• 検索条件の初期化の際に ID = 0 以外の方法を探す

TODO

• 複雑な検索結果の取得
• リレーションの貼り方、概念的な整理
• ソースの自動生成