first commit

zadanie7API/main/main.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"github.com/labstack/echo/v4"
+	echoSwagger "github.com/swaggo/echo-swagger"
+	"net/http"
+	"strconv"
+	"zadanie7API/main/models"
+	"zadanie7API/main/repository"
+)
+
+const connectionURL = "postgres://tgbot:pass@localhost:5432/vyatsu"
+
+//INPUT DATA IN FORM-DATA FORMAT
+
+func main() {
+	e := echo.New()
+	e.GET("/equipment/:id", getEquipment)
+	e.GET("/swagger/*", echoSwagger.WrapHandler)
+	e.PUT("/equipment", createEquipment)
+	// Start server
+	e.Logger.Fatal(e.Start(":8080"))
+}
+func getEquipment(e echo.Context) error {
+	db, err := repository.New(connectionURL)
+	id, err := strconv.Atoi(e.Param("id"))
+	equipment, err := db.GetEquipment(id)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
+	}
+	return e.JSON(http.StatusOK, equipment)
+}
+func createEquipment(e echo.Context) error {
+	db, err := repository.New(connectionURL)
+
+	var Equipment models.Equipment
+
+	name := e.FormValue("Name")
+	ownerId, err := strconv.Atoi(e.FormValue("OwnerId"))
+	receiptDate := e.FormValue("ReceiptDate")
+	liteTime := e.FormValue("LifeTime")
+	inventoryNumber, err := strconv.Atoi(e.FormValue("InventoryNumber"))
+
+	Equipment = models.Equipment{
+		Name:            name,
+		OwnerId:         ownerId,
+		ReceiptDate:     receiptDate,
+		LifeTime:        liteTime,
+		InventoryNumber: inventoryNumber,
+	}
+	db.CreateEquipment(Equipment)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
+	}
+	return e.JSON(http.StatusOK, Equipment)
+}

zadanie7API/main/models/models.go

@@ -0,0 +1,10 @@
+package models
+
+type Equipment struct {
+	GUID            int
+	InventoryNumber int
+	Name            string
+	OwnerId         int
+	ReceiptDate     string
+	LifeTime        string
+}

zadanie7API/main/repository/equipment.go

@@ -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)
+	}
+}

zadanie7API/main/repository/pgsql.go

@@ -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
+}