init

internal/adapter/storages/postgres/db.go

@@ -0,0 +1,38 @@
+package postgres
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/aykhans/oh-my-chat/internal/adapter/config"
+	postgresDriver "gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func NewDB(config *config.PostgresConfig) (*gorm.DB, error) {
+	dsn := fmt.Sprintf(
+		"host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=%s",
+		config.Host,
+		config.User,
+		config.Password,
+		config.DBName,
+		config.Port,
+		"disable",
+		"UTC",
+	)
+
+	var db *gorm.DB
+	var err error
+	for range 3 {
+		db, err = gorm.Open(postgresDriver.Open(dsn), &gorm.Config{})
+		if err == nil {
+			break
+		}
+		time.Sleep(3 * time.Second)
+	}
+	if err != nil {
+		return nil, err
+	}
+
+	return db, nil
+}

internal/adapter/storages/postgres/migrations/000001_create_users_table.down.sql

@@ -0,0 +1 @@
+DROP TABLE IF EXISTS "public"."users";

internal/adapter/storages/postgres/migrations/000001_create_users_table.up.sql

@@ -0,0 +1,11 @@
+CREATE TABLE "public"."users" (
+    "id" uuid NOT NULL DEFAULT gen_random_uuid(),
+    "username" character varying(50) NOT NULL,
+    "email" text NOT NULL,
+    "password" character varying(72) NOT NULL,
+    "created_at" timestamptz NOT NULL DEFAULT now(),
+    "updated_at" timestamptz NOT NULL DEFAULT now(),
+    PRIMARY KEY ("id"),
+    CONSTRAINT "uni_users_email" UNIQUE ("email"),
+    CONSTRAINT "uni_users_username" UNIQUE ("username")
+);

internal/adapter/storages/postgres/models/user.go

@@ -0,0 +1,24 @@
+package models
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+const (
+	UserTableName = "users"
+)
+
+type User struct {
+	ID        uuid.UUID `gorm:"primarykey;unique;type:uuid;default:gen_random_uuid()"`
+	Username  string    `gorm:"unique;not null;size:50"`
+	Email     string    `gorm:"unique;not null"`
+	Password  string    `gorm:"not null;size:72"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+}
+
+func (u User) TableName() string {
+	return UserTableName
+}

internal/adapter/storages/postgres/repository/user.go

@@ -0,0 +1,109 @@
+package repository
+
+import (
+	"context"
+	"errors"
+
+	"github.com/aykhans/oh-my-chat/internal/adapter/storages/postgres/models"
+	"github.com/aykhans/oh-my-chat/internal/core/domain"
+	"gorm.io/gorm"
+)
+
+type UserRepository struct {
+	db *gorm.DB
+}
+
+func NewUserRepository(db *gorm.DB) *UserRepository {
+	return &UserRepository{db}
+}
+
+func (userRepository *UserRepository) CreateUser(
+	ctx context.Context,
+	user *domain.User,
+) (*domain.User, error) {
+	userModel := &models.User{
+		Username: user.Username,
+		Email:    user.Email,
+		Password: user.Password,
+	}
+	tx := userRepository.db.Create(userModel)
+	if tx.Error != nil {
+		return nil, tx.Error
+	}
+
+	user.ID = userModel.ID
+	user.Username = userModel.Username
+	user.Email = userModel.Email
+	user.Password = userModel.Password
+	return user, nil
+}
+
+func (userRepository *UserRepository) IsUsernameExists(
+	ctx context.Context,
+	username string,
+) (bool, error) {
+	var count int64
+	tx := userRepository.db.
+		Table(models.UserTableName).
+		Where("username = ?", username).
+		Count(&count)
+	if tx.Error != nil {
+		return false, tx.Error
+	}
+
+	return count > 0, nil
+}
+
+func (userRepository *UserRepository) IsEmailExists(
+	ctx context.Context,
+	email string,
+) (bool, error) {
+	var count int64
+	tx := userRepository.db.
+		Table(models.UserTableName).
+		Where("email = ?", email).
+		Count(&count)
+	if tx.Error != nil {
+		return false, tx.Error
+	}
+
+	return count > 0, nil
+}
+
+func (userRepository *UserRepository) GetUserByEmail(
+	ctx context.Context,
+	email string,
+) (*domain.User, error) {
+	user := &domain.User{}
+	tx := userRepository.db.
+		Table(models.UserTableName).
+		Where("email = ?", email).
+		First(user)
+	if tx.Error != nil {
+		if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
+			return nil, domain.ErrDataNotFound
+		}
+		return nil, tx.Error
+	}
+
+	return user, nil
+}
+
+func (userRepository *UserRepository) GetUserByUsername(
+	ctx context.Context,
+	username string,
+) (*domain.User, error) {
+	user := &domain.User{}
+	tx := userRepository.db.
+		Table(models.UserTableName).
+		Where("username = ?", username).
+		First(user)
+	if tx.Error != nil {
+		if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
+			return nil, domain.ErrDataNotFound
+		}
+		return nil, tx.Error
+	}
+
+	return user, nil
+}