init

cmd/frabit-server/cmd/root.go

@@ -16,15 +16,26 @@
 package cmd
 
 import (
+	"context"
+	"fmt"
 	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/frabits/frabit/pkg/common/version"
+	"github.com/frabits/frabit/pkg/config"
+	"github.com/frabits/frabit/pkg/server"
 
 	"github.com/spf13/cobra"
 )
 
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
-	Use:   "frabit-server",
-	Short: "Frabit The next-generation database automatic platform",
+	Use:     "frabit-server",
+	Short:   "Frabit The next-generation database automatic platform",
+	RunE:    start,
+	Version: version.InfoStr.String(),
 }
 
 // Execute adds all child commands to the root command and sets flags appropriately.
@@ -36,6 +47,37 @@ func Execute() {
 	}
 }
 
+func start(cmd *cobra.Command, args []string) error {
+	ctx := context.Background()
+	cfg := config.Conf
+	srv := server.NewServer(cfg)
+
+	go listenToSystemSignals(ctx, srv)
+	return srv.Run()
+}
+
+func listenToSystemSignals(ctx context.Context, src *server.Server) {
+	signalChan := make(chan os.Signal, 1)
+	sighupChan := make(chan os.Signal, 1)
+
+	signal.Notify(sighupChan, syscall.SIGHUP)
+	signal.Notify(signalChan, os.Interrupt, syscall.SIGTERM)
+
+	for {
+		select {
+		case <-sighupChan:
+			fmt.Println("nothing,maybe reload config in the future")
+		case sig := <-signalChan:
+			ctx, cancel := context.WithTimeout(ctx, 30*time.Second)
+			defer cancel()
+			if err := src.Shutdown(ctx, fmt.Sprintf("System signal:%s", sig)); err != nil {
+				fmt.Fprintf(os.Stderr, "Timed out waiting for server to shut down")
+			}
+		}
+	}
+}
+
 func init() {
-	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+	rootCmd.Flags().BoolP("version", "v", false, "Display frabit-server version")
+	rootCmd.Flags().String("config", "/etc/frabit/frabit-server.yml", "Specific frabit config file")
 }