union-ark-web 1.0.1

package/ark-control/src/main/java/com/union/control/utils/AgentSupport.java

@@ -0,0 +1,103 @@
+package com.union.control.utils;
+
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.regex.Pattern;
+
+public final class AgentSupport {
+    private static final Pattern ID = Pattern.compile("[A-Za-z0-9._:@-]{1,64}");
+
+    private AgentSupport() {}
+
+    public static Map<String, Object> request(ObjectMapper json, String input) {
+        if (input == null) throw new IllegalArgumentException("请求参数非法");
+        try {
+            Map<String, Object> value = json.readValue(
+                    input, new TypeReference<Map<String, Object>>() {});
+            if (value == null) throw new IllegalArgumentException("请求参数非法");
+            return value;
+        } catch (IllegalArgumentException error) {
+            throw error;
+        } catch (Exception error) {
+            throw new IllegalArgumentException("请求参数不是合法 JSON", error);
+        }
+    }
+
+    public static String userId(Map<String, Object> request) {
+        return identity(request, "userId");
+    }
+
+    public static String identity(Map<String, Object> request, String field) {
+        String value = text(request, field, 64, true);
+        if (value == null || value.trim().isEmpty())
+            throw new IllegalArgumentException(field + " 非法");
+        return value;
+    }
+
+    public static void requireId(String value) {
+        if (value == null || !ID.matcher(value).matches())
+            throw new IllegalArgumentException("conversationId 非法");
+    }
+
+    public static void requireExecutionToken(String value) {
+        if (value == null || !ID.matcher(value).matches())
+            throw new IllegalArgumentException("runId 非法");
+    }
+
+    public static int integer(Map<String, Object> values, String key, int min, int max) {
+        Object raw = values.get(key);
+        if (!(raw instanceof Number)) throw new IllegalArgumentException(key + " 非法");
+        int value = ((Number) raw).intValue();
+        requireRange(value, min, max, key);
+        return value;
+    }
+
+    public static long number(Object value) {
+        return value instanceof Number
+                ? ((Number) value).longValue()
+                : Long.parseLong(String.valueOf(value));
+    }
+
+    public static String optionalText(Map<String, Object> values, String key, int max) {
+        return text(values, key, max, false);
+    }
+
+    public static String text(Map<?, ?> values, String key, int max, boolean required) {
+        Object raw = values.get(key);
+        if (raw != null && !(raw instanceof String))
+            throw new IllegalArgumentException(key + " 非法");
+        String value = raw == null ? null : ((String) raw).trim();
+        if (value != null && value.isEmpty()) value = null;
+        if ((required && value == null) || (value != null && value.length() > max))
+            throw new IllegalArgumentException(key + " 非法");
+        return value;
+    }
+
+    public static void requireRange(int value, int min, int max, String name) {
+        if (value < min || value > max)
+            throw new IllegalArgumentException(name + " 超出范围");
+    }
+
+    public static String likePrefix(String prefix) {
+        return prefix.replace("\\", "\\\\").replace("%", "\\%").replace("_", "\\_") + "%";
+    }
+
+    public static boolean same(String left, String right) {
+        return left == null ? right == null : left.equals(right);
+    }
+
+    public static Map<String, Object> success() {
+        Map<String, Object> response = new LinkedHashMap<>();
+        response.put("success", true);
+        return response;
+    }
+
+    public static Map<String, Object> ok(Object data) {
+        Map<String, Object> response = success();
+        response.put("data", data);
+        return response;
+    }
+}

package/ark-control/src/main/java/com/union/control/utils/security/SymmetricalSecurityUtils.java

