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
+54
View File
@@ -0,0 +1,54 @@
package main
import (
"github.com/labstack/echo/v4"
"github.com/swaggo/echo-swagger"
"net/http"
"strconv"
_ "vyatsuAPIInventory/docs"
"vyatsuAPIInventory/vyatsuAPI/models"
"vyatsuAPIInventory/vyatsuAPI/repository"
)
const connectionURL = "postgres://tgbot:pass@localhost:5432/vyatsu"
// @title Vyatsu
// @host localhost:8080
// @version 1.0
// @BasePath /
func main() {
e := echo.New()
e.GET("/employees", getEmployees)
e.GET("/swagger/*", echoSwagger.WrapHandler)
// Start server
e.Logger.Fatal(e.Start(":8080"))
}
// @Summary Get List Of Employees
// @Tags get
// @Description get list
// @Produce json
// @Success 200
func getEmployees(e echo.Context) error {
db, err := repository.New(connectionURL)
offset, err := strconv.Atoi(e.QueryParam("offset"))
limit, err := strconv.Atoi(e.QueryParam("limit"))
position := e.QueryParam("position")
var employees []models.Employee
if limit == 0 {
limit = 10
}
if position == "" {
employees, err = db.GetEmployees(offset, limit)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
} else {
employees, err = db.GetEmployeesWithFilterByPosition(offset, limit, position)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
}
return e.JSON(http.StatusOK, employees)
}
+9
View File
@@ -0,0 +1,9 @@
package models
type Employee struct {
ID int
Name string
Email string
Position string
PersonnelNumber int
}
+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
}