feat: add a service commnd

The following command will install atest as a Linux service. `atest service --action install`.

Set the default service port to be 7070 to avoid confliction

Author: LinuxSuRen

cmd/root.go

@@ -17,6 +17,7 @@ func NewRootCmd() (c *cobra.Command) {
 	c.Version = version.GetVersion()
 	c.AddCommand(createInitCommand(),
 		createRunCommand(), createSampleCmd(),
-		createServerCmd(), createJSONSchemaCmd())
+		createServerCmd(), createJSONSchemaCmd(),
+		createServiceCommand())
 	return
 }

cmd/server.go

@@ -19,7 +19,7 @@ func createServerCmd() (c *cobra.Command) {
 		RunE:  opt.runE,
 	}
 	flags := c.Flags()
-	flags.IntVarP(&opt.port, "port", "p", 9090, "The RPC server port")
+	flags.IntVarP(&opt.port, "port", "p", 7070, "The RPC server port")
 	flags.BoolVarP(&opt.printProto, "print-proto", "", false, "Print the proto content and exit")
 	return
 }

cmd/service.go

@@ -0,0 +1,54 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"runtime"
+
+	"github.com/spf13/cobra"
+)
+
+func createServiceCommand() (c *cobra.Command) {
+	opt := &serviceOption{}
+	c = &cobra.Command{
+		Use:     "service",
+		Aliases: []string{"s"},
+		Short:   "Install atest as a Linux service",
+		PreRunE: opt.preRunE,
+		RunE:    opt.runE,
+	}
+	flags := c.Flags()
+	flags.StringVarP(&opt.action, "action", "a", "", "The action of service, support actions: install")
+	return
+}
+
+type serviceOption struct {
+	action string
+}
+
+func (o *serviceOption) preRunE(c *cobra.Command, args []string) (err error) {
+	if runtime.GOOS != "linux" {
+		err = fmt.Errorf("only support on Linux")
+	}
+	return
+}
+
+func (o *serviceOption) runE(c *cobra.Command, args []string) (err error) {
+	switch o.action {
+	case "install", "i":
+		err = os.WriteFile("/lib/systemd/system/atest.service", []byte(script), os.ModeAppend)
+	default:
+		err = fmt.Errorf("not support action: '%s'", o.action)
+	}
+	return
+}
+
+var script = `[Unit]
+Description=API Testing
+
+[Service]
+ExecStart=atest server
+
+[Install]
+WantedBy=multi-user.target
+`

cmd/service_test.go

@@ -0,0 +1,14 @@
+package cmd
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestService(t *testing.T) {
+	root := NewRootCmd()
+	root.SetArgs([]string{"service", "--action", "fake"})
+	err := root.Execute()
+	assert.NotNil(t, err)
+}

sample/manifest.yaml

@@ -37,9 +37,9 @@ metadata:
 spec:
   ports:
   - name: server
-    port: 9090
+    port: 7070
     protocol: TCP
-    targetPort: 9090
+    targetPort: 7070
   selector:
     app: api-testing
   sessionAffinity: None