Add login

storage/dao/postgresql/database_postgresql.go

@@ -0,0 +1,43 @@
+package postgresql
+
+import (
+	"database/sql"
+	"fmt"
+	"hamster-tycoon/storage/dao"
+	"hamster-tycoon/utils"
+
+	"github.com/lib/pq"
+)
+
+const (
+	pgCodeUniqueViolation     = "23505"
+	pgCodeForeingKeyViolation = "23503"
+)
+
+func handlePgError(e *pq.Error) error {
+	if e.Code == pgCodeUniqueViolation {
+		return dao.NewDAOError(dao.ErrTypeDuplicate, e)
+	}
+
+	if e.Code == pgCodeForeingKeyViolation {
+		return dao.NewDAOError(dao.ErrTypeForeignKeyViolation, e)
+	}
+	return e
+}
+
+type DatabasePostgreSQL struct {
+	session *sql.DB
+}
+
+func NewDatabasePostgreSQL(connectionURI string) dao.Database {
+	db, err := sql.Open("postgres", connectionURI)
+	if err != nil {
+		utils.GetLogger().WithError(err).Fatal("Unable to get a connection to the postgres db")
+	}
+	err = db.Ping()
+	if err != nil {
+		fmt.Println(err)
+		utils.GetLogger().WithError(err).Fatal("Unable to ping the postgres db")
+	}
+	return &DatabasePostgreSQL{session: db}
+}

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
+}