Feat: add query params

cmds/client.go

@@ -12,17 +12,17 @@ func ping() {
 	for _, msg := range msgs {
 		fmt.Printf("client sending: '%s'\n", msg)
 
-		b, err := tcp.Post(":8080", "/", []byte(msg))
+		resp, err := tcp.Get(":8080", "/test", map[string]string{"msg": msg})
 		if err != nil {
 			panic(err)
 		}
 
-		fmt.Printf("server reply: '%s'\n", string(b))
+		fmt.Printf("server reply: '%s'\n", string(resp.Body))
 	}
 }
 
 func pull() {
-	socket, err := tcp.Dial(":8080", "/")
+	socket, err := tcp.Dial(":8080", "/test")
 	if err != nil {
 		panic(err)
 	}
@@ -40,6 +40,6 @@ func pull() {
 var clientCmd = &cobra.Command{
 	Use: "client",
 	Run: func(cmd *cobra.Command, args []string) {
-		pull()
+		ping()
 	},
 }

cmds/database.go

@@ -0,0 +1,39 @@
+package cmds
+
+import (
+	"database/sql"
+	"inp2025/handlers/database"
+	"inp2025/middlewares"
+	"inp2025/tcp"
+	"inp2025/utils"
+
+	"github.com/spf13/cobra"
+	"github.com/uptrace/bun"
+	"github.com/uptrace/bun/dialect/sqlitedialect"
+	"github.com/uptrace/bun/driver/sqliteshim"
+)
+
+var databaseCmd = &cobra.Command{
+	Use: "database",
+	Run: func(cmd *cobra.Command, args []string) {
+		router := tcp.NewRouter().
+			Use(middlewares.ErrorHandler).
+			Use(middlewares.AccessLog)
+
+		sqldb, err := sql.Open(sqliteshim.ShimName, "file::memory:?cache=shared")
+		if err != nil {
+			panic(err)
+		}
+		db := bun.NewDB(sqldb, sqlitedialect.New())
+		utils.InitDB(db)
+
+		handlers := database.NewHandlers(db)
+
+		router.GET("/users", handlers.GetUsers)
+	},
+}
+
+func init() {
+	databaseCmd.Flags().
+		String("port", "5432", "")
+}

cmds/server.go

@@ -8,12 +8,13 @@ import (
 	"time"
 
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 	"go.uber.org/zap"
 )
 
 func pongHandler(w tcp.ResponseWriter, req *tcp.Request) error {
 	w.WriteHeader(tcp.StatusOK)
-	_, err := io.WriteString(w, string(req.Body))
+	_, err := io.WriteString(w, string(req.Params["msg"]))
 	return err
 }
 
@@ -29,12 +30,20 @@ func tickHandler(w tcp.ResponseWriter, req *tcp.Request) error {
 var serverCmd = &cobra.Command{
 	Use: "server",
 	Run: func(cmd *cobra.Command, args []string) {
-		router := tcp.NewRouter()
-		router.Use(middlewares.ErrorHandler)
-		router.Use(middlewares.AccessLog)
-		router.Register(tcp.MethodSOCKET, "/", tickHandler)
-		router.Register(tcp.MethodPOST, "/", pongHandler)
+		router := tcp.NewRouter().
+			Use(middlewares.ErrorHandler).
+			Use(middlewares.AccessLog)
 
-		router.Listen(":8080")
+		router.Register(tcp.MethodSOCKET, "/test",
+			tickHandler)
+		router.Register(tcp.MethodGET, "/test",
+			pongHandler)
+
+		router.Listen(":" + viper.GetString("port"))
 	},
 }
+
+func init() {
+	serverCmd.Flags().
+		String("port", "8080", "")
+}