备份一下 datart 视图路径配置 feign调用

视图路径配置

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
package datart.api.service;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.lang.Dict;
import cn.hutool.core.lang.TypeReference;
import cn.hutool.core.util.BooleanUtil;
import cn.hutool.core.util.ReUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import datart.api.config.DatartProperties;
import datart.api.feign.DataProviderFeignClient;
import datart.api.feign.ViewFeignClient;
import datart.api.feign.dataobject.BaseEntity;
import datart.api.feign.dataobject.CheckNameParam;
import datart.api.feign.dataobject.Column;
import datart.api.feign.dataobject.Dataframe;
import datart.api.feign.dataobject.ResponseData;
import datart.api.feign.dataobject.TestExecuteParam;
import datart.api.feign.dataobject.View;
import datart.api.feign.dataobject.ViewDetailDTO;
import datart.api.service.custom.exception.DataProviderException;
import datart.api.service.custom.qry.SqlQry;
import datart.api.service.custom.qry.SqlVariable;
import datart.api.service.custom.qry.ViewQry;
import datart.api.service.custom.result.QueryResult;
import datart.api.service.custom.result.Row;
import feign.Response;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.mvel2.MVEL;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
* 描述
*
* @author 陶攀峰
* @date 2022-05-25 19:42
*/
@Component
@Slf4j
public class DataProviderServiceImpl implements DataProviderService {

@Autowired
private DatartProperties datartProperties;

@Autowired
private DataProviderFeignClient dataProviderFeignClient;

@Autowired
private ViewFeignClient viewFeignClient;

// -------------------------base-------------------------------

private String getOrgId() {
return datartProperties.getOrgId();
}

private Row one(String sourceId, String sql) {
QueryResult queryResult = list(sourceId, sql);
if (CollectionUtil.isEmpty(queryResult)) {
throw new DataProviderException("selectOne 错误,查询为空。sourceId=[{}], sql=[{}]", sourceId, sql);
}
if (queryResult.size() != 1) {
throw new DataProviderException("selectOne 错误,查询记录不唯一。sourceId=[{}], sql=[{}]", sourceId, sql);
}

return queryResult.get(0);
}

private QueryResult list(String sourceId, String sql) {
TestExecuteParam testExecuteParam = new TestExecuteParam();
testExecuteParam.setSourceId(sourceId);
testExecuteParam.setScript(sql);
testExecuteParam.setSize(Integer.MAX_VALUE);

checkTenantId(sql);
checkSqlVariable(sql);

try {
ResponseData<Dataframe> response = dataProviderFeignClient.testExecute(testExecuteParam);
Dataframe data = response.getData();
List<String> columnNameList = data.getColumns().stream().map(Column::getName).collect(Collectors.toList());

QueryResult queryResult = new QueryResult();

for (List<Object> e : data.getRows()) {
Row row = new Row();
for (int i = 0; i < columnNameList.size(); i++) {
String columnName = columnNameList.get(i);
Object columnValue = e.get(i);
row.put(columnName, columnValue);
}
queryResult.add(row);
}

printSql(queryResult.size(), sql);
return queryResult;
} catch (Exception e) {
throw new DataProviderException(e, "取数出现异常:sourceId=[{}], sql=[{}]", sourceId, sql);
}
}

/**
* 检查 sql 中必须包含 tenant_id 字段
*/
private void checkTenantId(String sql) {
if (!sql.contains("tenant_id")) {
throw new DataProviderException("sql 中不包含 'tenant_id' 字段,请检查 sql=[{}]", sql);
}
}

/**
* 检查 SQL 中是否有未替换的变量
*/
private void checkSqlVariable(String sql) {
List<String> variableList = ReUtil.findAllGroup1("(\\$\\w+\\$)", sql);

if (CollectionUtil.isNotEmpty(variableList)) {
throw new DataProviderException("存在未替换的变量:sql=[{}]", sql);
}
}

/**
* 打印 SQL
*
* @param sql 变量全部替换完成后的 SQL
*/
private void printSql(int size, String sql) {
if (datartProperties.isEnabledPrintSql()) {
log.info("datart print sql ===========================> size=[{}], sql={}", size, sql);
}
}

// -------------------------sql-------------------------------

@Override
public Row selectOne(SqlQry sqlQry) {
return one(sqlQry.getSourceId(), sqlQry.getSql());
}

@Override
public QueryResult selectList(SqlQry sqlQry) {
return list(sqlQry.getSourceId(), sqlQry.getSql());
}

// -------------------------view-------------------------------

@Override
public Row selectOne(ViewQry viewQry) {
SqlQry sqlQry = convert(viewQry);

return selectOne(sqlQry);
}

@Override
public QueryResult selectList(ViewQry viewQry) {
SqlQry sqlQry = convert(viewQry);

return selectList(sqlQry);
}

/**
* 视图查询对象 => SQL查询对象
*/
private SqlQry convert(ViewQry viewQry) {
String viewName = viewQry.getViewName();
SqlVariable sqlVariable = viewQry.getSqlVariable();

// 1、获取视图详情
ViewDetailDTO viewDetailDTO = viewDetailDTO(viewName);
String sqlTemplate = viewDetailDTO.getScript();
String sourceId = viewDetailDTO.getSourceId();

// 2、构建 sql
String sql = buildSql(sqlTemplate, sqlVariable);
return SqlQry.of(sourceId, sql);
}

/**
* 获取视图详情 DTO
*
* @param viewName 视图名。若以 "/" 开头,则以路径匹配查找。
* @return ViewDetailDTO(包含两个重要属性:sourceId=数据源id,sql=字符串SQL)
*/
private ViewDetailDTO viewDetailDTO(String viewName) {
// 1、路径匹配
if (viewName.startsWith("/")) {
return findByViewPath(viewName);
}

// 2、名称匹配
return findByViewName(viewName);
}

/**
* 按照 viewName 查询
*
* @param viewName 视图 全名称匹配
* @return ViewDetailDTO(包含两个重要属性:sourceId=数据源id,sql=字符串SQL)
*/
private ViewDetailDTO findByViewName(String viewName) {
// 1、检查 viewName 在当前组织下是否存在(不存在,抛异常)
checkViewName(viewName);

// 2、viewName 已存在,获取 viewId
ResponseData<List<View>> viewListResp = viewFeignClient.listViews(getOrgId());
List<View> viewList = viewListResp.getData().stream().filter(e -> e.getName().equals(viewName)).collect(Collectors.toList());
if (viewList.size() > 1) {
throw new DataProviderException("查询视图详情错误,视图名称不唯一!!!请使用以 '/' 开头的绝对路径。viewName=[{}], orgId=[{}]", viewName, getOrgId());
}
String viewId = viewList.get(0).getId();

// 3、根据 viewId 获取 view 详情
ResponseData<ViewDetailDTO> viewDetailResp = viewFeignClient.getViewDetail(viewId);
return viewDetailResp.getData();
}

/**
* 按照 viewPath 查询
*
* @param viewPath 视图 全路径匹配
* @return ViewDetailDTO(包含两个重要属性:sourceId=数据源id,sql=字符串SQL)
*/
private ViewDetailDTO findByViewPath(String viewPath) {

// viewPath 不能以 '/' 结尾
if (viewPath.endsWith("/")) {
throw new DataProviderException("查询视图详情错误,viewPath 不能以 '/' 结尾。viewPath=[{}], orgId=[{}]", viewPath, getOrgId());
}

// viewPath="/a/b/c/d/e"
// 找到 viewName="e"
String[] path = viewPath.replaceFirst("/", "").split("/");
String viewName = path[path.length - 1];

// 1、检查 viewName 在当前组织下是否存在(不存在,抛异常)
checkViewName(viewName);

// 2、viewName 已存在,获取 viewId
ResponseData<List<View>> viewListResp = viewFeignClient.listViews(getOrgId());
View view = findCandidate(viewListResp.getData(), path, viewName);
if (view == null) {
throw new DataProviderException("未找到相关视图,请检查路径。viewPath=[{}], orgId=[{}]", viewPath, getOrgId());
}

// 3、根据 viewId 获取 view 详情
return getViewDetail(view.getId());
}

/**
* 检查 viewName 在当前组织下是否存在(不存在,抛异常)
*
* @param viewName 视图名称
*/
@SneakyThrows
private void checkViewName(String viewName) {
// 1、构建参数
CheckNameParam checkNameParam = new CheckNameParam();
checkNameParam.setName(viewName);
checkNameParam.setOrgId(getOrgId());

// 2、feign => datart
Response response = viewFeignClient.checkViewName(checkNameParam);

// 3、读取响应
String body = IoUtil.readUtf8(response.body().asInputStream());
ResponseData<Boolean> checkResp = JSONUtil.toBean(body, new TypeReference<ResponseData<Boolean>>() {
}, false);

// 4、检查
// data == true => 名称不存在
// data == null => 名称已存在
if (BooleanUtil.isTrue(checkResp.getData())) {
throw new DataProviderException("查询视图详情错误,视图不存在。viewName=[{}], orgId=[{}]", viewName, getOrgId());
}
}

/**
* 根据 viewId 获取视图详情
*
* @param viewId 视图id(视图唯一标识符)
* @return ViewDetailDTO(包含两个重要属性:sourceId=数据源id,sql=字符串SQL)
*/
private ViewDetailDTO getViewDetail(String viewId) {
ResponseData<ViewDetailDTO> viewDetailResp = viewFeignClient.getViewDetail(viewId);
return viewDetailResp.getData();
}

/**
* 查找候选人(路径匹配)
*
* @param viewList 当前组织下所有的视图列表
* @param path 视图路径分割列表
* @param viewName 视图名
* @return 匹配到的视图(未找到匹配,返回 null)
*/
private View findCandidate(List<View> viewList, String[] path, String viewName) {
// viewPath="/a/b/c/d/e"
// 找到 viewName="e"

// k=viewId, v=View
Map<String, View> viewMap = viewList.stream().collect(Collectors.toMap(BaseEntity::getId, Function.identity()));
// 候选人列表 不是目录 && 名称="e"
List<View> candidateList = viewList.stream()
.filter(e -> !e.getIsFolder())
.filter(e -> e.getName().equals(viewName)).collect(Collectors.toList());

// 依次匹配候选人
for (View candidate : candidateList) {

View curr = candidate;
// 从倒数第一个开始匹配,也就是 "e"
for (int i = path.length - 2; i >= 0; i--) {
View parent = viewMap.get(curr.getParentId());// "e" 的父亲 "d"

if (parent.getName().trim().equals(path[i].trim())) {// 父目录是"d" ===> continue;继续匹配下一个父目录
if (i == 0) {// 已经是最后一个目录了,直接返回(全部匹配)
return candidate;
}
curr = parent;
continue;
}
break;// 父目录不是"d",当前候选人匹配失败,匹配下一个候选人
}
}

// 走到这里表示 viewPath 配置错误了,没有一个匹配成功的。
return null;
}

/**
* 根据变量,填充 SQL
*
* @param sqlTemplate sql模版
* @param sqlVariable sql变量
* @return 被变量填充后的 sql
*/
private static String buildSql(String sqlTemplate, SqlVariable sqlVariable) {
sqlTemplate = ifTagCheck(sqlTemplate, sqlVariable);

sqlTemplate = sqlVariableReplace(sqlTemplate, sqlVariable);

return sqlTemplate;
}

/**
* if 标签判断
*/
private static String ifTagCheck(String sqlTemplate, SqlVariable sqlVariable) {
String ifReg = "<if test=\"([\\s\\S]+?)\">([\\s\\S]+?)</if>";
List<String> g0 = ReUtil.findAll(ifReg, sqlTemplate, 0);// 整体
List<String> g1 = ReUtil.findAll(ifReg, sqlTemplate, 1);// test 引号内
List<String> g2 = ReUtil.findAll(ifReg, sqlTemplate, 2);// if 标签体

for (int i = 0; i < g0.size(); i++) {
String find = g0.get(i);
String test = " " + g1.get(i).trim() + " ";// 前后拼接空格,防止 SQL 粘连
String body = " " + g2.get(i).trim() + " ";// body 中的变量,在 buildSql 填充

String expression = "if(test) return body;".replace("test", test);

Dict mvelVars = Dict.create().set("body", body);

// 找到 test 所有变量
// sqlVariable => dict
List<String> testVars = ReUtil.findAllGroup1("\\$(\\w+)\\$", test);
testVars.forEach(e -> mvelVars.set("$" + e + "$", sqlVariable.get(e)));

// test通过,返回 if 标签体
// test不通过,返回 null => ""
String res = MVEL.eval(expression, mvelVars, String.class);// " and storeId in [$storeId$] "
if (StrUtil.isBlank(res)) {
res = "";
}
sqlTemplate = sqlTemplate.replace(find, res);

}
return sqlTemplate;
}

/**
* SQL 变量替换
*/
private static String sqlVariableReplace(String sqlTemplate, SqlVariable sqlVariable) {
if (sqlVariable == null) {
return sqlTemplate;
}

// 找到所有的变量列表
List<String> vars = ReUtil.findAllGroup1("\\$(\\w+)\\$", sqlTemplate);
for (String var : vars) {
String key = "$" + var + "$";
Set<String> value = sqlVariable.get(var);

if (CollectionUtil.isEmpty(value)) {
throw new DataProviderException("找不到变量 变量名=[{}]:sql=[{}]", var, sqlTemplate);
}

sqlTemplate = sqlTemplate.replace(key, CollectionUtil.join(value, ","));
}

return sqlTemplate;
}

}

