Golang DB 依存性の注入

Question

Golangを独学で学習しているものです。
ぜひお力を貸して頂けると幸いです。

● 現在の状況
golangでDIを実装したい（モックテストを実現するため)

● 困っていること
参考書などを引用してコードを実装↓
gormを使用しており、DIするためにハンドラーにdbを渡した。

main.go

func signupHandler(user models.UserService) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var u models.User
		json.NewDecoder(r.Body).Decode(&u)

		name := u.UserName
		email := u.Email
		pass := u.Password

		if name == "" {
			apiError(w, "UserName is required", http.StatusBadRequest)
			return
		}
		if email == "" {
			apiError(w, "Email is required", http.StatusBadRequest)
			return
		}
		if pass == "" {
			apiError(w, "Password is required", http.StatusBadRequest)
			return
		}

		if err := user.CreateUser(name, email, pass); err != nil {
			apiError(w, "username or email are duplicated", http.StatusConflict)
			return
		}

		apiError(w, "success", http.StatusCreated)
		return
	}
}

func userDeleteHandler(user models.UserService) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var u models.User
		json.NewDecoder(r.Body).Decode(&u)

		id, err := strconv.Atoi(path.Base(r.URL.Path))
		if err != nil {
			apiError(w, "cloud not find user", http.StatusNotFound)
			return
		}
		err = user.DeleteUser(id, u.Password)
		if err != nil {
			apiError(w, err.Error(), http.StatusBadRequest)
			return
		}
		apiError(w, "success", http.StatusOK)
		return
	}
}

func StartWebServer() error {
	db, err := db.SQLConnect()
	if err != nil {
		panic(err.Error())
	}
	defer db.Close()

	r := mux.NewRouter()
	http.Handle("/", r)

	// users
	r.HandleFunc("/users", signupHandler(&models.User{DB: db})).Methods("POST")
	r.HandleFunc("/users/{id:[0-9]+}", userDeleteHandler(&models.User{DB: db})).Methods("DELETE")

	return http.ListenAndServe(fmt.Sprintf(":%d", config.Config.Port), nil)
}

type UserService interface {
	CreateUser(name string, email string, pass string) (err error)
	DeleteUser(id int, pass string) (err error)
}

// User 構造体
type User struct {
	DB        *gorm.DB
	ID        int       `json:"id,omitempty" gorm:"primaryKey,unique"`
	UserName  string    `json:"user_name,omitempty" gorm:"unique"`
	Email     string    `json:"email,omitempty" gorm:"unique"`
	Password  string    `json:"password,omitempty"`
	CreatedAt time.Time `json:"created_at,omitempty"`
	UpdatedAt time.Time `json:"updated_at,omitempty"`
}

// CreateUser user登録
func (user *User) CreateUser(name, email, pass string) (err error) {
	user.ID = 0 //gorm auto increment
	user.UserName = name
	user.Email = email
	user.CreatedAt = time.Now()
	user.UpdatedAt = time.Now()

	hash, err := bcrypt.GenerateFromPassword([]byte(pass), 10)
	if err != nil {
		log.Fatal(err)
	}
	user.Password = string(hash)

	if err := user.DB.Create(&user).Error; err != nil {
		return err
	}

	return
}

func (user *User) DeleteUser(id int, pass string) (err error) {
	if err := user.DB.First(&user, id).Error; err != nil {
		return err
	}

	if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(pass)); err != nil {
		return err
	}

	if err := user.DB.Delete(&user).Error; err != nil {
		return err
	}
	return nil
}

エラー

apiを叩くと1度しか成功しない。
1度目に叩い情報が構造体に残ったままになってしまう。

質問

apiを叩くと1回目の処理は成功するが、2回目以降の処理は1度目に叩いた情報が残っているので毎回初期化しないといけないのでしょうか？
CreateUserメソッドではIDに0を渡すことで無理やりIDを初期化しております。
ID = 0　を書かなければ user.ID duplicateのエラーが出てしまいます。

DIしない場合はSQL処理の度にdb.Close()しているため毎回初期化されると予想しております。

箇条書きで羅列してしまいましたが、以上でぜひお力をお借りしたいです>_<
どうぞよろしくお願いいたします！

Answer 1

gormの「モデル定義の構造体型」はデータベースの１レコードごとに「インスタンス化と破棄」されることを前提とした作りになっています。ひとつのインスタンスを持ち回って使い回すことは想定されていません。
なので、モデルにDBインスタンスをもたせるのもミスマッチだと思います。

Comments

[Shika_Tech]

nobonobo様
ご指摘頂きありがとうございます。
> gormの「モデル定義の構造体型」はデータベースの１レコードごとに「インスタンス化と破棄」
> モデルにDBインスタンスをもたせるのもミスマッチ
なるほどですね。ここは標準のパッケージで書き直すか、コンストラクタインジェクション?
という方法で実装するのが良いでしょうか？