server.go

package main

import (
	"fmt"

	"github.com/gofiber/fiber/v2"

	"github.com/gofiber/websocket/v2"

	BackendHandler "tail-based-sampling/src/backend"
	CliendHandler "tail-based-sampling/src/client"
	Common "tail-based-sampling/src/common"
)

func main() {

	app := fiber.New()
	port := Common.GetEnvDefault("SERVER_PORT", "3000")

	app.Get("/", func(c *fiber.Ctx) error {
		return c.SendString("Hello Baby ~ Johnny is coming!")
	})

	if port == "8002" {
		app.Get("/ready", func(c *fiber.Ctx) error {
			return c.SendString(fmt.Sprintf("Server is running on port: %v", port))
		})
	} else {
		app.Get("/ready", func(c *fiber.Ctx) error {
			go CliendHandler.WsConnection()
			return c.SendString(fmt.Sprintf("Server is running on port: %v", port))
		})
	}

	if port == "8002" {
		app.Post("/setParameter", BackendHandler.SetParameterPostHandler)
		app.Get("/setParameter", BackendHandler.SetParameterGetHandler)
	} else {
		app.Post("/setParameter", CliendHandler.SetParameterPostHandler)
		app.Get("/setParameter", CliendHandler.SetParameterGetHandler)
	}

	// if port == "8002" {
	// 	app.Post("/setWrongTraceId", BackendHandler.SetWrongTraceIDHandler)
	// 	app.Post("/finish", BackendHandler.FinishHandler)
	// } else {
	// 	app.Get("/getWrongTrace/:traceID", CliendHandler.GetWrongTraceHandler)
	// }

	app.Use("/ws", func(c *fiber.Ctx) error {
		// IsWebSocketUpgrade returns true if the client
		// requested upgrade to the WebSocket protocol.
		if websocket.IsWebSocketUpgrade(c) {
			c.Locals("allowed", true)
			return c.Next()
		}
		return fiber.ErrUpgradeRequired
	})

	app.Get("/ws/:id/:type", websocket.New(BackendHandler.WsServerHandler))

	app.Listen(":" + port)
}