Goでルーティング

DBと接続できたのでルーティングしてみる。

/と/userにクエリをつけて検索するだけの簡単なもの。

package main

import (
	"fmt"
	"net/http"
	"net/url"
	"os"

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

func main() {

	initDB()

	http.HandleFunc("/", hello)
	http.HandleFunc("/user", handleUser)
	fmt.Println("listnening...")
	http.ListenAndServe(":"+os.Getenv("PORT"), nil)
}

func hello(res http.ResponseWriter, req *http.Request) {
	fmt.Fprintln(res, "Hello!!")
}
func handleUser(res http.ResponseWriter, req *http.Request) {
	m, _ := url.ParseQuery(req.URL.RawQuery)
	user := loadUser(m["user_name"][0])
	fmt.Fprintln(res, user.Name)
}

func loadUser(s string) User {
	databaseURL := os.Getenv("DATABASE_URL")
	db, err := gorm.Open("postgres", databaseURL)
	if err != nil {
		fmt.Println(err)
		panic("failed to connect database")
	}

	var user User
	db.Where("name = ?", s).First(&user)
	return user
}

func initDB() {
	databaseURL := os.Getenv("DATABASE_URL")
	db, err := gorm.Open("postgres", databaseURL)
	if err != nil {
		fmt.Println(err)
		panic("failed to connect database")
	}

	db.AutoMigrate(&User{})
	db.Create(&User{Name: "Motoki Narita"})
	db.Create(&User{Name: "Swift"})
	db.Create(&User{Name: "Gopher"})
}

type User struct {
	gorm.Model
	Name string
}

https://hogehoge.com/user?user_name=GopherでGopherとか表示されるようになった。単純なことしかできなかった。

とりあえずpostgresの環境がherokuにしかないので、次はローカルでpostgresの環境構築してデバッグが捗るようにする。

たぶんPG Commanderとか使う。