janus/main.go

package main

import (
	config "com.navi.medici.janus/config"
	lib "com.navi.medici.janus/lib"
	producer_client "com.navi.medici.janus/producer"
	"com.navi.medici.janus/schema"
	server "com.navi.medici.janus/server"

	"github.com/spf13/viper"
	"log"
)

var (
	port string
	cors string
)

const metrics_port = "4000"

func init() {
	configs := getConfigs()
	port = configs.Server.Port
	log.Printf("PORT IS: %v", port)
	log.Printf(configs.Kafka.Bootstrap_Servers)
	log.Printf(configs.SchemaRegistry.Endpoint)
	schema.InitializeSchemaHandler(configs)
	producer_client.InitializeProducers(configs.Kafka, configs.Env)

	for i := 0; i < 2; i++ {
		go lib.ProcessProtobufRequestChannel(configs.Kafka.Kafka_Topic_Protobuf)
	}

	for i := 0; i < configs.Server.Goroutines; i++ {
		go lib.ProcessJsonRequestChannel(configs.Kafka.Kafka_Topic_Json)
	}
}

func getConfigs() config.Configurations {
	viper.SetConfigName("config")
	viper.AddConfigPath("./config")
	viper.SetConfigType("yml")
	viper.AutomaticEnv()

	var configuration config.Configurations
	if err := viper.ReadInConfig(); err != nil {
		log.Fatalln("Error reading config file, %s", err)
	}

	err := viper.Unmarshal(&configuration)
	if err != nil {
		log.Fatalln("Unable to decode into struct, %v", err)
	}
	//log.Print(configuration.Kafka.Bootstrap_Servers)
	// Following coinfigurations read from environment variables
	configuration.Env = viper.GetString(configuration.Env)
	configuration.Kafka.Bootstrap_Servers = viper.GetString(configuration.Kafka.Bootstrap_Servers)
	configuration.SchemaRegistry.Endpoint = viper.GetString(configuration.SchemaRegistry.Endpoint)
	configuration.SchemaRegistry.Topics = viper.GetString(configuration.SchemaRegistry.Topics)
	configuration.Kafka.Kafka_Topic_Json = viper.GetString(configuration.Kafka.Kafka_Topic_Json)
	configuration.Kafka.Kafka_Topic_Protobuf = viper.GetString(configuration.Kafka.Kafka_Topic_Protobuf)

	//log.Printf("Env is: ", configuration.Env)
	//if configuration.Env == "PROD" {
	configuration.Kafka.Sasl_User = viper.GetString(configuration.Kafka.Sasl_User)
	configuration.Kafka.Sasl_Password = viper.GetString(configuration.Kafka.Sasl_Password)
	//}
	cors = viper.GetString(configuration.Server.Cors)
	return configuration
}

func main() {
	log.Printf("Serving on http://0.0.0.0:", port)
	httpServer, err1 := server.NewServer(port, cors)
	metricsServer, err2 := server.MetricServer(metrics_port)
	if err1 != nil {
		log.Fatalln("Unable to start server, ", err1)
	}
	if err2 != nil {
		log.Fatalln("Unable to start Metric server, ", err2)
	}
	go httpServer.HttpServer.Serve(httpServer.Listener)
	go metricsServer.HttpServer.Serve(metricsServer.Listener)

	select {}
}