topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 20:51:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][headless] add validateAndQuery interface in SqlQueryApiController

Browse Source
This commit is contained in:
lexluo09
2024-11-27 15:03:00 +08:00
parent dad065d0ba
commit 25559fdaa5
1 changed files with 29 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
headless/server/src/main/java/com/tencent/supersonic/headless/server/facade/rest/SqlQueryApiController.java
View File

@@ -6,6 +6,7 @@ import javax.servlet.http.HttpServletResponse;
import com.tencent.supersonic.auth.api.authentication.utils.UserHolder;
import com.tencent.supersonic.common.pojo.User;
import com.tencent.supersonic.common.util.StringUtil;
import com.tencent.supersonic.headless.api.pojo.SqlEvaluation;
import com.tencent.supersonic.headless.api.pojo.request.QuerySqlReq;
import com.tencent.supersonic.headless.api.pojo.request.QuerySqlsReq;
import com.tencent.supersonic.headless.api.pojo.request.SemanticQueryReq;
@@ -102,4 +103,28 @@ public class SqlQueryApiController {
querySqlReq.setSql(StringUtil.replaceBackticks(sql));
return chatLayerService.validate(querySqlReq, user);
}
@PostMapping("/validateAndQuery")
public Object validateAndQuery(@RequestBody QuerySqlsReq querySqlsReq,
HttpServletRequest request, HttpServletResponse response) throws Exception {
User user = UserHolder.findUser(request, response);
List<QuerySqlReq> convert = convert(querySqlsReq);
for (QuerySqlReq querySqlReq : convert) {
SqlEvaluation validate = chatLayerService.validate(querySqlReq, user);
if (!validate.getIsValidated()) {
throw new Exception(validate.getValidateMsg());
}
}
return queryBySqls(querySqlsReq, request, response);
}
private List<QuerySqlReq> convert(QuerySqlsReq querySqlsReq) {
return querySqlsReq.getSqls().stream().map(sql -> {
QuerySqlReq querySqlReq = new QuerySqlReq();
BeanUtils.copyProperties(querySqlsReq, querySqlReq);
querySqlReq.setSql(StringUtil.replaceBackticks(sql));
return querySqlReq;
}).collect(Collectors.toList());
}
}