gorpでサーバー起動時に自動でCREATE DADABASEする

gorpを使ってて、typeで各テーブルの定義をしておくと、こちらでCREATE TABLEしなくても、自動でやってくれる仕組みがあって感動していたのですが、どうやらDBはつくってくれない感じだったので、何も考えなくていいように最初にDBもつくるようにしました。

main.go

package main

func main() {
	// initialize the DbMap
	dbmap := initDb()
	defer dbmap.Db.Close()
	
	// なにかしらの処理…

次にinitDbを定義してるスクリプト。

自分はめんどくさいのでmain.goの隣に置いてます。

sql.Openの部分でまずはDB名を指定せずに接続し、クリエイトのためのSQLを流し、そのあとDB名を指定して接続しにいってます。あとは標準的なgorpの使い方になっているかと思います。

db.go

package main

import (
	"database/sql"
	"github.com/coopernurse/gorp"
	_ "github.com/go-sql-driver/mysql"
	"log"
)

type PostInfo struct {
	Title       string
	Body        string
}

type Post struct {
	Id int64
	PostInfo
	CreatedAt int64
	UpdatedAt int64
}

func checkErr(err error, msg string) {
	if err != nil {
		log.Fatalln(msg, err)
	}
}

func initDb() *gorp.DbMap {
	// connect mysql
	db, err := sql.Open("mysql", "user_name:pass_word@/")
	checkErr(err, "sql.Open failed")
	dbmap := &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}

	// create db if not exists
	_, err = dbmap.Exec("CREATE DATABASE IF NOT EXISTS your_project_name DEFAULT CHARACTER SET utf8;")
	checkErr(err, "create db failed")

	// connect db
	db, err = sql.Open("mysql", "user_name:pass_word@/your_project_name")
	checkErr(err, "sql.Open failed")
	dbmap = &gorp.DbMap{Db: db, Dialect: gorp.MySQLDialect{"InnoDB", "UTF8"}}

	// add table
	dbmap.AddTableWithName(Post{}, "posts").SetKeys(true, "Id")
	
	// create the table
	err = dbmap.CreateTablesIfNotExists()
	checkErr(err, "Create tables failed")

	return dbmap
}

余談

最近ずっとRailsでActiverecord使っていたので、gorpの中途半端なORM感がちょっと微妙だなぁって思ってました。

探したらgormが使いやすそうです。

乗り換えようかな。。。