refactor: mid refactor

2 files changed, 28 insertions, 7 deletions

diff --git a/cmd/server/main.go b/cmd/server/main.go
new file mode 100644
index 0000000..e9cb4ad
--- /dev/null
+++ b/cmd/server/main.go
@@ -0,0 +1,28 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/kyren223/eko/internal/server"
+)
+
+const port = 7223
+
+func main() {
+	server := server.NewServer(port)
+
+	ctx, cancel := context.WithCancel(context.Background())
+	signalChan := make(chan os.Signal, 1)
+	signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		signal := <-signalChan
+		log.Println("signal:", signal.String())
+		cancel()
+	}()
+
+	server.ListenAndServe(ctx)
+}
diff --git a/cmd/server/server.go b/cmd/server/server.go
deleted file mode 100644
index 168264a..0000000
--- a/cmd/server/server.go
+++ /dev/null
@@ -1,7 +0,0 @@
-package main
-
-import "github.com/kyren223/eko/internal/server"
-
-func main() {
-	server.Start()
-}