支持输出 mock 日志

Author: LinuxSuRen

console/atest-ui/src/views/MockManager.vue

@@ -21,6 +21,15 @@ API.GetSchema('mock').then((schema) => {
     }
 });
 
+const logOutput = ref('');
+API.GetStream((stream) => {
+    try {
+        const data = JSON.parse(stream);
+        logOutput.value += `${data.result.message}`;
+        console.log('Received schema:', logOutput.value);
+    } catch (e) {}
+});
+
 interface MockConfig {
   Config: string
   ConfigAsJSON: string
@@ -107,6 +116,14 @@ items:
                     :extensions="[jsonSchema(mockschema), jsonComplete, headerComplete]" />
             </el-tab-pane>
         </el-tabs>
+        <el-card class="log-output" shadow="hover">
+          <template #header>
+            <span>{{ t('label.logs') }}</span>
+          </template>
+          <el-scrollbar height="200px" ref="logScrollbar">
+            <pre style="white-space: pre-wrap; word-break: break-all;">{{ logOutput }}</pre>
+          </el-scrollbar>
+        </el-card>
     </div>
 </template>
 

console/atest-ui/src/views/net.ts

@@ -733,7 +733,38 @@ function GetMockConfig(callback: (d: any) => void) {
         .then(DefaultResponseProcess)
         .then(callback)
 }
-
+function GetStream(callback: (d: any) => void, errHandle?: (e: any) => void | null) {
+    const requestOptions = {
+        headers: {
+            'X-Auth': getToken()
+        }
+    }
+    fetch(`/api/v1/mock/log`, {
+        ...requestOptions,
+        headers: {
+            ...requestOptions.headers,
+            'Accept': 'text/event-stream'
+        }
+    })
+    .then((response: any) => {
+        if (response.ok && response.body) {
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            const read = () => {
+                reader.read().then(({ done, value }) => {
+                    if (done) return;
+                    const chunk = decoder.decode(value, { stream: true });
+                    callback(chunk);
+                    read();
+                });
+            };
+            read();
+        } else {
+            return DefaultResponseProcess(response);
+        }
+    })
+    .catch(emptyOrDefault(errHandle));
+}
 function getToken() {
     const token = sessionStorage.getItem('token')
     if (!token) {
@@ -944,7 +975,7 @@ export const API = {
     FunctionsQuery,
     GetSecrets, DeleteSecret, CreateOrUpdateSecret,
     GetSuggestedAPIs, GetSwaggers,
-    ReloadMockServer, GetMockConfig, SBOM, DataQuery, DataQueryAsync,
+    ReloadMockServer, GetMockConfig, GetStream, SBOM, DataQuery, DataQueryAsync,
     GetThemes, GetTheme, GetBinding,
     getToken
 }

pkg/mock/in_memory.go

@@ -89,6 +89,13 @@ func (s *inMemoryServer) WithTLS(certFile, keyFile string) DynamicServer {
 	return s
 }
 
+func (s *inMemoryServer) WithLogWriter(writer io.Writer) DynamicServer {
+	if writer != nil {
+		memLogger = memLogger.WithNameAndWriter("stream", writer)
+	}
+	return s
+}
+
 func (s *inMemoryServer) GetTLS() (string, string) {
 	return s.certFile, s.keyFile
 }
@@ -268,7 +275,7 @@ func (s *inMemoryServer) EnableMetrics() {
 func (s *inMemoryServer) startObject(obj Object) {
 	// create a simple CRUD server
 	s.mux.HandleFunc("/"+obj.Name, func(w http.ResponseWriter, req *http.Request) {
-		fmt.Println("mock server received request", req.URL.Path)
+		memLogger.Info("mock server received request", "path", req.URL.Path)
 		s.metrics.RecordRequest(req.URL.Path)
 		method := req.Method
 		w.Header().Set(util.ContentType, util.JSON)

pkg/mock/server.go

@@ -16,6 +16,7 @@ limitations under the License.
 package mock
 
 import (
+	"io"
 	"net/http"
 )
 
@@ -27,6 +28,7 @@ type DynamicServer interface {
 	Start(reader Reader, prefix string) error
 	SetupHandler(reader Reader, prefix string) (http.Handler, error)
 	WithTLS(certFile, keyFile string) DynamicServer
+	WithLogWriter(writer io.Writer) DynamicServer
 	GetTLS() (certFile, keyFile string)
 	Stop() error
 	GetPort() string

pkg/server/remote_server.go

@@ -1377,6 +1377,7 @@ type mockServerController struct {
 	mockWriter  mock.ReaderAndWriter
 	loader      mock.Loadable
 	reader      mock.Reader
+	logData     chan string
 	prefix      string
 	combinePort int
 }
@@ -1386,6 +1387,7 @@ func NewMockServerController(mockWriter mock.ReaderAndWriter, loader mock.Loadab
 		mockWriter:  mockWriter,
 		loader:      loader,
 		prefix:      "/mock/server",
+		logData:     make(chan string, 100),
 		combinePort: combinePort,
 	}
 }
@@ -1404,6 +1406,7 @@ func (s *mockServerController) Reload(ctx context.Context, in *MockConfig) (repl
 
 		server := mock.NewInMemoryServer(ctx, int(in.GetPort())).WithTLS(dServer.GetTLS())
 		server.Start(s.mockWriter, in.Prefix)
+		server.WithLogWriter(s)
 		s.loader = server
 	}
 	err = s.loader.Load()
@@ -1421,6 +1424,22 @@ func (s *mockServerController) GetConfig(ctx context.Context, in *Empty) (reply
 	}
 	return
 }
+func (s *mockServerController) LogWatch(e *Empty, logServer Mock_LogWatchServer) (err error) {
+	for msg := range s.logData {
+		logServer.Send(&CommonResult{
+			Success: true,
+			Message: msg,
+		})
+	}
+	return
+}
+func (s *mockServerController) Write(p []byte) (n int, err error) {
+	select {
+	case s.logData <- fmt.Sprintf("%s: %s", time.Now().Format(time.RFC3339), string(p)):
+	default:
+	}
+	return
+}
 
 func (s *server) getLoaderByStoreName(storeName string) (loader testing.Writer, err error) {
 	var store *testing.Store