main.go

//
//   date  : 2014-05-23 17:35
//   author: xjdrew
//

package main

import (
	"flag"
	"fmt"
	"net"
	"os"
	"sync"

	"github.com/ejoy/goscon/upstream"
	"github.com/spf13/viper"
	"github.com/xjdrew/glog"
)

// log rule:
// verbose:
// 0, default, 确定数量的日志
// 1, connection related, 跟连接相关的日志， 一般每条连接在生命周期内产生x条
// 2, packet releated, 跟包相关的日志，一般每个packet都会产生数条日志

// <2019-10-18> 1.0.0: 重写goscon的配置方式
// <2019-12-24> 1.1.0: 优化内部设计，提升kcp可靠性等
var _Version = "1.1.0"

func testConfigFile(filename string) error {
	viper.SetConfigFile(filename)
	return viper.ReadInConfig()
}

func main() {
	// set default log directory
	flag.Set("log_dir", "./")

	showVersion := flag.Bool("version", false, "show version and exit")
	testConfig := flag.Bool("t", false, "test configuration and exit")
	dumpConfig := flag.Bool("T", false, "test configuration, dump it and exit")
	configFile := flag.String("config", "./config.yaml", "set configuration file")
	flag.Parse()

	if *showVersion {
		fmt.Printf("goscon version: goscon/%s\n", _Version)
		os.Exit(0)
	}

	if *testConfig || *dumpConfig {
		if err := testConfigFile(*configFile); err != nil {
			fmt.Printf("read configuration file %s faield: %s\n", *configFile, err.Error())

			os.Exit(1)
		}

		fmt.Printf("the configuration file %s syntax is ok\n", *configFile)

		if *dumpConfig {
			fmt.Println(marshalConfigFile())
		}
		os.Exit(0)
	}

	viper.SetConfigFile(*configFile)
	if err := reloadConfig(); err != nil {
		os.Exit(1)
	}

	if err := startManager(viper.GetString("manager")); err != nil {
		glog.Errorf("start manager failed: err=%s", err.Error())
		os.Exit(1)
	}

	// 监听主机(非阻塞)
	upstream.WatchHost()

	var wg sync.WaitGroup

	// listen
	tcpListen := viper.GetString("tcp")
	if tcpListen != "" {
		l, err := NewTCPListener(tcpListen)
		if err != nil {
			glog.Errorf("tcp listen failed: addr=%s, err=%s", tcpListen, err.Error())
			os.Exit(1)
		}
		glog.Infof("tcp listen start: addr=%s", tcpListen)

		wg.Add(1)
		go func(l net.Listener) {
			defer l.Close()
			defer wg.Done()
			err := defaultServer.Serve(l)
			glog.Errorf("tcp listen stop: addr=%s, err=%s", tcpListen, err.Error())
		}(l)
	}

	kcpListen := viper.GetString("kcp")
	if kcpListen != "" {
		reuseport := viper.GetInt("kcp_option.reuseport")
		if reuseport <= 0 {
			reuseport = 1
		}
		for i := 0; i < reuseport; i++ {
			l, err := NewKCPListener(kcpListen)
			if err != nil {
				glog.Errorf("kcp listen failed: addr=%s, err=%s", kcpListen, err.Error())
				os.Exit(1)
			}
			glog.Infof("kcp listen start: addr=%s", kcpListen)

			wg.Add(1)
			go func(l net.Listener) {
				defer l.Close()
				defer wg.Done()
				err := defaultServer.Serve(l)
				glog.Errorf("kcp listen stop: addr=%s, err=%s", tcpListen, err.Error())
			}(l)
		}
	}

	wsListen := viper.GetString("ws")
	if wsListen != "" {
		l, err := NewWSListener(wsListen)
		if err != nil {
			glog.Errorf("ws listen failed: addr=%s, err=%s", wsListen, err.Error())
			os.Exit(1)
		}

		glog.Infof("ws listen start: addr=%s", wsListen)

		wg.Add(1)
		go func(l net.Listener) {
			defer l.Close()
			defer wg.Done()
			err := defaultServer.Serve(l)
			glog.Errorf("ws listen stop: addr=%s, err=%s", tcpListen, err.Error())
		}(l)
	}

	wg.Wait()
	glog.Flush()
}