mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-02 11:47:36 +08:00
102 lines
3.9 KiB
Go
102 lines
3.9 KiB
Go
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
|
|
// RAINBOND, Application Management Platform
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
|
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
|
// must be obtained first.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package option
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/goodrain/rainbond/util"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/spf13/pflag"
|
|
)
|
|
|
|
type GWServer struct {
|
|
Config
|
|
LogLevel string
|
|
}
|
|
|
|
func NewGWServer() *GWServer {
|
|
return &GWServer{}
|
|
}
|
|
|
|
//Config contains all configuration
|
|
type Config struct {
|
|
K8SConfPath string
|
|
ListenPorts ListenPorts
|
|
//This number should be, at maximum, the number of CPU cores on your system.
|
|
WorkerProcesses int
|
|
WorkerRlimitNofile int
|
|
ErrorLog string
|
|
WorkerConnections int
|
|
//essential for linux, optmized to serve many clients with each thread
|
|
EnableEpool bool
|
|
EnableMultiAccept bool
|
|
KeepaliveTimeout int
|
|
KeepaliveRequests int
|
|
NginxUser string
|
|
IP string
|
|
}
|
|
|
|
// ListenPorts describe the ports required to run the gateway controller
|
|
type ListenPorts struct {
|
|
HTTP int
|
|
HTTPS int
|
|
Status int
|
|
AuxiliaryPort int
|
|
}
|
|
|
|
// AddFlags adds flags
|
|
func (g *GWServer) AddFlags(fs *pflag.FlagSet) {
|
|
fs.StringVar(&g.LogLevel, "log-level", "debug", "the gateway log level")
|
|
fs.StringVar(&g.K8SConfPath, "kube-conf", "/opt/rainbond/etc/kubernetes/kubecfg/admin.kubeconfig", "absolute path to the kubeconfig file")
|
|
fs.IntVar(&g.ListenPorts.AuxiliaryPort, "auxiliary-port", 10253, "port of auxiliary server")
|
|
fs.IntVar(&g.WorkerProcesses, "worker-processes", 0, "Default get current compute cpu core number.This number should be, at maximum, the number of CPU cores on your system.")
|
|
fs.IntVar(&g.WorkerConnections, "worker-connections", 4000, "Determines how many clients will be served by each worker process.")
|
|
fs.IntVar(&g.WorkerRlimitNofile, "worker-rlimit-nofile", 200000, "Number of file descriptors used for Nginx. This is set in the OS with 'ulimit -n 200000'")
|
|
fs.BoolVar(&g.EnableEpool, "enable-epool", true, "essential for linux, optmized to serve many clients with each thread")
|
|
fs.BoolVar(&g.EnableMultiAccept, "enable-multi-accept", true, "Accept as many connections as possible, after nginx gets notification about a new connection.")
|
|
fs.StringVar(&g.ErrorLog, "error-log", "/dev/stderr crit", "only log critical errors")
|
|
fs.StringVar(&g.NginxUser, "nginx-user", "root", "nginx user name")
|
|
fs.IntVar(&g.KeepaliveRequests, "keepalive-requests", 100000, "Number of requests a client can make over the keep-alive connection. This is set high for testing.")
|
|
fs.IntVar(&g.KeepaliveTimeout, "keepalive-timeout", 30, "Timeout for keep-alive connections. Server will close connections after this time.")
|
|
fs.StringVar(&g.IP, "ip", "127.0.0.1", "Node ip.") // TODO: more detail
|
|
}
|
|
|
|
// SetLog sets log
|
|
func (g *GWServer) SetLog() {
|
|
level, err := logrus.ParseLevel(g.LogLevel)
|
|
if err != nil {
|
|
fmt.Println("set log level error." + err.Error())
|
|
return
|
|
}
|
|
logrus.SetLevel(level)
|
|
}
|
|
|
|
//CheckConfig check config
|
|
func (g *GWServer) CheckConfig() error {
|
|
if g.K8SConfPath == "" {
|
|
return fmt.Errorf("kube config file path can not be empty")
|
|
}
|
|
if exist, _ := util.FileExists(g.K8SConfPath); !exist {
|
|
return fmt.Errorf("kube config file %s not exist", g.K8SConfPath)
|
|
}
|
|
return nil
|
|
}
|