@@ -0,0 +1,160 @@
+package com.union.control.utils.security;
+
+import com.epcc.commons.securityproxy.api.SecurityResult;
+import com.epcc.commons.securityproxy.api.SymmetricalSecurityService;
+import com.epcc.dubbo.result.Result;
+import com.nucc.channel.ark.common.exception.BaseDataErrorCode;
+import com.nucc.channel.ark.common.exception.CheckException;
+import org.apache.commons.codec.binary.Base64;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import static com.nucc.channel.ark.common.util.Constant.ENCRYPT_END;
+import static com.nucc.channel.ark.common.util.Constant.ENCRYPT_START;
+import static com.nucc.channel.ark.common.util.Constant.REGEX_DECRYPT_TAG;
+import static com.nucc.channel.ark.common.util.Constant.REGEX_EMAIL;
+import static com.nucc.channel.ark.common.util.Constant.REGEX_MOBILE;
+import static com.nucc.channel.ark.common.util.Constant.REGEX_TELEPHONE;
+
+/** Production-equivalent crypto utility; cryptography remains behind sensitiveProxy. */
+@Component
+public class SymmetricalSecurityUtils {
+    private static final Logger LOG = LoggerFactory.getLogger(SymmetricalSecurityUtils.class);
+    private static final Pattern MOBILE = Pattern.compile(REGEX_MOBILE);
+    private static final Pattern EMAIL = Pattern.compile(REGEX_EMAIL);
+    private static final Pattern TELEPHONE = Pattern.compile(REGEX_TELEPHONE);
+    private static final Pattern DECRYPT_TAG = Pattern.compile(REGEX_DECRYPT_TAG);
+
+    @Resource(name = "sensitiveProxy")
+    private SymmetricalSecurityService symmetricalSecurityService;
+
+    public SymmetricalSecurityUtils() {}
+
+    public SymmetricalSecurityUtils(SymmetricalSecurityService symmetricalSecurityService) {
+        this.symmetricalSecurityService = symmetricalSecurityService;
+    }
+
+    public String encryptLongString(String plaintext, int chunkSize) throws CheckException {
+        if (chunkSize == 0) return encryptByFixKey(plaintext);
+        List<String> chunks = new ArrayList<>();
+        for (int i = 0; i < plaintext.length(); i += chunkSize) {
+            chunks.add(encryptByFixKey(plaintext.substring(i, Math.min(plaintext.length(), i + chunkSize))));
+        }
+        return join(chunks, "|");
+    }
+
+    public String encryptWithTag(String plaintext) throws CheckException {
+        if (isBlank(plaintext)) return "";
+        String value = replaceEncrypted(MOBILE, plaintext);
+        value = replaceEncrypted(EMAIL, value);
+        return replaceEncrypted(TELEPHONE, value);
+    }
+
+    public String decryptLongString(String ciphertext) throws CheckException {
+        String[] chunks = ciphertext.split("\\|");
+        StringBuilder result = new StringBuilder();
+        for (String chunk : chunks) result.append(decryptWithCheckNoLog(chunk));
+        return result.toString();
+    }
+
+    public String decryptWithTag(String ciphertext) throws CheckException {
+        if (isBlank(ciphertext)) return "";
+        StringBuilder value = new StringBuilder(ciphertext);
+        Matcher matcher = DECRYPT_TAG.matcher(value);
+        while (matcher.find()) {
+            value.replace(matcher.start(), matcher.end(), decryptWithCheckNoLog(matcher.group(1)));
+            matcher.reset(value);
+        }
+        return value.toString();
+    }
+
+    public String encryptWithCheck(String plaintext) throws CheckException {
+        return isBlank(plaintext) ? plaintext : encryptByFixKey(plaintext);
+    }
+
+    public String decryptWithCheck(String ciphertext) {
+        if (isBlank(ciphertext)) return ciphertext;
+        try {
+            return decryptByFixKey(ciphertext);
+        } catch (CheckException error) {
+            LOG.error("decryptWithCheck failed error_type={}", error.getClass().getSimpleName());
+            return ciphertext;
+        }
+    }
+
+    public String decryptWithCheckNoLog(String ciphertext) throws CheckException {
+        return isBlank(ciphertext) ? ciphertext : decryptByFixKey(ciphertext);
+    }
+
+    private String replaceEncrypted(Pattern pattern, String plaintext) throws CheckException {
+        StringBuilder value = new StringBuilder(plaintext);
+        Matcher matcher = pattern.matcher(value);
+        while (matcher.find()) {
+            value.replace(matcher.start(), matcher.end(),
+                    ENCRYPT_START + encryptByFixKey(matcher.group()) + ENCRYPT_END);
+            matcher.reset(value);
+        }
+        return value.toString();
+    }
+
+    private String encryptByFixKey(String plaintext) throws CheckException {
+        long started = System.currentTimeMillis();
+        Result<SecurityResult> result = symmetricalSecurityService.encryptByFixedKey(
+                SymmetricalSecurityService.Algorithm.AES256,
+                plaintext.getBytes(StandardCharsets.UTF_8));
+        if (!result.isSuccess()) throw remoteFailure(result);
+        debug("encryptByFixKey", started);
+        return Base64.encodeBase64String(result.getResult().getContent());
+    }
+
+    private String decryptByFixKey(String ciphertext) throws CheckException {
+        long started = System.currentTimeMillis();
+        final byte[] decoded;
+        try {
+            decoded = Base64.decodeBase64(ciphertext);
+        } catch (IllegalArgumentException error) {
+            throw new CheckException(BaseDataErrorCode.SYSTEM_INNER_ERROR, "Invalid ciphertext");
+        }
+        Result<SecurityResult> result = symmetricalSecurityService.decryptByFixedKey(
+                SymmetricalSecurityService.Algorithm.AES256, decoded);
+        if (!result.isSuccess()) throw remoteFailure(result);
+        String plaintext = new String(result.getResult().getContent(), StandardCharsets.UTF_8);
+        debug("decryptByFixKey", started);
+        return plaintext;
+    }
+
+    private void debug(String operation, long started) {
+        if (LOG.isDebugEnabled()) LOG.debug("SymmetricalSecurityUtils.{} elapsedMs={}",
+                operation, System.currentTimeMillis() - started);
+    }
+
+    private static CheckException remoteFailure(Result<SecurityResult> result) {
+        return new CheckException(BaseDataErrorCode.SYSTEM_INNER_ERROR,
+                result.getErrorCode() + "|" + result.getErrorMsg());
+    }
+
+    private static boolean isBlank(String value) {
+        if (value == null || value.isEmpty()) return true;
+        for (int i = 0; i < value.length(); i++) {
+            if (!Character.isWhitespace(value.charAt(i))) return false;
+        }
+        return true;
+    }
+
+    private static String join(List<String> values, String delimiter) {
+        StringBuilder result = new StringBuilder();
+        for (String value : values) {
+            if (result.length() > 0) result.append(delimiter);
+            result.append(value);
+        }
+        return result.toString();
+    }
+}

