router.go

package main

import (
	"github.com/sinhashubham95/go-actuator"
	actuatorCore "github.com/sinhashubham95/go-actuator/core"
	actuatorModels "github.com/sinhashubham95/go-actuator/models"
	"github.com/valyala/fasthttp"
	"net/http"
	"strings"

	"github.com/sinhashubham95/moxy/commons"
	"github.com/sinhashubham95/moxy/controllers"
)

func getFastHTTPHandler() fasthttp.RequestHandler {
	actuatorHandler := actuator.GetFastHTTPActuatorHandler(&actuatorModels.Config{
		Prefix: commons.ActuatorPrefix,
	})
	return actuatorCore.WrapFastHTTPHandler(func(ctx *fasthttp.RequestCtx) {
		if strings.HasPrefix(string(ctx.Path()), commons.ActuatorPrefix) && string(ctx.Method()) == http.MethodGet {
			// this means it is an actuator endpoint
			actuatorHandler(ctx)
			return
		}
		switch string(ctx.Path()) {
		case commons.BasePath:
			handle(ctx, http.MethodGet, controllers.HandleBase)
		case commons.MockEndpointPath:
			handle(ctx, http.MethodPost, controllers.HandleMock)
		case commons.UnMockEndpointPath:
			handle(ctx, http.MethodDelete, controllers.HandleUnMock)
		default:
			controllers.Handle(ctx)
		}
	})
}

func handle(ctx *fasthttp.RequestCtx, method string, handler fasthttp.RequestHandler) {
	if string(ctx.Method()) == method {
		handler(ctx)
		return
	}
	ctx.SetStatusCode(http.StatusNotFound)
}