janus/server/server.go

package server

import (
	"fmt"
	"net"
	"net/http"
	"strings"
	"sync"

	"github.com/gorilla/mux"
	"github.com/pkg/errors"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/rs/cors"
)

type Server struct {
	HttpServer *http.Server
	Listener   net.Listener
	wg         sync.WaitGroup
}

var (
	networkTCP = "tcp"
)

func NewServer(port string, corsList string) (*Server, error) {

	network := networkTCP

	listener, err := net.Listen(network, ":"+port)
	if err != nil {
		return nil, errors.Wrap(err, "failed to create listener")
	}

	router := mux.NewRouter()
	router.HandleFunc("/events/json", eventsHandlerJson).Methods("POST")
	router.HandleFunc("/health", healthHandler).Methods("GET")

	httpServer := &http.Server{Addr: ":" + port, Handler: enableCors(router, corsList)}
	newServer := &Server{HttpServer: httpServer, Listener: listener}
	return newServer, nil

}

func enableCors(handler http.Handler, corsList string) http.Handler {
	corsArray := strings.Split(corsList, ",")
	fmt.Print(corsArray)
	c := cors.New(cors.Options{
		AllowedOrigins:   corsArray,
		AllowCredentials: false,
		// Enable Debugging for testing, consider disabling in production
		Debug: false,
	})
	return c.Handler(handler)
}
func MetricServer(port string) (*Server, error) {
	network := networkTCP

	listener, err := net.Listen(network, ":"+port)
	if err != nil {
		return nil, errors.Wrap(err, "failed to create listener")
	}

	router := mux.NewRouter()

	router.Path("/metrics").Handler(promhttp.Handler())

	httpServer := &http.Server{Addr: ":" + port, Handler: router}
	mServer := &Server{HttpServer: httpServer, Listener: listener}
	return mServer, nil
}