add: constraints, validations and gorm configs

WeDias · WeDias · commit e73bc59ae436 · 2022-04-21T09:49:25.000-03:00
diff --git a/database/database.go b/database/database.go
@@ -6,16 +6,27 @@ import (
 
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
+	"gorm.io/gorm/schema"
 )
 
 var Conn *gorm.DB
 
 func InitDatabase() {
 	var err error
+
 	dsn := fmt.Sprintf(
-		"host=%s user=postgres password=postgres dbname=postgres port=5432",
-		os.Getenv("PG_HOST"))
-	Conn, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
+		"host=%v user=%v password=%v dbname=%v port=%v",
+		os.Getenv("PG_HOST"), os.Getenv("PG_USER"), os.Getenv("PG_PASS"),
+		os.Getenv("PG_DBNM"), os.Getenv("PG_PORT"),
+	)
+
+	gormConfig := &gorm.Config{
+		NamingStrategy: schema.NamingStrategy{
+			SingularTable: true,
+		},
+	}
+
+	Conn, err = gorm.Open(postgres.Open(dsn), gormConfig)
 	if err != nil {
 		panic("failed to connect database")
 	}
diff --git a/entities/productEntity.go b/entities/productEntity.go
@@ -1,8 +1,8 @@
 package entities
 
 type Product struct {
-	ProCod            int64   `gorm:"primaryKey" json:"id"`
-	ProName           string  `json:"name"`
-	ProPrice          float32 `json:"price"`
-	ProAvailableStock uint    `json:"stock"`
+	Cod            int64   `json:"id"    gorm:"column:pro_cod; primaryKey; <-:create"`
+	Name           string  `json:"name"  gorm:"column:pro_name"`
+	Price          float32 `json:"price" gorm:"column:pro_price"`
+	AvailableStock uint    `json:"stock" gorm:"column:pro_available_stock"`
 }
diff --git a/resources/ddl-database.sql b/resources/ddl-database.sql
@@ -1,7 +1,12 @@
-create table if not exists products (
+create table if not exists product (
     pro_cod bigint not null generated by default as identity,
     pro_name varchar not null,
-    pro_price numeric not null,
+    pro_price numeric(12, 2) not null,
     pro_available_stock integer not null,
-    constraint products_pro_cod_pkey primary key (pro_cod)
+
+    constraint products_pro_cod_pkey primary key (pro_cod),
+    constraint products_pro_name_ukey unique (pro_name),
+    constraint products_pro_name_check check (pro_name <> ''),
+    constraint products_pro_price_check check (pro_price >= 0),
+    constraint products_pro_available_stock_check check (pro_available_stock >= 0)
 );
diff --git a/resources/dev.env b/resources/dev.env
@@ -1 +1,5 @@
-PG_HOST=localhost
+PG_HOST=localhost
+PG_USER=postgres
+PG_PASS=postgres
+PG_DBNM=postgres
+PG_PORT=5432
diff --git a/resources/docker.env b/resources/docker.env
@@ -1 +1,5 @@
-PG_HOST=postgres-go
+PG_HOST=postgres-go
+PG_USER=postgres
+PG_PASS=postgres
+PG_DBNM=postgres
+PG_PORT=5432
diff --git a/services/productService.go b/services/productService.go
@@ -9,18 +9,20 @@ import (
 
 func NewProduct(product *entities.Product) (*entities.Product, error) {
 	var err error
-	database.Conn.Create(&product)
-	if product.ProCod == 0 {
-		err = errors.New("product was not created")
+	tx := database.Conn.Create(&product)
+	if tx.Error != nil {
+		err = errors.New(tx.Error.Error())
 	}
 	return product, err
 }
 
 func GetProduct(productId string) (*entities.Product, error) {
 	var err error
 	var product *entities.Product
-	database.Conn.Find(&product, productId)
-	if product.ProCod == 0 {
+	tx := database.Conn.Find(&product, productId)
+	if tx.Error != nil {
+		err = errors.New(tx.Error.Error())
+	} else if product.Cod == 0 {
 		err = errors.New("product not found")
 	}
 	return product, err
@@ -29,21 +31,33 @@ func GetProduct(productId string) (*entities.Product, error) {
 func GetProducts() (*[]entities.Product, error) {
 	var err error
 	var products *[]entities.Product
-	database.Conn.Find(&products)
-	if *products == nil {
+	tx := database.Conn.Order("pro_cod").Find(&products)
+	if tx.Error != nil {
+		err = errors.New(tx.Error.Error())
+	} else if *products == nil {
 		err = errors.New("products not found")
 	}
 	return products, err
 }
 
 func UpdateProduct(newProduct *entities.Product, productId string) (*entities.Product, error) {
 	product, err := GetProduct(productId)
-	database.Conn.Model(product).Updates(newProduct)
+	if err == nil {
+		tx := database.Conn.Model(product).Updates(newProduct)
+		if tx.Error != nil {
+			err = errors.New(tx.Error.Error())
+		}
+	}
 	return product, err
 }
 
 func DeleteProduct(productId string) (*entities.Product, error) {
 	product, err := GetProduct(productId)
-	database.Conn.Delete(product)
+	if err == nil {
+		tx := database.Conn.Delete(product)
+		if tx.Error != nil {
+			err = errors.New(tx.Error.Error())
+		}
+	}
 	return product, err
 }
