package com.hx.other.service.controller.admin;
|
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.hx.common.BaseController;
|
import com.hx.mybatisTool.SqlSentence;
|
import com.hx.other.service.model.BaseEntity;
|
import com.hx.other.service.model.QueryRecord;
|
import com.hx.other.service.model.SqlQueryTemp;
|
import com.hx.other.service.service.QueryRecordService;
|
import com.hx.other.service.service.SqlQueryTempService;
|
import com.hx.other.service.vo.BaseVo;
|
import com.hx.other.service.vo.ai.QueryRecordVo;
|
import com.hx.resultTool.Result;
|
import com.hx.util.StringUtils;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.annotation.Resource;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* @Author: cmg
|
* @Date: 2023/5/5 14:37
|
*/
|
@RestController
|
@RequestMapping("/ai/query/record")
|
public class AdminQueryRecordController extends BaseController {
|
|
@Resource
|
private QueryRecordService queryRecordService;
|
|
@Resource
|
private SqlQueryTempService sqlQueryTempService;
|
|
@PostMapping("/list")
|
public Result list(@RequestBody(required = false) QueryRecordVo baseVo)
|
{
|
if(baseVo == null)
|
{
|
baseVo = new QueryRecordVo();
|
}
|
|
if(baseVo.getPageNum() == null || baseVo.getPageNum() <= 0)
|
{
|
baseVo.setPageNum(1);
|
}
|
|
if(baseVo.getPageSize() == null || baseVo.getPageSize() <= 0 || baseVo.getPageSize() > 100)
|
{
|
baseVo.setPageSize(20);
|
}
|
|
SqlSentence sqlSentence = new SqlSentence();
|
Map<String, Object> map = new HashMap<>();
|
sqlSentence.setM(map);
|
map.put("isDel", BaseEntity.NO);
|
|
StringBuilder stringBuilder = new StringBuilder();
|
stringBuilder.append("select * from query_record where isDel = #{m.isDel}");
|
|
if(!StringUtils.isEmpty(baseVo.getKeyWord()))
|
{
|
stringBuilder.append(" and content like #{m.content} ");
|
map.put("content", "%" + baseVo.getKeyWord() + "%");
|
}
|
|
if(baseVo.getIsFromTemp() != null)
|
{
|
stringBuilder.append(" and isFromQuery = #{m.isFromQuery} ");
|
map.put("isFromQuery", baseVo.getIsFromTemp());
|
}
|
|
sqlSentence.setSqlSentence(stringBuilder.toString());
|
PageHelper.startPage(baseVo.getPageNum(), baseVo.getPageSize(), " createTime desc ");
|
List<QueryRecord> list = queryRecordService.selectList(sqlSentence);
|
PageInfo<QueryRecord> pageInfo = new PageInfo<>(list);
|
|
return Result.success(pageInfo);
|
}
|
|
@PostMapping("/change/temp")
|
public Result changeToTemp(@RequestBody(required = false) BaseVo baseVo)
|
{
|
if(baseVo == null || StringUtils.isEmpty(baseVo.getId()))
|
{
|
throwParamException("请选择记录");
|
}
|
|
QueryRecord queryRecord = queryRecordService.selectOneByKey(baseVo.getId());
|
if(queryRecord == null)
|
{
|
throwParamException("找不到记录");
|
}
|
|
String sql = sqlQueryTempService.selectByContent(queryRecord.getContent());
|
if(!StringUtils.isEmpty(sql))
|
{
|
throwServiceException("已加入模板");
|
}
|
|
SqlQueryTemp sqlQueryTemp = new SqlQueryTemp();
|
sqlQueryTemp.setSqlStr(queryRecord.getSqlStr());
|
sqlQueryTemp.setContent(queryRecord.getContent());
|
sqlQueryTemp.setIsUp(BaseEntity.YES);
|
sqlQueryTempService.insert(sqlQueryTemp);
|
|
return Result.success();
|
}
|
}
|
