server.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"

	"github.com/bervProject/go-microservice-boilerplate/graph_base"
	"github.com/bervProject/go-microservice-boilerplate/models"
	"github.com/bervProject/go-microservice-boilerplate/services"
	"github.com/bervProject/go-microservice-boilerplate/utils"
	"github.com/labstack/echo/v4"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

func main() {
	dsn := utils.GetEnv("DB_CONNECTION_STRING", "")
	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		logger.Config{
			LogLevel:                  logger.Info, // Log level
			IgnoreRecordNotFoundError: true,        // Ignore ErrRecordNotFound error for logger
			Colorful:                  false,       // Disable color
		},
	)
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		Logger: newLogger,
	})
	if err != nil {
		log.Fatalf("failed to connect database, error: %v", err)
	}
	if err = db.AutoMigrate(&models.User{}); err != nil {
		log.Fatalf("Failed migrate, err: %v", err)
	}
	graph_base.InitGraphQL(db)
	e := echo.New()
	e.GET("/", func(c echo.Context) error {
		return c.String(http.StatusOK, "Hello, World!")
	})
	e.POST("/graphql", graph_base.GraphQLHandler)
	e.POST("/start-ec2", func(c echo.Context) error {
		ec2Handler := services.NewEC2Handler()
		return ec2Handler.StartInstance(c.Request().Context())
	})
	e.POST("/stop-ec2", func(c echo.Context) error {
		ec2Handler := services.NewEC2Handler()
		return ec2Handler.StopInstance(c.Request().Context())
	})
	e.POST("/status", func(c echo.Context) error {
		ec2Handler := services.NewEC2Handler()
		data, err := ec2Handler.InstanceStatus(c.Request().Context())
		if err != nil {
			return err
		}
		if err := c.JSON(200, data); err != nil {
			return err
		}
		return nil
	})
	e.POST("/publish-message", func(c echo.Context) error {
		messageRequest := &services.KafkaRequest{}
		if err := c.Bind(messageRequest); err != nil {
			return err
		}
		producer := services.NewProducer()
		return producer.Publish(c.Request().Context(), messageRequest.Message)
	})
	e.Logger.Fatal(e.Start(fmt.Sprintf(":%s", utils.GetEnv("PORT", "1323"))))
}