add serve monitoring

Author: Allyson-English

cmd/contract-event-processor/main.go

@@ -10,6 +10,9 @@ import (
 
 	"github.com/DIMO-Network/shared"
 	"github.com/Shopify/sarama"
+	"github.com/gofiber/adaptor/v2"
+	"github.com/gofiber/fiber/v2"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/rs/zerolog"
 )
 
@@ -39,6 +42,8 @@ func main() {
 		}
 	}
 
+	monApp := serveMonitoring(settings.MonitoringPort, &logger)
+
 	kafkaClient, err := services.StartKafkaStream(settings)
 	if err != nil {
 		log.Fatal(err)
@@ -57,4 +62,24 @@ func main() {
 
 	listener.CompileRegistryMap("config.yaml")
 	listener.ChainIndexer(blockNum)
+
+	monApp.Shutdown()
+}
+
+func serveMonitoring(port string, logger *zerolog.Logger) *fiber.App {
+	monApp := fiber.New(fiber.Config{DisableStartupMessage: true})
+
+	// Health check.
+	monApp.Get("/", func(c *fiber.Ctx) error { return nil })
+	monApp.Get("/metrics", adaptor.HTTPHandler(promhttp.Handler()))
+
+	go func() {
+		if err := monApp.Listen(":" + port); err != nil {
+			logger.Fatal().Err(err).Str("port", port).Msg("Failed to start monitoring web server.")
+		}
+	}()
+
+	logger.Info().Str("port", port).Msg("Started monitoring web server.")
+
+	return monApp
 }

internal/block_listener.go

@@ -48,6 +48,8 @@ type Settings struct {
 	PostgresDB       string `yaml:"POSTGRES_DB"`
 	PostgresHOST     string `yaml:"POSTGRES_HOST"`
 	PostgresPort     int    `yaml:"POSTGRES_PORT"`
+
+	MonitoringPort string `yaml:"MONITORING_PORT"`
 }
 
 type BlockListener struct {

sample.settings.yaml

@@ -1,6 +1,6 @@
 WEB_SOCKET_ADDRESS: wss://polygon-mainnet.g.alchemy.com/v2/
 API_KEY:
-EVENT_STREAM_TOPIC: topic.event.stream
+EVENT_STREAM_TOPIC: topic.contract.event
 PARTITIONS: 1
 KAFKA_BROKER: localhost:9092
 POSTGRES_USER: postgres
@@ -9,3 +9,4 @@ POSTGRES_DB: postgres
 POSTGRES_HOST: localhost
 POSTGRES_PORT: 5432
 BLOCK_CONFIRMATIONS: 5
+MONITORING_PORT: 8888

sqlboiler.toml

@@ -1,5 +1,5 @@
 [psql]
-schema = "chain_indexer"
+schema = "contract_event_processor"
 dbname = "postgres"
 blacklist = ["migrations"]
 host = "localhost"