Feat: done tcp

cmds/client.go

@@ -0,0 +1,45 @@
+package cmds
+
+import (
+	"fmt"
+	"inp2025/tcp"
+
+	"github.com/spf13/cobra"
+)
+
+func ping() {
+	msgs := []string{"hello, world", "goodbye"}
+	for _, msg := range msgs {
+		fmt.Printf("client sending: '%s'\n", msg)
+
+		b, err := tcp.Post(":8080", "/", []byte(msg))
+		if err != nil {
+			panic(err)
+		}
+
+		fmt.Printf("server reply: '%s'\n", string(b))
+	}
+}
+
+func pull() {
+	socket, err := tcp.Dial(":8080", "/")
+	if err != nil {
+		panic(err)
+	}
+	defer socket.Shutdown()
+	for i := 0; i < 5; i++ {
+		b, err := socket.Read()
+		if err != nil {
+			panic(err)
+		}
+
+		fmt.Printf("server send: '%s'\n", string(b))
+	}
+}
+
+var clientCmd = &cobra.Command{
+	Use: "client",
+	Run: func(cmd *cobra.Command, args []string) {
+		pull()
+	},
+}

cmds/root.go

@@ -0,0 +1,34 @@
+package cmds
+
+import (
+	"strings"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"go.uber.org/zap"
+)
+
+var RootCmd = &cobra.Command{
+	Use: "lab4",
+	PersistentPreRun: func(cmd *cobra.Command, args []string) {
+		viper.AutomaticEnv()
+		viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
+		viper.BindPFlags(cmd.PersistentFlags())
+		viper.BindPFlags(cmd.Flags())
+
+		cfg := zap.NewProductionConfig()
+		logger, err := cfg.Build()
+		if err != nil {
+			panic(err)
+		}
+
+		zap.ReplaceGlobals(logger)
+	},
+}
+
+func init() {
+	cobra.EnableTraverseRunHooks = true
+
+	RootCmd.AddCommand(serverCmd)
+	RootCmd.AddCommand(clientCmd)
+}

cmds/server.go

@@ -0,0 +1,40 @@
+package cmds
+
+import (
+	"fmt"
+	"inp2025/middlewares"
+	"inp2025/tcp"
+	"io"
+	"time"
+
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+func pongHandler(w tcp.ResponseWriter, req *tcp.Request) error {
+	w.WriteHeader(tcp.StatusOK)
+	_, err := io.WriteString(w, string(req.Body))
+	return err
+}
+
+func tickHandler(w tcp.ResponseWriter, req *tcp.Request) error {
+	zap.L().Info("Run tickHandler")
+	for {
+		w.SocketSendString(fmt.Sprintf("time: %s", time.Now().String()))
+		time.Sleep(time.Second)
+	}
+	return nil
+}
+
+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.Listen(":8080")
+	},
+}