diff --git a/app.py b/app.py
index 6960aa4..714e56b 100644
--- a/app.py
+++ b/app.py
@@ -1175,7 +1175,7 @@ def create_connection(config):
return None, None
def execute_query(session, table, keys, fields, values, exclude_fields=None):
- """执行查询"""
+ """执行查询，支持单主键和复合主键"""
try:
# 参数验证
if not keys or len(keys) == 0:
@@ -1187,8 +1187,37 @@ def execute_query(session, table, keys, fields, values, exclude_fields=None):
return []
# 构建查询条件
- quoted_values = [f"'{value}'" for value in values]
- query_conditions = f"{keys[0]} IN ({', '.join(quoted_values)})"
+ if len(keys) == 1:
+ # 单主键查询（保持原有逻辑）
+ quoted_values = [f"'{value}'" for value in values]
+ query_conditions = f"{keys[0]} IN ({', '.join(quoted_values)})"
+ else:
+ # 复合主键查询
+ conditions = []
+ for value in values:
+ # 检查value是否包含复合主键分隔符
+ if isinstance(value, str) and ',' in value:
+ # 解析复合主键值
+ key_values = [v.strip() for v in value.split(',')]
+ if len(key_values) == len(keys):
+ # 构建单个复合主键条件: (key1='val1' AND key2='val2')
+ key_conditions = []
+ for i, (key, val) in enumerate(zip(keys, key_values)):
+ key_conditions.append(f"{key} = '{val}'")
+ conditions.append(f"({' AND '.join(key_conditions)})")
+ else:
+ logger.warning(f"复合主键值 '{value}' 的字段数量({len(key_values)})与主键字段数量({len(keys)})不匹配")
+ # 将其作为第一个主键的值处理
+ conditions.append(f"{keys[0]} = '{value}'")
+ else:
+ # 单值，作为第一个主键的值处理
+ conditions.append(f"{keys[0]} = '{value}'")
+
+ if conditions:
+ query_conditions = ' OR '.join(conditions)
+ else:
+ logger.error("无法构建有效的查询条件")
+ return []
# 确定要查询的字段
if fields:
@@ -1200,7 +1229,10 @@ def execute_query(session, table, keys, fields, values, exclude_fields=None):
# 记录查询SQL日志
logger.info(f"执行查询SQL: {query_sql}")
- logger.info(f"查询参数: 表={table}, 字段={fields_str}, Key数量={len(values)}")
+ if len(keys) > 1:
+ logger.info(f"复合主键查询参数: 表={table}, 主键字段={keys}, 字段={fields_str}, Key数量={len(values)}")
+ else:
+ logger.info(f"单主键查询参数: 表={table}, 主键字段={keys[0]}, 字段={fields_str}, Key数量={len(values)}")
# 执行查询
start_time = time.time()
@@ -1367,15 +1399,29 @@ def execute_mixed_query(pro_session, test_session, pro_config, test_config, keys
return results
def compare_results(pro_data, test_data, keys, fields_to_compare, exclude_fields, values):
- """比较查询结果"""
+ """比较查询结果，支持复合主键"""
differences = []
field_diff_count = {}
identical_results = [] # 存储相同的结果
+ def match_composite_key(row, composite_value, keys):
+ """检查数据行是否匹配复合主键值"""
+ if len(keys) == 1:
+ # 单主键匹配
+ return getattr(row, keys[0]) == composite_value
+ else:
+ # 复合主键匹配
+ if isinstance(composite_value, str) and ',' in composite_value:
+ key_values = [v.strip() for v in composite_value.split(',')]
+ if len(key_values) == len(keys):
+ return all(str(getattr(row, key)) == key_val for key, key_val in zip(keys, key_values))
+ # 如果不是复合值，只匹配第一个主键
+ return getattr(row, keys[0]) == composite_value
+
for value in values:
- # 查找原表和测试表中该ID的相关数据
- rows_pro = [row for row in pro_data if getattr(row, keys[0]) == value]
- rows_test = [row for row in test_data if getattr(row, keys[0]) == value]
+ # 查找生产表和测试表中该主键值的相关数据
+ rows_pro = [row for row in pro_data if match_composite_key(row, value, keys)]
+ rows_test = [row for row in test_data if match_composite_key(row, value, keys)]
for row_pro in rows_pro:
# 在测试表中查找相同主键的行
@@ -2180,4 +2226,4 @@ def api_get_query_logs_by_history(history_id):
return jsonify({'success': False, 'error': str(e)}), 500
if __name__ == '__main__':
- app.run(debug=True, port=5000)
+ app.run(debug=True, port=5001)
diff --git a/static/js/app.js b/static/js/app.js
index f6a8ee5..d873b7d 100644
--- a/static/js/app.js
+++ b/static/js/app.js
@@ -52,24 +52,24 @@ function toggleShardingMode() {
// 启用分表模式
shardingConfig.style.display = 'block';
executeButton.textContent = '执行分表查询比对';
- keyInputHint.textContent = '分表模式：Key值应包含时间戳用于计算分表索引';
- keysField.placeholder = ' (推荐使用包含时间戳的字段)';
+ keyInputHint.textContent = '分表模式：Key值应包含时间戳用于计算分表索引，支持单主键和复合主键（逗号分隔）';
+ keysField.placeholder = '单主键：wmid 或 复合主键：docid,id (推荐使用包含时间戳的字段)';
keysField.value = '';
// 更新查询Key输入框的占位符
const queryValues = document.getElementById('query_values');
- queryValues.placeholder = '请输入查询的Key值，一行一个\n分表查询示例（包含时间戳）：\nwmid_1609459200\nwmid_1610064000\nwmid_1610668800';
+ queryValues.placeholder = '请输入查询的Key值，一行一个\n单主键示例（包含时间戳）：\nwmid_1609459200\nwmid_1610064000\nwmid_1610668800\n\n复合主键示例（逗号分隔）：\ndocid_1609459200,id1\ndocid_1610064000,id2\ndocid_1610668800,id3';
} else {
// 禁用分表模式
shardingConfig.style.display = 'none';
executeButton.textContent = '执行查询比对';
- keyInputHint.textContent = '单表模式：输入普通Key值';
- keysField.placeholder = '';
+ keyInputHint.textContent = '单表模式：支持单主键和复合主键查询，复合主键用逗号分隔';
+ keysField.placeholder = '单主键：docid 或 复合主键：docid,id';
keysField.value = '';
// 更新查询Key输入框的占位符
const queryValues = document.getElementById('query_values');
- queryValues.placeholder = '请输入查询的Key值，一行一个\n单表查询示例：\nkey1\nkey2\nkey3';
+ queryValues.placeholder = '请输入查询的Key值，一行一个\n单主键示例：\nkey1\nkey2\nkey3\n\n复合主键示例（逗号分隔）：\ndocid1,id1\ndocid2,id2\ndocid3,id3';
}
updateTableNameHints();
@@ -975,7 +975,7 @@ function displayDifferences() {
- 主键: ${JSON.stringify(diff.key)}
+ 主键: ${formatCompositeKey(diff.key)}
${diff.message}
@@ -1008,7 +1008,7 @@ function displayDifferences() {
- 主键: ${JSON.stringify(diff.key)}
+ 主键: ${formatCompositeKey(diff.key)}
差异字段: ${diff.field}
@@ -1137,7 +1137,7 @@ function displayIdenticalResults() {
- 主键: ${JSON.stringify(result.key)}
+ 主键: ${formatCompositeKey(result.key)}
@@ -1559,16 +1559,34 @@ function showRawData(keyStr) {
// 查找生产环境数据
if (currentResults.raw_pro_data) {
proData = currentResults.raw_pro_data.find(item => {
- // 比较主键
- return JSON.stringify(item[Object.keys(key)[0]]) === JSON.stringify(key[Object.keys(key)[0]]);
+ // 支持复合主键比较
+ if (typeof key === 'object' && !Array.isArray(key)) {
+ // 复合主键情况：比较所有主键字段
+ return Object.keys(key).every(keyField =>
+ JSON.stringify(item[keyField]) === JSON.stringify(key[keyField])
+ );
+ } else {
+ // 单主键情况：保持原有逻辑
+ const keyField = Object.keys(key)[0];
+ return JSON.stringify(item[keyField]) === JSON.stringify(key[keyField]);
+ }
});
}
// 查找测试环境数据
if (currentResults.raw_test_data) {
testData = currentResults.raw_test_data.find(item => {
- // 比较主键
- return JSON.stringify(item[Object.keys(key)[0]]) === JSON.stringify(key[Object.keys(key)[0]]);
+ // 支持复合主键比较
+ if (typeof key === 'object' && !Array.isArray(key)) {
+ // 复合主键情况：比较所有主键字段
+ return Object.keys(key).every(keyField =>
+ JSON.stringify(item[keyField]) === JSON.stringify(key[keyField])
+ );
+ } else {
+ // 单主键情况：保持原有逻辑
+ const keyField = Object.keys(key)[0];
+ return JSON.stringify(item[keyField]) === JSON.stringify(key[keyField]);
+ }
});
}
@@ -3867,7 +3885,7 @@ function renderRawDataContent() {
const globalIndex = startIndex + index + 1;
const envBadgeClass = item.environment === 'production' ? 'bg-primary' : 'bg-success';
- // 获取主键值用于标识 - 修复主键解析问题
+ // 获取主键值用于标识 - 修复主键解析问题，支持复合主键
let keyValue = 'N/A';
// 从当前配置获取keys，因为query_config可能不在results中
@@ -3884,11 +3902,12 @@ function renderRawDataContent() {
}).filter(val => val !== 'N/A');
if (keyValues.length > 0) {
+ // 复合主键显示格式：docid1,id1 或 单主键显示：value1
keyValue = keyValues.join(', ');
}
} else {
// 如果没有配置主键，尝试常见的主键字段名
- const commonKeyFields = ['id', 'statusid', 'key', 'uuid'];
+ const commonKeyFields = ['id', 'statusid', 'key', 'uuid', 'docid'];
for (const field of commonKeyFields) {
if (item.data[field] !== undefined) {
keyValue = item.data[field];
@@ -4122,4 +4141,21 @@ function copyRawRecord(recordStr) {
} catch (error) {
showAlert('danger', '复制失败: ' + error.message);
}
+}
+
+// 格式化复合主键显示
+function formatCompositeKey(key) {
+ if (!key) {
+ return 'N/A';
+ }
+
+ if (typeof key === 'object' && !Array.isArray(key)) {
+ // 复合主键情况：显示为 "docid: value1, id: value2" 格式
+ return Object.entries(key)
+ .map(([field, value]) => `${field}: ${value}`)
+ .join(', ');
+ } else {
+ // 单主键或其他情况：使用JSON格式
+ return JSON.stringify(key);
+ }
}
\ No newline at end of file
diff --git a/templates/db_compare.html b/templates/db_compare.html
index de3ce57..eb8bb69 100644
--- a/templates/db_compare.html
+++ b/templates/db_compare.html
@@ -539,9 +539,9 @@