feature: Test run parameter

limanmys · Jan 12, 2024 · 14829b9 · 14829b9
1 parent ce4ecf8
commit 14829b9
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 12 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"log"
 
 	_ "github.com/joho/godotenv/autoload"
@@ -17,6 +18,9 @@ import (
 )
 
 func main() {
+	runType := flag.String("type", "admin", "Server's run type.")
+	flag.Parse()
+
 	// Migrate tables
 	if !fiber.IsChild() {
 		//Migrate tables
@@ -46,6 +50,11 @@ func main() {
 	// Mount routes
 	routes.Routes(app)
 
-	// Start server
-	log.Fatal(app.Listen("127.0.0.1:7806"))
+	if *runType == "admin" {
+		// Start server
+		log.Fatal(app.Listen("127.0.0.1:7806"))
+	} else if *runType == "test" {
+		// Start test server
+		log.Fatal(app.Listen("0.0.0.0:7825"))
+	}
 }
diff --git a/scripts/debian/postinst b/scripts/debian/postinst
@@ -7,11 +7,11 @@ fi
 
 echo """
 [Unit]
-Description=Inventory Server
+Description=Inventory Server (%I)
 [Service]
 Type=simple
 WorkingDirectory=/opt/inventory-server
-ExecStart=/opt/inventory-server/inventory-server
+ExecStart=/opt/inventory-server/inventory-server -type=%i
 Restart=always
 RestartSec=10
 SyslogIdentifier=inventory
@@ -20,8 +20,8 @@ User=root
 Group=root
 [Install]
 WantedBy=multi-user.target
-    """ > /etc/systemd/system/inventory-server.service
+    """ > /etc/systemd/system/inventory-server@.service
 
 systemctl daemon-reload
-systemctl enable inventory-server.service
-systemctl restart inventory-server.service
+systemctl enable inventory-server@admin.service
+systemctl restart inventory-server@admin.service
diff --git a/scripts/redhat/inventory-server.spec b/scripts/redhat/inventory-server.spec
@@ -29,11 +29,11 @@ fi
 
 echo """
 [Unit]
-Description=Inventory Server
+Description=Inventory Server (%I)
 [Service]
 Type=simple
 WorkingDirectory=/opt/inventory-server
-ExecStart=/opt/inventory-server/inventory-server
+ExecStart=/opt/inventory-server/inventory-server -type=%i
 Restart=always
 RestartSec=10
 SyslogIdentifier=inventory
@@ -42,11 +42,11 @@ User=root
 Group=root
 [Install]
 WantedBy=multi-user.target
-    """ > /etc/systemd/system/inventory-server.service
+    """ > /etc/systemd/system/inventory-server@.service
 
 systemctl daemon-reload
-systemctl enable inventory-server.service
-systemctl restart inventory-server.service
+systemctl enable inventory-server@admin.service
+systemctl restart inventory-server@admin.service
 
 %clean