add tcp proxy support

Author: LinuxSuRen

cmd/mock-compose.go

@@ -17,12 +17,11 @@ limitations under the License.
 package cmd
 
 import (
-	"os"
-	"os/signal"
-	"syscall"
-
-	"github.com/linuxsuren/api-testing/pkg/mock"
-	"github.com/spf13/cobra"
+    "github.com/linuxsuren/api-testing/pkg/mock"
+    "github.com/spf13/cobra"
+    "os"
+    "os/signal"
+    "syscall"
 )
 
 func createMockComposeCmd() (c *cobra.Command) {
@@ -33,36 +32,36 @@ func createMockComposeCmd() (c *cobra.Command) {
 		RunE: func(cmd *cobra.Command, args []string) (err error) {
 			reader := mock.NewLocalFileReader(args[0])
 
-			var server *mock.Server
-			if server, err = reader.Parse(); err != nil {
-				return
-			}
+            var server *mock.Server
+            if server, err = reader.Parse(); err != nil {
+                return
+            }
 
-			var subServers []mock.DynamicServer
-			for _, proxy := range server.Proxies {
-				subProxy := &mock.Server{
-					Proxies: []mock.Proxy{proxy},
-				}
+            var subServers []mock.DynamicServer
+            for _, proxy := range server.Proxies {
+                subProxy := &mock.Server{
+                    Proxies: []mock.Proxy{proxy},
+                }
 
-				subReader := mock.NewObjectReader(subProxy)
-				subServer := mock.NewInMemoryServer(c.Context(), proxy.Port)
-				if err = subServer.Start(subReader, proxy.Prefix); err != nil {
-					return
-				}
-				subServers = append(subServers, subServer)
-			}
+                subReader := mock.NewObjectReader(subProxy)
+                subServer := mock.NewInMemoryServer(c.Context(), proxy.Port)
+                if err = subServer.Start(subReader, proxy.Prefix); err != nil {
+                    return
+                }
+                subServers = append(subServers, subServer)
+            }
 
-			clean := make(chan os.Signal, 1)
-			signal.Notify(clean, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
-			select {
-			case <-c.Context().Done():
-			case <-clean:
-			}
-			for _, server := range subServers {
-				server.Stop()
-			}
-			return
-		},
-	}
-	return
+            clean := make(chan os.Signal, 1)
+            signal.Notify(clean, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
+            select {
+            case <-c.Context().Done():
+            case <-clean:
+            }
+            for _, server := range subServers {
+                server.Stop()
+            }
+            return
+        },
+    }
+    return
 }

docs/site/content/zh/latest/tasks/mock.md

@@ -178,6 +178,12 @@ proxies:
     port: 17001
     path: /{path:.*}
     target: http://192.168.123.58:17001
+  - protocol: tcp
+    port: 33060
+    path: /
+    target: 192.168.123.58:33060
 ```
 
+当前代理支持 HTTP 和 TCP 协议，上面的例子中代理了 MySQL 的 `33060` 端口。
+
 > 更多 URL 中通配符的用法，请参考 https://github.com/gorilla/mux