first commit

This commit is contained in:
Garrou_5
2024-07-05 11:09:19 +03:00
commit 2a79781b93
9 changed files with 259 additions and 0 deletions
+27
View File
@@ -0,0 +1,27 @@
package repository
import (
"context"
"zadanie7API/main/models"
)
func (repo *PGrepo) GetEquipment(id int) (models.Equipment, error) {
rows, err := repo.pool.Query(context.Background(), "select * from equipment WHERE GUID = $1", id)
defer rows.Close()
equipment := models.Equipment{}
rows.Next()
err = rows.Scan(&equipment.GUID, &equipment.InventoryNumber, &equipment.Name, &equipment.OwnerId, &equipment.ReceiptDate,
&equipment.LifeTime)
if err != nil {
panic(err)
}
return equipment, nil
}
func (repo *PGrepo) CreateEquipment(equipment models.Equipment) {
_, err := repo.pool.Query(context.Background(), "INSERT INTO equipment (inventoryNumber, Name, OwnerId, receiptDate, lifeTime) values ($1, $2, $3, $4, $5)",
equipment.InventoryNumber, equipment.Name, equipment.OwnerId, equipment.ReceiptDate, equipment.LifeTime)
if err != nil {
panic(err)
}
}
+20
View File
@@ -0,0 +1,20 @@
package repository
import (
"context"
"github.com/jackc/pgx/v5/pgxpool"
"sync"
)
type PGrepo struct {
mu sync.Mutex
pool *pgxpool.Pool
}
func New(connStr string) (*PGrepo, error) {
pool, err := pgxpool.New(context.Background(), connStr)
if err != nil {
panic(err)
}
return &PGrepo{mu: sync.Mutex{}, pool: pool}, nil
}