modify valid

Author: ma5d

src/main/java/org/ma5d/javamock/controller/MockController.java

@@ -1,8 +1,10 @@
 package org.ma5d.javamock.controller;
 
 import jakarta.annotation.Resource;
+import jakarta.servlet.http.HttpServletRequest;
 import org.ma5d.javamock.dto.SaveParam;
 import org.ma5d.javamock.service.JavaMockServiceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.*;
@@ -14,8 +16,15 @@
 public class MockController {
     @Resource
     JavaMockServiceImpl javaMockService;
+    @Autowired
+    private JavaMockServiceImpl javaMockServiceImpl;
 
     @GetMapping("/")
+    public String index() {
+        return "redirect:/index";
+    }
+
+    @GetMapping("/index")
     public String index(Model model) throws SQLException {
         List<SaveParam> saveParams = javaMockService.queryAllLines();
         model.addAttribute("saveParams", saveParams);
@@ -35,4 +44,22 @@ public String detail(@PathVariable String timeStamp, Model model) throws SQLExce
         return "detail";
     }
 
+    @ResponseBody
+    @GetMapping("/p/**")
+    public String proxyGet(HttpServletRequest request) throws SQLException {
+        String requestURI = request.getRequestURI();
+        String pathWithParam = "/" + requestURI.split("/p/")[1];
+        SaveParam saveParam = javaMockServiceImpl.getSaveParamByPathWithParam(pathWithParam);
+        return saveParam.getResponse();
+    }
+
+    @ResponseBody
+    @PostMapping("/p/**")
+    public String proxyPost(HttpServletRequest request) throws SQLException {
+        String requestURI = request.getRequestURI();
+        String pathWithParam = "/" + requestURI.split("/p/")[1];
+        SaveParam saveParam = javaMockServiceImpl.getSaveParamByPathWithParam(pathWithParam);
+        return saveParam.getResponse();
+    }
+
 }

src/main/java/org/ma5d/javamock/dto/SaveParam.java

@@ -1,17 +1,24 @@
 package org.ma5d.javamock.dto;
 
 import lombok.Data;
+import lombok.NoArgsConstructor;
 
 import java.io.Serializable;
+import java.sql.ResultSet;
+import java.sql.SQLException;
 
 @Data