package/ark-control/src/main/resources/META-INF/dubbo/com.alibaba.dubbo.rpc.Filter

@@ -0,0 +1 @@
+providerAccessLog=com.union.control.dubbo.ProviderAccessLogFilter

package/ark-control/src/main/resources/application.properties

@@ -0,0 +1,4 @@
+sensitive.reveal.enabled=${SENSITIVE_REVEAL_ENABLED:true}
+sensitive.reveal.strict_mode=${SENSITIVE_REVEAL_STRICT_MODE:true}
+sensitive.reveal.ttl-seconds=${SENSITIVE_REVEAL_TTL_SECONDS:300}
+sensitive.address-book.plaintext-roles=${SENSITIVE_ADDRESS_BOOK_PLAINTEXT_ROLES:001,002,003}

package/ark-control/src/main/resources/application.yml

@@ -0,0 +1,32 @@
+agent:
+  py-app-base-url: ${PY_APP_BASE_URL:http://127.0.0.1:8083}
+  scheduled-enabled: ${SCHEDULED_TASK_ENABLED:true}
+  scheduled-worker-threads: ${SCHEDULED_TASK_WORKER_THREADS:2}
+  scheduled-max-run-seconds: ${SCHEDULED_TASK_MAX_RUN_SECONDS:930}
+  scheduled-token-ttl-seconds: ${SCHEDULED_TASK_TOKEN_TTL_SECONDS:960}
+
+dubbo:
+  application:
+    name: ark-control
+  registry:
+    address: ${DUBBO_REGISTRY_ADDRESS:zookeeper://127.0.0.1:2181}
+  protocol:
+    port: ${DUBBO_PROTOCOL_PORT:20880}
+
+spring:
+  main:
+    web-environment: false
+  datasource:
+    url: ${MYSQL_URL:jdbc:mysql://127.0.0.1:3306/union_control?createDatabaseIfNotExist=true&allowPublicKeyRetrieval=true&useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=UTC}
+    username: ${MYSQL_USER:root}
+    password: ${MYSQL_PASSWORD:mmls@HW902}
+    driver-class-name: com.mysql.cj.jdbc.Driver
+    initialize: true
+  redis:
+    host: ${REDIS_HOST:127.0.0.1}
+    port: ${REDIS_PORT:6379}
+    password: ${REDIS_PASSWORD:mmls@HW902}
+    timeout: ${REDIS_TIMEOUT_MS:2000}
+
+mybatis:
+  mapper-locations: classpath*:mapper/*.xml

package/ark-control/src/main/resources/dubbo-provider.xml

@@ -0,0 +1,20 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<beans xmlns="http://www.springframework.org/schema/beans"
+       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+       xmlns:dubbo="http://code.alibabatech.com/schema/dubbo"
+       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
+                           http://code.alibabatech.com/schema/dubbo http://code.alibabatech.com/schema/dubbo/dubbo.xsd">
+    <dubbo:application name="${dubbo.application.name}"/>
+    <dubbo:registry address="${dubbo.registry.address}"/>
+    <dubbo:protocol name="dubbo" port="${dubbo.protocol.port}" transporter="netty4"/>
+    <dubbo:provider filter="providerAccessLog"/>
+
+    <dubbo:service interface="com.union.control.service.AgentExecutionService" ref="agentExecutionService"/>
+    <dubbo:service interface="com.union.control.service.AgentProxyService" ref="agentProxyService"/>
+    <dubbo:service interface="com.union.control.service.ConversationService" ref="conversationService"/>
+    <dubbo:service interface="com.union.control.service.MemoryStoreService" ref="memoryStoreService"/>
+    <dubbo:service interface="com.union.control.service.RunningAnalysisMockService" ref="runningAnalysisMockService"/>
+    <dubbo:service interface="com.union.control.service.ScheduledTaskService" ref="scheduledTaskService"/>
+    <dubbo:service interface="com.union.control.service.SensitiveDataDemoService" ref="sensitiveDataDemoService"/>
+    <dubbo:service interface="com.union.control.service.sensitive.SensitiveRevealService" ref="sensitiveRevealService"/>
+</beans>

package/ark-control/src/main/resources/mapper/AgentExecutionMapper.xml

@@ -0,0 +1,75 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper
+        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
+        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.union.control.mapper.AgentExecutionMapper">
+
+    <update id="softDeleteExecutions">
+        UPDATE ai_agent_execution
+        SET delete_flag=0,updated_at=CURRENT_TIMESTAMP
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+    </update>
+
+    <insert id="insertRootExecution">
+        INSERT INTO ai_agent_execution
+            (run_id,conversation_id,user_id,parent_execution_id,agent_name,status,delete_flag)
+        VALUES (#{runId},#{conversationId},#{userId},NULL,NULL,'running',1)
+    </insert>
+
+    <insert id="insertCompletedRootExecution" parameterType="map"
+            useGeneratedKeys="true" keyProperty="id">
+        INSERT INTO ai_agent_execution
+            (run_id,conversation_id,user_id,parent_execution_id,agent_name,status,
+             finished_at,delete_flag,created_at,updated_at)
+        VALUES (#{runId},#{conversationId},#{userId},NULL,#{agentName},'completed',
+                CURRENT_TIMESTAMP,1,CURRENT_TIMESTAMP,CURRENT_TIMESTAMP)
+    </insert>
+
+    <update id="updateRootAgent">
+        UPDATE ai_agent_execution
+        SET agent_name=#{agentName},updated_at=CURRENT_TIMESTAMP
+        WHERE id=#{id} AND delete_flag=1
+    </update>
+
+    <insert id="insertChildExecution">
+        INSERT INTO ai_agent_execution
+            (run_id,conversation_id,user_id,parent_execution_id,agent_name,
+             delegation_tool_call_id,task,status,delete_flag)
+        VALUES (#{runId},#{conversationId},#{userId},#{parentExecutionId},#{agentName},
+                #{delegationCallId},#{task},'running',1)
+    </insert>
+
+    <select id="findExecutions" resultType="map">
+        SELECT e.id,e.run_id AS runId,e.conversation_id AS conversationId,
+               e.parent_execution_id AS parentExecutionId,p.run_id AS parentRunId,
+               e.agent_name AS agentName,e.delegation_tool_call_id AS delegationToolCallId,
+               e.task,e.status,e.error_code AS errorCode,
+               DATE_FORMAT(e.finished_at,'%Y-%m-%dT%H:%i:%s') AS finishedAt,
+               DATE_FORMAT(e.created_at,'%Y-%m-%dT%H:%i:%s') AS createdAt
+        FROM ai_agent_execution e
+        LEFT JOIN ai_agent_execution p ON p.id=e.parent_execution_id
+        WHERE e.user_id=#{userId} AND e.conversation_id=#{conversationId} AND e.delete_flag=1
+        ORDER BY e.created_at,e.id
+    </select>
+
+    <select id="findExecution" resultType="map">
+        SELECT e.id,e.run_id AS runId,e.conversation_id AS conversationId,
+               e.parent_execution_id AS parentExecutionId,p.run_id AS parentRunId,
+               e.agent_name AS agentName,e.delegation_tool_call_id AS delegationToolCallId,
+               e.task,e.status,e.error_code AS errorCode,
+               DATE_FORMAT(e.finished_at,'%Y-%m-%dT%H:%i:%s') AS finishedAt,
+               DATE_FORMAT(e.created_at,'%Y-%m-%dT%H:%i:%s') AS createdAt
+        FROM ai_agent_execution e
+        LEFT JOIN ai_agent_execution p ON p.id=e.parent_execution_id
+        WHERE e.user_id=#{userId} AND e.conversation_id=#{conversationId}
+          AND e.run_id=#{runId} AND e.delete_flag=1
+        <if test="lock">FOR UPDATE</if>
+    </select>
+
+    <update id="finishExecution">
+        UPDATE ai_agent_execution
+        SET status=#{status},error_code=#{errorCode},
+            finished_at=CURRENT_TIMESTAMP,updated_at=CURRENT_TIMESTAMP
+        WHERE id=#{id} AND delete_flag=1
+    </update>
+</mapper>

package/ark-control/src/main/resources/mapper/ConversationMapper.xml

@@ -0,0 +1,110 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper
+        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
+        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.union.control.mapper.ConversationMapper">
+
+    <select id="findConversations" resultType="map">
+        SELECT c.conversation_id AS conversationId,c.title,c.status,
+               DATE_FORMAT(c.created_at,'%Y-%m-%dT%H:%i:%s') AS createdAt,
+               DATE_FORMAT(c.updated_at,'%Y-%m-%dT%H:%i:%s') AS updatedAt,
+               (SELECT e.status FROM ai_agent_execution e
+                WHERE e.conversation_id=c.conversation_id AND e.user_id=c.user_id
+                  AND e.parent_execution_id IS NULL AND e.delete_flag=1
+                ORDER BY e.created_at DESC,e.id DESC LIMIT 1) AS executionStatus
+        FROM ai_conversation c
+        WHERE c.user_id=#{userId} AND c.delete_flag=1
+          AND c.status='active'
+        ORDER BY c.created_at DESC,c.id DESC
+        LIMIT #{limit}
+    </select>
+
+    <select id="findConversation" resultType="map">
+        SELECT c.conversation_id AS conversationId,c.title,c.status,
+               DATE_FORMAT(c.created_at,'%Y-%m-%dT%H:%i:%s') AS createdAt,
+               DATE_FORMAT(c.updated_at,'%Y-%m-%dT%H:%i:%s') AS updatedAt,
+               (SELECT e.status FROM ai_agent_execution e
+                WHERE e.conversation_id=c.conversation_id AND e.user_id=c.user_id
+                  AND e.parent_execution_id IS NULL AND e.delete_flag=1
+                ORDER BY e.created_at DESC,e.id DESC LIMIT 1) AS executionStatus
+        FROM ai_conversation c
+        WHERE c.conversation_id=#{conversationId}
+          AND c.user_id=#{userId} AND c.delete_flag=1
+    </select>
+
+    <update id="updateConversationTitle">
+        UPDATE ai_conversation
+        SET title=#{title},updated_at=CURRENT_TIMESTAMP
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+    </update>
+
+    <update id="softDeleteMessages">
+        UPDATE ai_conversation_message
+        SET delete_flag=0,updated_at=CURRENT_TIMESTAMP
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+    </update>
+
+    <update id="softDeleteConversation">
+        UPDATE ai_conversation
+        SET delete_flag=0,updated_at=CURRENT_TIMESTAMP
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+    </update>
+
+    <insert id="insertConversation">
+        INSERT INTO ai_conversation
+            (conversation_id,user_id,title,delete_flag)
+        VALUES (#{conversationId},#{userId},#{title},1)
+    </insert>
+
+    <select id="currentMessageSequence" resultType="long">
+        SELECT COALESCE(MAX(sequence_no),0)
+        FROM ai_conversation_message
+        WHERE conversation_id=#{conversationId}
+    </select>
+
+    <insert id="insertMessage">
+        INSERT INTO ai_conversation_message
+            (conversation_id,user_id,message_id,agent_execution_id,role,sequence_no,payload,delete_flag)
+        VALUES (#{conversationId},#{userId},#{messageId},#{executionId},#{role},#{sequence},#{payload},1)
+    </insert>
+
+    <update id="touchConversation">
+        UPDATE ai_conversation
+        SET updated_at=CURRENT_TIMESTAMP
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+    </update>
+
+    <select id="findRootMessages" resultType="map">
+        SELECT m.message_id AS messageId,m.role,m.payload
+        FROM ai_conversation_message m
+        JOIN ai_agent_execution e ON e.id=m.agent_execution_id
+        WHERE m.conversation_id=#{conversationId} AND m.user_id=#{userId} AND m.delete_flag=1
+          AND e.delete_flag=1 AND e.parent_execution_id IS NULL
+        ORDER BY m.sequence_no
+    </select>
+
+    <select id="findBrowserMessages" resultType="map">
+        SELECT m.message_id AS messageId,m.role,m.payload,e.run_id AS runId,
+               e.parent_execution_id AS parentExecutionId
+        FROM ai_conversation_message m
+        JOIN ai_agent_execution e ON e.id=m.agent_execution_id
+        WHERE m.conversation_id=#{conversationId} AND m.user_id=#{userId} AND m.delete_flag=1
+          AND e.delete_flag=1
+        ORDER BY m.sequence_no
+    </select>
+
+    <select id="requireOwnedActive" resultType="long">
+        SELECT id
+        FROM ai_conversation
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId}
+          AND status='active' AND delete_flag=1
+    </select>
+
+    <select id="requireOwned" resultType="long">
+        SELECT id
+        FROM ai_conversation
+        WHERE conversation_id=#{conversationId} AND user_id=#{userId} AND delete_flag=1
+        <if test="lock">FOR UPDATE</if>
+    </select>
+</mapper>
+

package/ark-control/src/main/resources/mapper/MemoryStoreMapper.xml

@@ -0,0 +1,74 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper
+        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
+        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.union.control.mapper.MemoryStoreMapper">
+
+    <select id="readMemory" resultType="map">
+        SELECT content,version,last_operation_id AS operationId
+        FROM ai_memory_file
+        WHERE user_id=#{userId} AND path=#{path} AND delete_flag=1
+    </select>
+
+    <select id="listMemoryPaths" resultType="string">
+        SELECT path
+        FROM ai_memory_file
+        WHERE user_id=#{userId} AND delete_flag=1 AND path LIKE #{prefix}
+        ORDER BY path
+        LIMIT #{limit}
+    </select>
+
+    <select id="findMemoryOperation" resultType="map">
+        SELECT fingerprint,result_version AS resultVersion,existed
+        FROM ai_memory_operation
+        WHERE user_id=#{userId} AND operation_id=#{operationId}
+        <if test="lock">FOR UPDATE</if>
+    </select>
+
+    <select id="currentMemoryVersion" resultType="long">
+        SELECT COALESCE(MAX(version),0)
+        FROM ai_memory_file
+        WHERE user_id=#{userId} AND path=#{path}
+    </select>
+
+    <update id="updateMemoryFile">
+        UPDATE ai_memory_file
+        SET content=#{content},version=#{version},last_operation_id=#{operationId},
+            updated_at=CURRENT_TIMESTAMP
+        WHERE id=#{id} AND delete_flag=1
+    </update>
+
+    <insert id="insertMemoryFile">
+        INSERT INTO ai_memory_file
+            (user_id,path,content,version,last_operation_id,delete_flag)
+        VALUES (#{userId},#{path},#{content},#{version},#{operationId},1)
+    </insert>
+
+    <update id="softDeleteMemoryFile">
+        UPDATE ai_memory_file
+        SET delete_flag=0,updated_at=CURRENT_TIMESTAMP
+        WHERE id=#{id} AND delete_flag=1
+    </update>
+
+    <select id="searchMemory" resultType="map">
+        SELECT path,content
+        FROM ai_memory_file
+        WHERE user_id=#{userId} AND delete_flag=1 AND path LIKE #{prefix}
+        ORDER BY path
+        LIMIT #{limit}
+    </select>
+
+    <select id="findMemoryRow" resultType="map">
+        SELECT id,version
+        FROM ai_memory_file
+        WHERE user_id=#{userId} AND path=#{path} AND delete_flag=1
+        FOR UPDATE
+    </select>
+
+    <insert id="insertMemoryOperation">
+        INSERT INTO ai_memory_operation
+            (user_id,operation_id,fingerprint,result_version,existed,delete_flag)
+        VALUES (#{userId},#{operationId},#{fingerprint},#{version},#{existed},1)
+    </insert>
+</mapper>
+