center/dataease-dm
center/dataease-dm
13
0
Fork 0
forked from github/dataease
Code Pull Requests Activity

Merge pull request #10864 from dataease/pr@dev-v2@feat_copilot

Browse Source 
feat: 新增copilot所需接口
This commit is contained in:
Junjun 2024-07-09 17:31:12 +08:00 committed by GitHub
commit 3f00a2e3b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
core/core-backend/src/main/java/io/dataease/copilot/manage/CopilotManage.java
View File

@ -1,10 +1,7 @@
package io.dataease.copilot.manage;
import com.fasterxml.jackson.core.type.TypeReference;
import io.dataease.api.copilot.dto.DESendDTO;
import io.dataease.api.copilot.dto.MsgDTO;
import io.dataease.api.copilot.dto.ReceiveDTO;
import io.dataease.api.copilot.dto.TokenDTO;
import io.dataease.api.copilot.dto.*;
import io.dataease.api.dataset.union.DatasetGroupInfoDTO;
import io.dataease.api.dataset.union.UnionDTO;
import io.dataease.chart.utils.ChartDataBuild;
@ -176,7 +173,8 @@ public class CopilotManage {
// String querySQL = SQLProvider.createQuerySQL(sqlMeta, false, false, needOrder);
// querySQL = provider.rebuildSQL(querySQL, sqlMeta, crossDs, dsMap);
// logger.info("preview sql: " + querySQL);
// todo test
// 无法加行权限的情况下，直接用sql
String querySQL = sql;
String copilotSQL = receiveDTO.getSql();
@ -228,15 +226,11 @@ public class CopilotManage {
Map<String, Object> previewData = buildPreviewData(data, fields, desensitizationList);
map.put("data", previewData);
// map.put("allFields", allFields);// map.data 中包含了fields和data
// if (ObjectUtils.isEmpty(dto.getId())) {
// map.put("allFields", allFields);
// } else {
// List<DatasetTableFieldDTO> fieldList = datasetTableFieldManage.selectByDatasetGroupId(dto.getId());
// map.put("allFields", fieldList);
// }
map.put("sql", Base64.getEncoder().encodeToString(s.getBytes()));
// 重构chart结构，补全缺失字段
rebuildChart(receiveDTO, fields);
MsgDTO result = new MsgDTO();
result.setDatasetGroupId(dto.getId());
result.setMsgType("api");
@ -294,6 +288,17 @@ public class CopilotManage {
return receiveDTO;
}
public void rebuildChart(ReceiveDTO receiveDTO, List<TableField> fields) {
if (StringUtils.equalsIgnoreCase(receiveDTO.getChart().getType(), "pie")) {
if (fields.size() != 2) {
DEException.throwException("当前字段不足以构建饼图");
}
AxisDTO axisDTO = new AxisDTO();
axisDTO.setX(fields.get(0).getOriginName());
axisDTO.setY(fields.get(1).getOriginName());
}
}
public List<MsgDTO> getList(Long userId) {
MsgDTO lastMsg = msgManage.getLastMsg(userId);
if (lastMsg == null) {