first commit

This commit is contained in:
RedokPaw
2024-07-03 21:30:49 +03:00
commit aaf436b992
16 changed files with 361 additions and 0 deletions
+47
View File
@@ -0,0 +1,47 @@
package repository
import (
"context"
"fmt"
"log"
"vyatsuAPIInventory/vyatsuAPI/models"
)
func (repo *PGrepo) GetEmployees(offset int, limit int) ([]models.Employee, error) {
rows, err := repo.pool.Query(context.Background(), "select * from employees LIMIT $1 OFFSET $2", limit, offset)
if err != nil {
log.Fatal(err)
return nil, err
}
defer rows.Close()
var employees []models.Employee
for rows.Next() {
e := models.Employee{}
err := rows.Scan(&e.ID, &e.Name, &e.Email, &e.Position, &e.PersonnelNumber)
if err != nil {
fmt.Println(err)
continue
}
employees = append(employees, e)
}
return employees, nil
}
func (repo *PGrepo) GetEmployeesWithFilterByPosition(offset int, limit int, position string) ([]models.Employee, error) {
rows, err := repo.pool.Query(context.Background(), "select * from employees where position = $3 LIMIT $1 OFFSET $2", limit, offset, position)
if err != nil {
log.Fatal(err)
return nil, err
}
defer rows.Close()
var employees []models.Employee
for rows.Next() {
e := models.Employee{}
err := rows.Scan(&e.ID, &e.Name, &e.Email, &e.Position, &e.PersonnelNumber)
if err != nil {
fmt.Println(err)
continue
}
employees = append(employees, e)
}
return employees, nil
}
+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
}