+@NoArgsConstructor
 public class SaveParam implements Serializable {
     private String timeStamp;
     private String domainCom;
     private String pathWithParam;
     private String response;
 
-    public String[] toStringArray(){
-        return new String[]{timeStamp, domainCom, pathWithParam, response};
+    public SaveParam(ResultSet resultSet) throws SQLException {
+        this.timeStamp = resultSet.getString("time_stamp");
+        this.domainCom = resultSet.getString("adomain");
+        this.pathWithParam = resultSet.getString("path_param");
+        this.response = resultSet.getString("response");
     }
 }

src/main/java/org/ma5d/javamock/service/JavaMockServiceImpl.java

@@ -22,7 +22,7 @@ public boolean saveConfig(SaveParam saveParam) throws SQLException {
         Connection connection = dataSource.getConnection();
         // language=SQLite
         String query = "insert into java_mock(time_stamp, adomain, path_param, response) values(?, ?, ?, ?)";
-        PreparedStatement stmt = connection.prepareStatement(query, saveParam.toStringArray());
+        PreparedStatement stmt = connection.prepareStatement(query);
         stmt.setString(1, saveParam.getTimeStamp());
         stmt.setString(2, saveParam.getDomainCom());
         stmt.setString(3, saveParam.getPathWithParam());
@@ -37,11 +37,7 @@ public List<SaveParam> queryAllLines() throws SQLException {
         ResultSet resultSet = connection.prepareStatement(query).executeQuery();
         List<SaveParam> saveParams = new ArrayList<>();
         while (resultSet.next()) {
-            SaveParam saveParam = new SaveParam();
-            saveParam.setTimeStamp(resultSet.getString("time_stamp"));
-            saveParam.setDomainCom(resultSet.getString("adomain"));
-            saveParam.setPathWithParam(resultSet.getString("path_param"));
-            saveParam.setResponse(resultSet.getString("response"));
+            SaveParam saveParam = new SaveParam(resultSet);
             saveParams.add(saveParam);
         }
         connection.close();
@@ -52,13 +48,27 @@ public SaveParam queryLine(String timeStamp) throws SQLException {
         Connection connection = dataSource.getConnection();
         // language=SQLite
         String query = "select time_stamp, adomain, path_param, response from java_mock where time_stamp = ?";
-        ResultSet resultSet = connection.prepareStatement(query, new String[]{timeStamp}).executeQuery();
+        PreparedStatement preparedStatement = connection.prepareStatement(query);
+        preparedStatement.setString(1, timeStamp);
+        ResultSet resultSet = preparedStatement.executeQuery();
         SaveParam saveParam = new SaveParam();
         while (resultSet.next()) {
-            saveParam.setTimeStamp(resultSet.getString("time_stamp"));
-            saveParam.setDomainCom(resultSet.getString("adomain"));
-            saveParam.setPathWithParam(resultSet.getString("path_param"));
-            saveParam.setResponse(resultSet.getString("response"));
+            saveParam = new SaveParam(resultSet);
+        }
+        connection.close();
+        return saveParam;
+    }
+
+    public SaveParam getSaveParamByPathWithParam(String pathWithParam) throws SQLException {
+        Connection connection = dataSource.getConnection();
+        // language=SQLite
+        String query = "select time_stamp, adomain, path_param, response from java_mock where path_param = ?";
+        PreparedStatement preparedStatement = connection.prepareStatement(query);
+        preparedStatement.setString(1, pathWithParam);
+        ResultSet resultSet = preparedStatement.executeQuery();
+        SaveParam saveParam = new SaveParam();
+        while (resultSet.next()) {
+            saveParam = new SaveParam(resultSet);
         }
         connection.close();
         return saveParam;

src/main/resources/templates/index.html

@@ -2,6 +2,16 @@
 <html xmlns:th="http://www.thymeleaf.org" lang="zh">
 <head>
     <title>Form Demo</title>
+    <style>
+        table {
+            border: 1px solid black;
+            border-collapse: collapse;
+        }
+        td, th {
+            border: 1px solid black;
+            padding: 8px; /* 可选：为单元格增加内边距 */
+        }
+    </style>
 </head>
 <body>
 <table>
@@ -11,22 +21,34 @@
         <th>域名</th>
         <th>路径及参数</th>
         <th>返回数据</th>
-        <th>操作</th>
     </tr>
     </thead>
     <tbody>
     <tr th:each="saveParam : ${saveParams}">
-        <td th:text="${saveParam.timeStamp}"></td>
-        <td th:text="${saveParam.domainCom}"></td>
-        <td th:text="${saveParam.pathWithParam}"></td>
-        <td th:text="${saveParam.response}"></td>
+
         <td>
             <a th:href="@{http://{domainCom}/detail/{timestamp}(domainCom=${saveParam.domainCom}, timestamp=${saveParam.timeStamp})}"
-               th:text="'访问 ' + ${saveParam.timeStamp}"></a>
+                    th:text="${saveParam.timeStamp}"></a>
         </td>
+
+        <td th:text="${saveParam.domainCom}"></td>
+
+        <td>
+            <a th:href="@{http://{domainCom}/p{pathWithParam}(domainCom=${saveParam.domainCom}, pathWithParam=${saveParam.pathWithParam})}"
+               th:text="${saveParam.pathWithParam}">
+            </a>
+        </td>
+
+        <td th:text="${#strings.length(saveParam.response) > 100 ? saveParam.response.substring(0, 100) + '...' : saveParam.response}"></td>
+
     </tr>
     </tbody>
 </table>
+<a href="/new" style="text-decoration: none;">
+    <div style="display: inline-block; padding: 8px 16px; background-color: #4CAF50; color: white; border: none; border-radius: 4px; cursor: pointer;">
+        新增
+    </div>
+</a>
 
 </body>
 </html>