多线程

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
package datart.api.service;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.lang.Assert;
import cn.hutool.core.lang.Dict;
import cn.hutool.core.thread.AsyncUtil;
import cn.hutool.core.util.ReUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import datart.api.config.DatartProperties;
import datart.api.feign.DataProviderFeignClient;
import datart.api.feign.dataobject.Column;
import datart.api.feign.dataobject.Dataframe;
import datart.api.feign.dataobject.ResponseData;
import datart.api.feign.dataobject.TestExecuteParam;
import datart.api.feign.dataobject.ViewDetailDTO;
import datart.api.service.custom.exception.DataProviderException;
import datart.api.service.custom.qry.SqlQry;
import datart.api.service.custom.qry.SqlVariable;
import datart.api.service.custom.qry.ViewQry;
import datart.api.service.custom.result.QueryResult;
import datart.api.service.custom.result.Row;
import lombok.extern.slf4j.Slf4j;
import org.mvel2.MVEL;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;

/**
* 数据供应商服务 => 实现
*
* @author 陶攀峰
* @date 2022-05-25 19:42
*/
@Component
@Slf4j
public class DataProviderServiceImpl implements DataProviderService {

@Autowired
private DatartProperties datartProperties;

@Autowired
private DataProviderFeignClient dataProviderFeignClient;

@Autowired
private StringRedisTemplate redis;

// -------------------------base-------------------------------

private Row one(String sourceId, String sql) {
QueryResult queryResult = list(sourceId, sql);
if (CollectionUtil.isEmpty(queryResult)) {
throw new DataProviderException("selectOne 错误,查询为空。sourceId=[{}], sql=[{}]", sourceId, sql);
}
if (queryResult.size() != 1) {
throw new DataProviderException("selectOne 错误,查询记录不唯一。sourceId=[{}], sql=[{}]", sourceId, sql);
}

return queryResult.get(0);
}

private QueryResult list(String sourceId, String sql) {
TestExecuteParam testExecuteParam = new TestExecuteParam();
testExecuteParam.setSourceId(sourceId);
testExecuteParam.setScript(sql);
testExecuteParam.setSize(Integer.MAX_VALUE);

checkTenantId(sql);
checkSqlVariable(sql);

try {
ResponseData<Dataframe> response = dataProviderFeignClient.testExecute(testExecuteParam);
Dataframe data = response.getData();
List<String> columnNameList = data.getColumns().stream().map(Column::getName).collect(Collectors.toList());

QueryResult queryResult = new QueryResult();

for (List<Object> e : data.getRows()) {
Row row = new Row();
for (int i = 0; i < columnNameList.size(); i++) {
String columnName = columnNameList.get(i);
Object columnValue = e.get(i);
row.put(columnName, columnValue);
}
queryResult.add(row);
}

printSql(queryResult.size(), sql);
return queryResult;
} catch (Exception e) {
throw new DataProviderException(e, "取数出现异常:sourceId=[{}], sql=[{}]", sourceId, sql);
}
}

/**
* 检查 sql 中必须包含 tenant_id 字段
*/
private void checkTenantId(String sql) {
if (!sql.contains("tenant_id")) {
throw new DataProviderException("sql 中不包含 'tenant_id' 字段,请检查 sql=[{}]", sql);
}
}

/**
* 检查 SQL 中是否有未替换的变量
*/
private void checkSqlVariable(String sql) {
List<String> variableList = ReUtil.findAllGroup1("(\\$\\w+\\$)", sql);

if (CollectionUtil.isNotEmpty(variableList)) {
throw new DataProviderException("存在未替换的变量:sql=[{}]", sql);
}
}

/**
* 打印 SQL
*
* @param sql 变量全部替换完成后的 SQL
*/
private void printSql(int size, String sql) {
if (datartProperties.isEnabledPrintSql()) {
log.info("datart print sql ===========================> size=[{}], sql={}", size, sql);
}
}

// -------------------------sql-------------------------------

@Override
public Row selectOne(SqlQry sqlQry) {
return one(sqlQry.getSourceId(), sqlQry.getSql());
}

@Override
public QueryResult selectList(SqlQry sqlQry) {
return list(sqlQry.getSourceId(), sqlQry.getSql());
}

@Override
public List<QueryResult> selectListSql(List<SqlQry> sqlQryList) {
// 1、遍历列表
// 2、异步处理 ===> 每个元素都会被处理为 QueryResult 对象
// 3、获取异步处理结果
return sqlQryList.stream()
.map(e -> CompletableFuture.supplyAsync(() -> list(e.getSourceId(), e.getSql())))
.map(AsyncUtil::get)
.collect(Collectors.toList());
}

// -------------------------view-------------------------------

@Override
public Row selectOne(ViewQry viewQry) {
SqlQry sqlQry = convert(viewQry);

return selectOne(sqlQry);
}

@Override
public QueryResult selectList(ViewQry viewQry) {
SqlQry sqlQry = convert(viewQry);

return selectList(sqlQry);
}

@Override
public List<QueryResult> selectListView(List<ViewQry> viewQryList) {
// 1、遍历列表
// 2、异步处理 ===> 每个元素都会被处理为 QueryResult 对象
// 3、获取异步处理结果
return viewQryList.stream()
.map(e -> CompletableFuture.supplyAsync(() -> {
SqlQry sqlQry = convert(e);
return list(sqlQry.getSourceId(), sqlQry.getSql());
}))
.map(AsyncUtil::get)
.collect(Collectors.toList());
}

/**
* 视图查询对象 => SQL查询对象
*/
private SqlQry convert(ViewQry viewQry) {
String viewName = viewQry.getViewName();
SqlVariable sqlVariable = viewQry.getSqlVariable();

// 0、检查视图名
checkViewName(viewName);

// 1、获取视图详情
ViewDetailDTO viewDetailDTO = viewDetailDTO(viewName);
String sqlTemplate = viewDetailDTO.getScript();
String sourceId = viewDetailDTO.getSourceId();

// 2、构建 sql
String sql = buildSql(sqlTemplate, sqlVariable);
return SqlQry.of(sourceId, sql);
}

/**
* 检查视图名
*/
private void checkViewName(String viewName) {
Assert.notBlank(viewName, "viewName=[{}] 不能为空,请设置!!!", viewName);
Assert.isTrue(viewName.startsWith("/"), "viewName=[{}] 必须以 '/' 开头!!!", viewName);
}

/**
* 获取视图详情 DTO
*
* @param viewName 视图名。必须以 "/" 开头。
* @return ViewDetailDTO(包含两个重要属性:sourceId=数据源id,sql=字符串SQL)
*/
private ViewDetailDTO viewDetailDTO(String viewName) {
String redisKey = datartProperties.redisKeyView();
Object redisValue = redis.opsForHash().get(redisKey, viewName);
Assert.notNull(redisValue, "viewName=[{}] 视图查询无果", viewName);
return JSONUtil.toBean(redisValue.toString(), ViewDetailDTO.class);
}

/**
* 根据变量,填充 SQL
*
* @param sqlTemplate sql模版
* @param sqlVariable sql变量
* @return 被变量填充后的 sql
*/
private static String buildSql(String sqlTemplate, SqlVariable sqlVariable) {
sqlTemplate = ifTagCheck(sqlTemplate, sqlVariable);

sqlTemplate = sqlVariableReplace(sqlTemplate, sqlVariable);

return sqlTemplate;
}

/**
* if 标签判断
*/
private static String ifTagCheck(String sqlTemplate, SqlVariable sqlVariable) {
String ifReg = "<if test=\"([\\s\\S]+?)\">([\\s\\S]+?)</if>";
List<String> g0 = ReUtil.findAll(ifReg, sqlTemplate, 0);// 整体
List<String> g1 = ReUtil.findAll(ifReg, sqlTemplate, 1);// test 引号内
List<String> g2 = ReUtil.findAll(ifReg, sqlTemplate, 2);// if 标签体

for (int i = 0; i < g0.size(); i++) {
String find = g0.get(i);
String test = " " + g1.get(i).trim() + " ";// 前后拼接空格,防止 SQL 粘连
String body = " " + g2.get(i).trim() + " ";// body 中的变量,在 buildSql 填充

String expression = "if(test) return body;".replace("test", test);

Dict mvelVars = Dict.create().set("body", body);

// 找到 test 所有变量
// sqlVariable => dict
List<String> testVars = ReUtil.findAllGroup1("\\$(\\w+)\\$", test);
testVars.forEach(e -> mvelVars.set("$" + e + "$", sqlVariable.get(e)));

// test通过,返回 if 标签体
// test不通过,返回 null => ""
String res = MVEL.eval(expression, mvelVars, String.class);// " and storeId in [$storeId$] "
if (StrUtil.isBlank(res)) {
res = "";
}
sqlTemplate = sqlTemplate.replace(find, res);

}
return sqlTemplate;
}

/**
* SQL 变量替换
*/
private static String sqlVariableReplace(String sqlTemplate, SqlVariable sqlVariable) {
if (sqlVariable == null) {
return sqlTemplate;
}

// 找到所有的变量列表
List<String> vars = ReUtil.findAllGroup1("\\$(\\w+)\\$", sqlTemplate);
for (String var : vars) {
String key = "$" + var + "$";
Set<String> value = sqlVariable.get(var);

if (CollectionUtil.isEmpty(value)) {
throw new DataProviderException("找不到变量 变量名=[{}]:sql=[{}]", var, sqlTemplate);
}

sqlTemplate = sqlTemplate.replace(key, CollectionUtil.join(value, ","));
}

return sqlTemplate;
}

}