Initial commit

cmd/controller-manager/controller-manager.go

@@ -0,0 +1,87 @@
+/*
+Copyright 2019 The KubeSphere authors.
+
+Licensed 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 (
+	"flag"
+	"os"
+
+	_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
+	"sigs.k8s.io/controller-runtime/pkg/client/config"
+	"sigs.k8s.io/controller-runtime/pkg/manager"
+	logf "sigs.k8s.io/controller-runtime/pkg/runtime/log"
+	"sigs.k8s.io/controller-runtime/pkg/runtime/signals"
+
+	"kubesphere.io/kubesphere/pkg/apis"
+	"kubesphere.io/kubesphere/pkg/controller"
+	"kubesphere.io/kubesphere/pkg/webhook"
+)
+
+func main() {
+	var metricsAddr string
+	flag.StringVar(&metricsAddr, "metrics-addr", ":8080", "The address the metric endpoint binds to.")
+	flag.Parse()
+	logf.SetLogger(logf.ZapLogger(false))
+	log := logf.Log.WithName("entrypoint")
+
+	// Get a config to talk to the apiserver
+	log.Info("setting up client for manager")
+	cfg, err := config.GetConfig()
+	if err != nil {
+		log.Error(err, "unable to set up client config")
+		os.Exit(1)
+	}
+
+	// Create a new Cmd to provide shared dependencies and start components
+	log.Info("setting up manager")
+	mgr, err := manager.New(cfg, manager.Options{})
+	if err != nil {
+		log.Error(err, "unable to set up overall controller manager")
+		os.Exit(1)
+	}
+
+	log.Info("Registering Components.")
+
+	// Setup Scheme for all resources
+	log.Info("setting up scheme")
+	if err := apis.AddToScheme(mgr.GetScheme()); err != nil {
+		log.Error(err, "unable add APIs to scheme")
+		os.Exit(1)
+	}
+
+	// Setup all Controllers
+	log.Info("Setting up controller")
+	if err := controller.AddToManager(mgr); err != nil {
+		log.Error(err, "unable to register controllers to the manager")
+		os.Exit(1)
+	}
+
+	log.Info("setting up webhooks")
+	if err := webhook.AddToManager(mgr); err != nil {
+		log.Error(err, "unable to register webhooks to the manager")
+		os.Exit(1)
+	}
+
+	controller.Run(signals.SetupSignalHandler())
+
+	// Start the Cmd
+	log.Info("Starting the Cmd.")
+	if err := mgr.Start(signals.SetupSignalHandler()); err != nil {
+		log.Error(err, "unable to run the manager")
+		os.Exit(1)
+	}
+}

cmd/ks-apiserver/apiserver.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	"fmt"
+	"kubesphere.io/kubesphere/cmd/ks-apiserver/app"
+	"os"
+
+	// Install apis
+	_ "kubesphere.io/kubesphere/pkg/apis/metrics/install"
+	_ "kubesphere.io/kubesphere/pkg/apis/operations/install"
+	_ "kubesphere.io/kubesphere/pkg/apis/resources/install"
+)
+
+func main() {
+
+	cmd := app.NewAPIServerCommand()
+
+	if err := cmd.Execute(); err != nil {
+		fmt.Fprintf(os.Stderr, "error: %v\n", err)
+		os.Exit(1)
+	}
+}

cmd/ks-apiserver/app/options/options.go

@@ -0,0 +1,67 @@
+package options
+
+import (
+	"fmt"
+	"github.com/spf13/pflag"
+	"os"
+)
+
+type ServerRunOptions struct {
+
+	// server bind address
+	BindAddress string
+
+	// insecure port number
+	InsecurePort int
+
+	// secure port number
+	SecurePort int
+
+	// OpenPitrix api gateway service url
+	OpenPitrixAddress string
+
+	// database connection string in MySQL like
+	// user:password@tcp(host)/dbname?charset=utf8&parseTime=True
+	DatabaseConnectionString string
+
+	// tls cert file
+	TlsCertFile string
+
+	// tls private key file
+	TlsPrivateKey string
+
+	// host openapi doc
+	ApiDoc bool
+
+	// kubeconfig file path
+	KubeConfig string
+}
+
+func NewServerRunOptions() *ServerRunOptions {
+	// create default server run options
+	s := ServerRunOptions{
+		BindAddress:              "0.0.0.0",
+		InsecurePort:             9090,
+		SecurePort:               0,
+		OpenPitrixAddress:        "openpitrix-api-gateway.openpitrix-system.svc",
+		DatabaseConnectionString: "",
+		TlsCertFile:              "",
+		TlsPrivateKey:            "",
+		ApiDoc:                   true,
+	}
+
+	return &s
+}
+
+func (s *ServerRunOptions) AddFlags(fs *pflag.FlagSet) {
+
+	fs.StringVar(&s.BindAddress, "bind-address", "0.0.0.0", "server bind address")
+	fs.IntVar(&s.InsecurePort, "insecure-port", 9090, "insecure port number")
+	fs.IntVar(&s.SecurePort, "secure-port", 0, "secure port number")
+	fs.StringVar(&s.OpenPitrixAddress, "openpitrix", "openpitrix-api-gateway.openpitrix-system.svc", "openpitrix api gateway address")
+	fs.StringVar(&s.DatabaseConnectionString, "database-connection", "", "database connection string")
+	fs.StringVar(&s.TlsCertFile, "tls-cert-file", "", "tls cert file")
+	fs.StringVar(&s.TlsPrivateKey, "tls-private-key", "", "tls private key")
+	fs.BoolVar(&s.ApiDoc, "api-doc", true, "host OpenAPI doc")
+	fs.StringVar(&s.KubeConfig, "kubeconfig", fmt.Sprintf("%s/.kube/config", os.Getenv("HOME")), "path to kubeconfig file")
+}

cmd/ks-apiserver/app/server.go

@@ -0,0 +1,73 @@
+package app
+
+import (
+	"fmt"
+	"github.com/emicklei/go-restful-openapi"
+	"github.com/spf13/cobra"
+	"kubesphere.io/kubesphere/cmd/ks-apiserver/app/options"
+	"kubesphere.io/kubesphere/pkg/apiserver/runtime"
+	"kubesphere.io/kubesphere/pkg/client"
+	"kubesphere.io/kubesphere/pkg/filter"
+	"kubesphere.io/kubesphere/pkg/informers"
+	"kubesphere.io/kubesphere/pkg/signals"
+	"log"
+	"net/http"
+)
+
+func NewAPIServerCommand() *cobra.Command {
+	s := options.NewServerRunOptions()
+
+	cmd := &cobra.Command{
+		Use: "ks-apiserver",
+		Long: `The KubeSphere API server validates and configures data
+for the api objects. The API Server services REST operations and provides the frontend to the
+cluster's shared state through which all other components interact.`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+
+			//s.AddFlags(cmd.Flags())
+
+			return Run(s)
+		},
+	}
+
+	s.AddFlags(cmd.Flags())
+
+	return cmd
+}
+
+func Run(s *options.ServerRunOptions) error {
+
+	var err error
+
+	stopChan := signals.SetupSignalHandler()
+	informers.SharedInformerFactory().Start(stopChan)
+	informers.SharedInformerFactory().WaitForCacheSync(stopChan)
+	log.Println("resources sync success")
+
+	container := runtime.Container
+	container.Filter(filter.Logging)
+
+	if len(s.KubeConfig) > 0 {
+		client.KubeConfigFile = s.KubeConfig
+	}
+
+	if s.ApiDoc {
+		config := restfulspec.Config{
+			WebServices: container.RegisteredWebServices(),
+			APIPath:     "/apidoc.json",
+		}
+		container.Add(restfulspec.NewOpenAPIService(config))
+	}
+
+	log.Printf("Server listening on %d.", s.InsecurePort)
+
+	if s.InsecurePort != 0 {
+		err = http.ListenAndServe(fmt.Sprintf("%s:%d", s.BindAddress, s.InsecurePort), container)
+	}
+
+	if s.SecurePort != 0 && len(s.TlsCertFile) > 0 && len(s.TlsPrivateKey) > 0 {
+		err = http.ListenAndServeTLS(fmt.Sprintf("%s:%d", s.BindAddress, s.SecurePort), s.TlsCertFile, s.TlsPrivateKey, container)
+	}
+
+	return err
+}