bottin/v4/handlers/insert.go

package handlers

import (
	"net/http"

	"git.agecem.com/agecem/bottin/v4/data"
	"git.agecem.com/agecem/bottin/v4/models"
	"github.com/labstack/echo/v4"
	"github.com/spf13/viper"
)

func PostMembres(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 membres []models.Membre

	if err := c.Bind(&membres); err != nil {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "Could not bind membres",
			"error":   err.Error(),
		})
	}

	if len(membres) == 0 {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "Nothing to do",
			"error":   "No valid membres to insert were found",
		})
	}

	newMembres, err := client.InsertMembres(membres)
	if err != nil {
		return c.JSON(http.StatusInternalServerError, map[string]string{
			"message": "Could not insert membres",
			"error":   err.Error(),
		})
	}

	return c.JSON(http.StatusOK, map[string]interface{}{
		"message": "Insert successful",
		"data": map[string]interface{}{
			"membres": newMembres,
		},
	})
}

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,
		},
	})
}