Add login

2020-01-05 23:20:38 +01:00
parent dc396a300f
commit c0926ffea0
35 changed files with 1647 additions and 39 deletions
--- a/storage/dao/postgresql/database_postgresql_user.go
+++ b/storage/dao/postgresql/database_postgresql_user.go
@@ -0,0 +1,119 @@
+package postgresql
+
+import (
+	"database/sql"
+	"hamster-tycoon/storage/dao"
+	"hamster-tycoon/storage/model"
+
+	"github.com/lib/pq"
+)
+
+func (db *DatabasePostgreSQL) GetAllUsers() ([]*model.User, error) {
+	q := `
+		SELECT u.id, u.email, u.created_at, u.updated_at, u.nickname
+		FROM public.user u
+	`
+	rows, err := db.session.Query(q)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	us := make([]*model.User, 0)
+	for rows.Next() {
+		u := model.User{}
+		err := rows.Scan(&u.ID, &u.Email, &u.CreatedAt, &u.UpdatedAt, &u.Nickname)
+		if err != nil {
+			return nil, err
+		}
+		us = append(us, &u)
+	}
+	return us, nil
+}
+
+func (db *DatabasePostgreSQL) GetUsersByID(id string) (*model.User, error) {
+	q := `
+		SELECT u.id, u.email, u.created_at, u.updated_at, u.nickname
+		FROM public.user u
+		WHERE u.id = $1
+	`
+	row := db.session.QueryRow(q, id)
+
+	u := model.User{}
+	err := row.Scan(&u.ID, &u.Email, &u.CreatedAt, &u.UpdatedAt, &u.Nickname)
+	if errPq, ok := err.(*pq.Error); ok {
+		return nil, handlePgError(errPq)
+	}
+	if err == sql.ErrNoRows {
+		return nil, dao.NewDAOError(dao.ErrTypeNotFound, err)
+	}
+	return &u, err
+}
+
+func (db *DatabasePostgreSQL) CreateUser(user *model.User) error {
+	q := `
+		INSERT INTO public.user
+			(email, nickname, google_id)
+		VALUES
+			($1, $2, $3)
+		RETURNING id, created_at
+	`
+
+	err := db.session.
+		QueryRow(q, user.Email, user.Nickname, user.GoogleId).
+		Scan(&user.ID, &user.CreatedAt)
+	if errPq, ok := err.(*pq.Error); ok {
+		return handlePgError(errPq)
+	}
+	return err
+}
+
+func (db *DatabasePostgreSQL) DeleteUser(id string) error {
+	q := `
+		DELETE FROM public.user
+		WHERE id = $1
+	`
+
+	_, err := db.session.Exec(q, id)
+	if errPq, ok := err.(*pq.Error); ok {
+		return handlePgError(errPq)
+	}
+	return err
+}
+
+func (db *DatabasePostgreSQL) UpdateUser(user *model.User) error {
+	q := `
+		UPDATE public.user
+		SET
+			email = $2
+		WHERE id = $1
+		RETURNING updated_at
+	`
+
+	err := db.session.
+		QueryRow(q, user.ID, user.Email).
+		Scan(&user.UpdatedAt)
+	if errPq, ok := err.(*pq.Error); ok {
+		return handlePgError(errPq)
+	}
+	return err
+}
+
+func (db *DatabasePostgreSQL) GetUsersByGoogleID(id string) (*model.User, error) {
+	q := `
+		SELECT u.id, u.email, u.created_at, u.updated_at, u.nickname, u.google_id
+		FROM public.user u
+		WHERE u.google_id = $1
+	`
+	row := db.session.QueryRow(q, id)
+
+	u := model.User{}
+	err := row.Scan(&u.ID, &u.Email, &u.CreatedAt, &u.UpdatedAt, &u.Nickname, &u.GoogleId)
+	if errPq, ok := err.(*pq.Error); ok {
+		return nil, handlePgError(errPq)
+	}
+	if err == sql.ErrNoRows {
+		return nil, dao.NewDAOError(dao.ErrTypeNotFound, err)
+	}
+	return &u, err
+}