main.go

package main

import (
	"fmt"
	"os"
	"strings"

	"github.com/openfaas-incubator/faas-federation/handlers"
	"github.com/openfaas-incubator/faas-federation/routing"
	"github.com/openfaas-incubator/faas-federation/types"
	"github.com/openfaas-incubator/faas-federation/version"
	bootstrap "github.com/openfaas/faas-provider"
	"github.com/openfaas/faas-provider/proxy"

	bootTypes "github.com/openfaas/faas-provider/types"
	log "github.com/sirupsen/logrus"
)

func init() {
	logFormat := os.Getenv("LOG_FORMAT")
	logLevel := os.Getenv("LOG_LEVEL")
	if strings.EqualFold(logFormat, "json") {
		log.SetFormatter(&log.JSONFormatter{
			FieldMap: log.FieldMap{
				log.FieldKeyMsg:  "message",
				log.FieldKeyTime: "@timestamp",
			},
			TimestampFormat: "2006-01-02T15:04:05.999Z07:00",
		})
	} else {
		log.SetFormatter(&log.TextFormatter{
			FullTimestamp: true,
		})
	}

	if level, err := log.ParseLevel(logLevel); err == nil {
		log.SetLevel(level)
	}
}

func main() {

	log.Infof("faas-federation version: %s. Last commit message: %s, commit SHA: %s", version.BuildVersion(), version.GitCommitMessage, version.GitCommitSHA)

	readConfig := types.ReadConfig{}
	osEnv := types.OsEnv{}
	cfg := readConfig.Read(osEnv)

	providerLookup, err := routing.NewDefaultProviderRouting(cfg.Providers, cfg.DefaultProvider)
	if err != nil {
		panic(fmt.Errorf("could not create provider lookup, error: %v", err))
	}

	err = providerLookup.ReloadCache()
	if err != nil {
		panic(fmt.Errorf("could not reload provider cache, error: %v", err))
	}

	proxyFunc := proxy.NewHandlerFunc(cfg.ReadTimeout,
		handlers.NewFunctionLookup(providerLookup))

	bootstrapHandlers := bootTypes.FaaSHandlers{
		FunctionProxy:  handlers.MakeProxyHandler(proxyFunc),
		DeleteHandler:  handlers.MakeDeleteHandler(proxyFunc),
		DeployHandler:  handlers.MakeDeployHandler(proxyFunc, providerLookup),
		FunctionReader: handlers.MakeFunctionReader(cfg.Providers),
		ReplicaReader:  handlers.MakeReplicaReader(),
		ReplicaUpdater: handlers.MakeReplicaUpdater(),
		UpdateHandler:  handlers.MakeUpdateHandler(proxyFunc, providerLookup),
		HealthHandler:  handlers.MakeHealthHandler(),
		InfoHandler:    handlers.MakeInfoHandler(version.BuildVersion(), version.GitCommitSHA),
	}

	bootstrapConfig := bootTypes.FaaSConfig{
		ReadTimeout:     cfg.ReadTimeout,
		WriteTimeout:    cfg.WriteTimeout,
		TCPPort:         &cfg.Port,
		EnableHealth:    true,
		EnableBasicAuth: false,
	}

	log.Infof("listening on port %d", cfg.Port)
	bootstrap.Serve(&bootstrapHandlers, &bootstrapConfig)
}