add tcp proxy support

Author: LinuxSuRen

cmd/mock-compose.go

@@ -17,52 +17,49 @@ 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) {
-	c = &cobra.Command{
-		Use:   "mock-compose",
-		Short: "Mock multiple servers",
-		Args:  cobra.ExactArgs(1),
-		RunE: func(cmd *cobra.Command, args []string) (err error) {
-			reader := mock.NewLocalFileReader(args[0])
+    c = &cobra.Command{
+        Use:   "mock-compose",
+        Short: "Mock multiple servers",
+        Args:  cobra.ExactArgs(1),
+        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)
+                go subServer.Start(subReader, proxy.Prefix)
+                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

@@ -38,7 +38,7 @@ docker pull localhost:6060/repo/name:tag
 * 面向对象的 CRUD
 * 自定义 HTTP 服务
 
-### 面对对象
+### 面向对象
 
 ```yaml
 #!api-testing-mock
@@ -160,6 +160,16 @@ proxies:
     target: http://192.168.123.58:9200
 ```
 
+## TCP 协议代理
+
+```yaml
+proxies:
+  - protocol: tcp
+    port: 3306
+    path: /
+    target: 192.168.123.58:33060
+```
+
 ## 代理多个服务
 
 ```shell
@@ -178,6 +188,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