Skip to content

標準ライブラリでデータベースに接続

データベース(今回はMariaDB)に接続します。

事前準備

MariaDB向けのGoドライバをインストールします:

shell
go get -u github.com/go-sql-driver/mysql

下記DDLでデータベースを用意しておきます:

sql
-- データベース作成
CREATE DATABASE snaildb;

USE snaildb;

-- テーブル作成
CREATE TABLE users(
    id INT PRIMARY KEY AUTO_INCREMENT
    , name VARCHAR(20) NOT NULL
    , age INT NOT NULL
);

サンプルコード

go
package main

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	// DB接続
	db, err := connectDB()
	if err != nil {
		log.Fatal("DB access failed")
	}

	// ユーザ登録実行
	insert(db, users{name: "nob", age: 13})

	// 複数ユーザ検索実行
	fmt.Println(findByAge(db))

	// 単一ユーザ検索実行
	fmt.Println(findById(db))
}

// データベースに接続します。
func connectDB() (*sql.DB, error) {

	// ユーザー名:パスワード@tcp(ホスト:ポート)/データベース名
	dsn := "root:password@tcp(localhost:3306)/snaildb"
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		return nil, err
	}

	// 実際に接続できるかを確認
	err = db.Ping()
	if err != nil {
		return nil, err
	}

	return db, nil
}

// ユーザを登録します。
func insert(db *sql.DB, u users) {

	// クエリ実行
	_, err := db.Exec("INSERT INTO users (name, age) VALUES (?, ?)", u.name, u.age)
	if err != nil {
		log.Fatal(err)
	}
}

// 30歳未満のユーザ(複数)を検索します。
func findByAge(db *sql.DB) []users {

	rows, err := db.Query("SELECT * FROM users WHERE age < ?", 30) // 30歳未満を抽出
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	var usersList []users
	for rows.Next() {
		var id int
		var name string
		var age int
		if err := rows.Scan(&id, &name, &age); err != nil {
			log.Fatal(err)
		}
		u := users{id: id, name: name, age: age}
		usersList = append(usersList, u)
	}

	return usersList
}

// ID=1のユーザ(単一)を検索します。
func findById(db *sql.DB) users {

	row := db.QueryRow("SELECT * FROM users WHERE id = ?", 1) // ID=1を抽出
	var id int
	var name string
	var age int
	row.Scan(&id, &name, &age)

	return users{id, name, age}
}

// usersテーブル向けのモデルです。
type users struct {
	id   int    // ID
	name string // 名前
	age  int    // 年齢
}