first commit

This commit is contained in:
Артём
2024-07-06 07:54:22 +03:00
commit f986c82755
10 changed files with 349 additions and 0 deletions
+116
View File
@@ -0,0 +1,116 @@
package main
import (
"github.com/labstack/echo/v4"
echoSwagger "github.com/swaggo/echo-swagger"
"net/http"
"strconv"
"time"
"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("/request/:id", getRequest)
e.GET("/swagger/*", echoSwagger.WrapHandler)
e.PUT("/request", createRequest)
e.DELETE("/request/:id", deleteRequest)
e.POST("/request/:id", updateRequest)
// Start server
e.Logger.Fatal(e.Start(":8080"))
}
func getRequest(e echo.Context) error {
db, err := repository.New(connectionURL)
id, err := strconv.Atoi(e.Param("id"))
request, err := db.GetRequest(id)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
return e.JSON(http.StatusOK, request)
}
const layout = "2006-01-02"
func stringToTime(dateString string) (time.Time, error) {
t, err := time.Parse(layout, dateString)
if err != nil {
return time.Time{}, err
}
return t, nil
}
func createRequest(e echo.Context) error {
db, err := repository.New(connectionURL)
var Request models.Request
name := e.FormValue("Name")
description := e.FormValue("Description")
Status := e.FormValue("Status")
CreationDate, err := stringToTime(e.FormValue("CreationDate"))
LastUpdateDate, err := stringToTime(e.FormValue("LastUpdateDate"))
Request = models.Request{
Name: name,
Description: description,
Status: Status,
CreationDate: CreationDate,
LastUpdateDate: LastUpdateDate,
}
db.CreateRequest(Request)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
return e.JSON(http.StatusOK, Request)
}
func deleteRequest(c echo.Context) error {
db, err := repository.New(connectionURL)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, "Ошибка подключения к базе данных")
}
id, err := strconv.Atoi(c.Param("id"))
if err != nil {
return echo.NewHTTPError(http.StatusBadRequest, "Неверный формат ID")
}
err = db.DeleteRequest(id)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
return c.NoContent(http.StatusOK)
}
func updateRequest(e echo.Context) error {
db, err := repository.New(connectionURL)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
id, err := strconv.Atoi(e.FormValue("ID"))
if err != nil {
return echo.NewHTTPError(http.StatusBadRequest, "Invalid ID")
}
var request models.Request
request.ID = id
request.Name = e.FormValue("Name")
request.Description = e.FormValue("Description")
request.Status = e.FormValue("Status")
request.CreationDate, err = stringToTime(e.FormValue("CreationDate"))
request.LastUpdateDate, err = stringToTime(e.FormValue("LastUpdateDate"))
err = db.UpdateRequest(request)
if err != nil {
return echo.NewHTTPError(http.StatusInternalServerError, err.Error())
}
return e.JSON(http.StatusOK, request)
}
+12
View File
@@ -0,0 +1,12 @@
package models
import "time"
type Request struct {
ID int
Name string
Description string
Status string
CreationDate time.Time
LastUpdateDate time.Time
}
+41
View File
@@ -0,0 +1,41 @@
package repository
import (
"context"
"zadanie7API/main/models"
)
func (repo *PGrepo) GetRequest(id int) (models.Request, error) {
rows, err := repo.pool.Query(context.Background(), "select * from request WHERE ID = $1", id)
defer rows.Close()
request := models.Request{}
rows.Next()
err = rows.Scan(&request.ID, &request.Name, &request.Description, &request.Status, &request.CreationDate,
&request.LastUpdateDate)
if err != nil {
panic(err)
}
return request, nil
}
func (repo *PGrepo) CreateRequest(request models.Request) {
_, err := repo.pool.Query(context.Background(), "INSERT INTO request (name, description, status, creationdate, lastupdatedate) values ($1, $2, $3, $4, $5)",
request.Name, request.Description, request.Status, request.CreationDate, request.LastUpdateDate)
if err != nil {
panic(err)
}
}
func (repo *PGrepo) DeleteRequest(id int) error {
_, err := repo.pool.Exec(context.Background(), "DELETE FROM request WHERE id = $1", id)
if err != nil {
return err
}
return nil
}
func (repo *PGrepo) UpdateRequest(request models.Request) error {
_, err := repo.pool.Exec(context.Background(), "UPDATE request SET name = $2, description = $3, status = $4, creationdate = $5, lastupdatedate = $6 WHERE id = $1",
request.ID, request.Name, request.Description, request.Status, request.CreationDate, request.LastUpdateDate)
return 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
}