cmd/controller/main.go

// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License.  You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"fmt"
	_ "github.com/apache/incubator-eventmesh/eventmesh-catalog-go/cmd/controller/docs"
	"github.com/apache/incubator-eventmesh/eventmesh-catalog-go/internal/dal"
	"github.com/apache/incubator-eventmesh/eventmesh-catalog-go/internal/util"
	"github.com/apache/incubator-eventmesh/eventmesh-server-go/config"
	"github.com/gin-gonic/gin"
	swaggerFiles "github.com/swaggo/files"
	ginSwagger "github.com/swaggo/gin-swagger"
	"log"
	"net/http"
)

type Server struct {
	server  *gin.Engine
	catalog *CatalogController
}

// @title           Event Catalog API
// @version         1.0
// @description     This is a event catalog server.

// @license.name  Apache 2.0
// @license.url   http://www.apache.org/licenses/LICENSE-2.0.html
func main() {
	s, err := initServer()
	if err != nil {
		log.Fatal("new server fail: " + err.Error())
	}
	s.router()
	if err := s.run(); err != nil {
		log.Fatal("run server fail: " + err.Error())
	}
}

func initServer() (*Server, error) {
	var s Server
	if err := s.setupConfig(); err != nil {
		return nil, err
	}
	if err := dal.Open(); err != nil {
		return nil, err
	}
	r := gin.New()
	r.Use(cors()).Use(gin.Recovery())
	swagger(r)
	s.server = r
	s.catalog = NewCatalogController()
	return &s, nil
}

func cors() gin.HandlerFunc {
	return func(c *gin.Context) {
		method := c.Request.Method
		origin := c.Request.Header.Get("Origin")
		if origin != "" {
			c.Header("Access-Control-Allow-Origin", "*")
			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
			c.Header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, "+
				"Authorization")
			c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, "+
				"Access-Control-Allow-Headers, Cache-Control, Content-Language, Content-Type")
			c.Header("Access-Control-Allow-Credentials", "true")
		}
		if method == "OPTIONS" {
			c.AbortWithStatus(http.StatusNoContent)
		}
		c.Next()
	}
}

func swagger(r *gin.Engine) {
	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
}

func (s *Server) router() {
	s.server.POST("/catalog", s.catalog.Save)
	s.server.GET("/catalog", s.catalog.QueryList)
	s.server.GET("/catalog/:eventId", s.catalog.QueryDetail)
}

func (s *Server) setupConfig() error {
	config.ServerConfigPath = "./configs/controller.yaml"
	// compatible local environment
	if !util.Exists(config.ServerConfigPath) {
		config.ServerConfigPath = "../configs/controller.yaml"
	}
	// compatible deploy environment
	if !util.Exists(config.ServerConfigPath) {
		config.ServerConfigPath = "../conf/controller.yaml"
	}
	cfg, err := config.LoadConfig(config.ServerConfigPath)
	if err != nil {
		return err
	}
	config.SetGlobalConfig(cfg)
	return config.Setup(cfg)
}

func (s *Server) run() error {
	return s.server.Run(fmt.Sprintf(":%d", config.GlobalConfig().Server.Port))
}