From f1a4d190dfc1d4935e32fa22e52bd2701dbec22e Mon Sep 17 00:00:00 2001 From: Victor Lacasse-Beaudoin Date: Thu, 25 May 2023 03:44:41 -0400 Subject: [PATCH] Ajouter insertion de programmes --- v4/cmd/server.go | 2 ++ v4/handlers/insert.go | 49 +++++++++++++++++++++++++++++++++++++++++++ v4/models/models.go | 2 +- 3 files changed, 52 insertions(+), 1 deletion(-) diff --git a/v4/cmd/server.go b/v4/cmd/server.go index a2c8799..e2c4c48 100644 --- a/v4/cmd/server.go +++ b/v4/cmd/server.go @@ -45,6 +45,8 @@ var serverCmd = &cobra.Command{ e.GET("/v4/membres/:membre_id/", handlers.ReadMembre) e.POST("/v4/membres/", handlers.PostMembres) + e.POST("/v4/programmes/", handlers.PostProgrammes) + // Execution connection := data.PostgresConnection{ diff --git a/v4/handlers/insert.go b/v4/handlers/insert.go index 4d07443..f534d9c 100644 --- a/v4/handlers/insert.go +++ b/v4/handlers/insert.go @@ -57,3 +57,52 @@ func PostMembres(c echo.Context) error { }, }) } + +func PostProgrammes(c echo.Context) error { + connection := data.PostgresConnection{ + User: viper.GetString("db.user"), + Password: viper.GetString("db.password"), + Host: viper.GetString("db.host"), + Database: viper.GetString("db.database"), + Port: viper.GetInt("db.port"), + } + + client, err := data.NewDataClient(connection) + if err != nil { + return c.JSON(http.StatusInternalServerError, map[string]string{ + "message": "Could not establish database connection", + "error": err.Error(), + }) + } + + var programmes []models.Programme + + if err := c.Bind(&programmes); err != nil { + return c.JSON(http.StatusBadRequest, map[string]string{ + "message": "Could not bind programmes", + "error": err.Error(), + }) + } + + if len(programmes) == 0 { + return c.JSON(http.StatusBadRequest, map[string]string{ + "message": "Nothing to do", + "error": "No valid programmes to insert were found", + }) + } + + newProgrammes, err := client.InsertProgrammes(programmes) + if err != nil { + return c.JSON(http.StatusInternalServerError, map[string]string{ + "message": "Could not insert programmes", + "error": err.Error(), + }) + } + + return c.JSON(http.StatusOK, map[string]interface{}{ + "message": "Insert successful", + "data": map[string]interface{}{ + "programmes": newProgrammes, + }, + }) +} diff --git a/v4/models/models.go b/v4/models/models.go index 2c29a82..c3f4af8 100644 --- a/v4/models/models.go +++ b/v4/models/models.go @@ -17,7 +17,7 @@ CREATE TABLE membres ( type Programme struct { ID string `db:"id" json:"programme_id"` - Titre string `db:"titre" json:"titre"` + Titre string `db:"titre" json:"nom_programme"` } type Membre struct {