Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
K
know-base
概览
概览
详情
活动
周期分析
版本库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
张京坤
know-base
Commits
96c2c882
提交
96c2c882
authored
2月 21, 2024
作者:
chenshiqiang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix
上级
252f3326
隐藏空白字符变更
内嵌
并排
正在显示
3 个修改的文件
包含
37 行增加
和
23 行删除
+37
-23
KnowledgeController.java
...ava/com/zzsn/knowbase/controller/KnowledgeController.java
+24
-16
IKnowledgeService.java
...ain/java/com/zzsn/knowbase/service/IKnowledgeService.java
+1
-1
KnowledgeServiceImpl.java
.../com/zzsn/knowbase/service/impl/KnowledgeServiceImpl.java
+12
-6
没有找到文件。
src/main/java/com/zzsn/knowbase/controller/KnowledgeController.java
浏览文件 @
96c2c882
...
@@ -28,7 +28,10 @@ import javax.servlet.http.HttpServletRequest;
...
@@ -28,7 +28,10 @@ import javax.servlet.http.HttpServletRequest;
import
javax.servlet.http.HttpServletResponse
;
import
javax.servlet.http.HttpServletResponse
;
import
java.io.File
;
import
java.io.File
;
import
java.io.IOException
;
import
java.io.IOException
;
import
java.util.*
;
import
java.util.ArrayList
;
import
java.util.Date
;
import
java.util.List
;
import
java.util.Map
;
import
java.util.concurrent.CompletableFuture
;
import
java.util.concurrent.CompletableFuture
;
/**
/**
...
@@ -66,17 +69,18 @@ public class KnowledgeController {
...
@@ -66,17 +69,18 @@ public class KnowledgeController {
IPage
<
KnowledgeVO
>
pageList
=
knowledgeService
.
queryPageList
(
knowledgeParam
,
pageNo
,
pageSize
,
column
,
order
);
IPage
<
KnowledgeVO
>
pageList
=
knowledgeService
.
queryPageList
(
knowledgeParam
,
pageNo
,
pageSize
,
column
,
order
);
return
Result
.
OK
(
pageList
);
return
Result
.
OK
(
pageList
);
}
}
/**
/**
* 详情
* 详情
*/
*/
@GetMapping
(
value
=
"/getById"
)
@GetMapping
(
value
=
"/getById"
)
public
Result
<?>
getById
(
@RequestParam
String
id
)
{
public
Result
<?>
getById
(
@RequestParam
String
id
)
{
log
.
info
(
"{}===query begin"
,
new
Date
());
log
.
info
(
"{}===query begin"
,
new
Date
());
KnowledgeVO
knowledgeVO
=
knowledgeService
.
getById
(
id
);
KnowledgeVO
knowledgeVO
=
knowledgeService
.
getById
(
id
);
if
(
null
!=
knowledgeVO
.
getPublishDate
()&&(!
knowledgeVO
.
getPublishDate
().
isEmpty
()))
{
if
(
null
!=
knowledgeVO
.
getPublishDate
()
&&
(!
knowledgeVO
.
getPublishDate
().
isEmpty
()))
{
knowledgeVO
.
setPublishDate
(
EsDateUtil
.
esFieldDateMapping
(
knowledgeVO
.
getPublishDate
()));
knowledgeVO
.
setPublishDate
(
EsDateUtil
.
esFieldDateMapping
(
knowledgeVO
.
getPublishDate
()));
}
}
if
(
null
!=
knowledgeVO
.
getVerifyTime
()&&(!
knowledgeVO
.
getVerifyTime
().
isEmpty
()))
{
if
(
null
!=
knowledgeVO
.
getVerifyTime
()
&&
(!
knowledgeVO
.
getVerifyTime
().
isEmpty
()))
{
knowledgeVO
.
setVerifyTime
(
EsDateUtil
.
esFieldDateMapping
(
knowledgeVO
.
getVerifyTime
()));
knowledgeVO
.
setVerifyTime
(
EsDateUtil
.
esFieldDateMapping
(
knowledgeVO
.
getVerifyTime
()));
}
}
return
Result
.
OK
(
knowledgeVO
);
return
Result
.
OK
(
knowledgeVO
);
...
@@ -106,24 +110,26 @@ public class KnowledgeController {
...
@@ -106,24 +110,26 @@ public class KnowledgeController {
Map
<
String
,
MultipartFile
>
fileMap
=
multipartRequest
.
getFileMap
();
Map
<
String
,
MultipartFile
>
fileMap
=
multipartRequest
.
getFileMap
();
Result
<
List
<
KnowFile
>>
result
=
localFileService
.
upload
(
fileMap
);
Result
<
List
<
KnowFile
>>
result
=
localFileService
.
upload
(
fileMap
);
if
(!
Integer
.
valueOf
(
"200"
).
equals
(
result
.
getCode
())){
log
.
info
(
"upload result:{}"
,
result
);
if
(!
Integer
.
valueOf
(
"200"
).
equals
(
result
.
getCode
()))
{
return
Result
.
error
(
result
.
getMessage
());
return
Result
.
error
(
result
.
getMessage
());
}
}
List
<
KnowFile
>
resultList
=
result
.
getResult
();
List
<
KnowFile
>
resultList
=
result
.
getResult
();
KbAuthorizedUser
userInfo
=
SpringContextUtils
.
getUserInfo
();
KbAuthorizedUser
userInfo
=
SpringContextUtils
.
getUserInfo
();
knowledge
.
setImportData
(
0
);
knowledge
.
setImportData
(
0
);
CompletableFuture
.
runAsync
(()
->
this
.
add
(
resultList
,
knowledge
,
userInfo
));
CompletableFuture
.
runAsync
(()
->
this
.
add
(
resultList
,
knowledge
,
userInfo
));
return
Result
.
OK
(
"添加成功!"
);
return
Result
.
OK
(
"添加成功!"
);
}
}
private
void
add
(
List
<
KnowFile
>
resultList
,
Knowledge
knowledge
,
KbAuthorizedUser
userInfo
){
private
void
add
(
List
<
KnowFile
>
resultList
,
Knowledge
knowledge
,
KbAuthorizedUser
userInfo
)
{
if
(
null
==
resultList
||
resultList
.
isEmpty
())
{
if
(
null
==
resultList
||
resultList
.
isEmpty
())
{
return
;
return
;
}
}
for
(
KnowFile
knowFileResult
:
resultList
)
{
for
(
KnowFile
knowFileResult
:
resultList
)
{
knowledge
.
setId
(
null
);
knowledge
.
setId
(
null
);
knowledge
.
setTitle
(
null
);
knowledge
.
setTitle
(
null
);
knowledgeService
.
addKnowledge
(
knowFileResult
,
knowledge
,
userInfo
);
knowledgeService
.
addKnowledge
(
knowFileResult
,
knowledge
,
userInfo
);
}
}
}
}
...
@@ -147,10 +153,11 @@ public class KnowledgeController {
...
@@ -147,10 +153,11 @@ public class KnowledgeController {
* @return
* @return
*/
*/
@DeleteMapping
(
value
=
"/delete"
)
@DeleteMapping
(
value
=
"/delete"
)
public
Result
<?>
delete
(
@RequestParam
(
name
=
"ids"
)
String
ids
)
{
public
Result
<?>
delete
(
@RequestParam
(
name
=
"ids"
)
String
ids
,
@RequestParam
(
name
=
"knowledgeBaseId"
)
String
knowledgeBaseId
)
{
knowledgeService
.
deleteKnowledge
(
ids
);
knowledgeService
.
deleteKnowledge
(
ids
,
knowledgeBaseId
);
return
Result
.
OK
(
"删除成功!"
);
return
Result
.
OK
(
"删除成功!"
);
}
}
@GetMapping
(
value
=
"/parse"
)
@GetMapping
(
value
=
"/parse"
)
public
Result
<?>
parse
(
@RequestParam
(
name
=
"filePath"
)
String
filePath
)
throws
IOException
{
public
Result
<?>
parse
(
@RequestParam
(
name
=
"filePath"
)
String
filePath
)
throws
IOException
{
String
s
=
DocUtil
.
docParseHtml
(
filePath
);
String
s
=
DocUtil
.
docParseHtml
(
filePath
);
...
@@ -166,7 +173,7 @@ public class KnowledgeController {
...
@@ -166,7 +173,7 @@ public class KnowledgeController {
@PostMapping
(
value
=
"/IntelligentQa"
)
@PostMapping
(
value
=
"/IntelligentQa"
)
public
Result
<?>
intelligentQa
(
@RequestBody
IntelligentQaParam
intelligentQaParam
)
throws
IOException
{
public
Result
<?>
intelligentQa
(
@RequestBody
IntelligentQaParam
intelligentQaParam
)
throws
IOException
{
JSONObject
params
=
new
JSONObject
();
JSONObject
params
=
new
JSONObject
();
params
.
put
(
"score_threshold"
,
intelligentQaParam
.
getScoreThreshold
()
);
params
.
put
(
"score_threshold"
,
intelligentQaParam
.
getScoreThreshold
()
);
params
.
put
(
"question"
,
intelligentQaParam
.
getQuestion
());
params
.
put
(
"question"
,
intelligentQaParam
.
getQuestion
());
params
.
put
(
"knowledge_base_id"
,
intelligentQaParam
.
getKbKnowledgeIds
().
split
(
","
));
params
.
put
(
"knowledge_base_id"
,
intelligentQaParam
.
getKbKnowledgeIds
().
split
(
","
));
String
result
=
HttpUtil
.
doPost
(
intelligentQaUrl
,
params
,
120000
);
String
result
=
HttpUtil
.
doPost
(
intelligentQaUrl
,
params
,
120000
);
...
@@ -186,9 +193,10 @@ public class KnowledgeController {
...
@@ -186,9 +193,10 @@ public class KnowledgeController {
public
Result
<?>
importInfo
(
HttpServletRequest
request
,
HttpServletResponse
response
)
{
public
Result
<?>
importInfo
(
HttpServletRequest
request
,
HttpServletResponse
response
)
{
KbAuthorizedUser
userInfo
=
SpringContextUtils
.
getUserInfo
();
KbAuthorizedUser
userInfo
=
SpringContextUtils
.
getUserInfo
();
assert
userInfo
!=
null
;
assert
userInfo
!=
null
;
knowledgeService
.
doImportInfo
(
request
,
userInfo
.
getId
());
knowledgeService
.
doImportInfo
(
request
,
userInfo
.
getId
());
return
Result
.
OK
(
"导入正在后台进行"
);
return
Result
.
OK
(
"导入正在后台进行"
);
}
}
/**
/**
* 下载导入模板
* 下载导入模板
*/
*/
...
...
src/main/java/com/zzsn/knowbase/service/IKnowledgeService.java
浏览文件 @
96c2c882
...
@@ -31,7 +31,7 @@ public interface IKnowledgeService {
...
@@ -31,7 +31,7 @@ public interface IKnowledgeService {
/**
/**
* 删除
* 删除
*/
*/
void
deleteKnowledge
(
String
ids
);
void
deleteKnowledge
(
String
ids
,
String
knowledgeBaseId
);
/**
/**
* 分页检索
* 分页检索
...
...
src/main/java/com/zzsn/knowbase/service/impl/KnowledgeServiceImpl.java
浏览文件 @
96c2c882
...
@@ -46,7 +46,10 @@ import org.springframework.web.multipart.MultipartFile;
...
@@ -46,7 +46,10 @@ import org.springframework.web.multipart.MultipartFile;
import
org.springframework.web.multipart.MultipartHttpServletRequest
;
import
org.springframework.web.multipart.MultipartHttpServletRequest
;
import
javax.servlet.http.HttpServletRequest
;
import
javax.servlet.http.HttpServletRequest
;
import
java.io.*
;
import
java.io.ByteArrayInputStream
;
import
java.io.File
;
import
java.io.FileInputStream
;
import
java.io.IOException
;
import
java.util.*
;
import
java.util.*
;
import
java.util.concurrent.CompletableFuture
;
import
java.util.concurrent.CompletableFuture
;
import
java.util.stream.Collectors
;
import
java.util.stream.Collectors
;
...
@@ -226,7 +229,7 @@ class KnowledgeServiceImpl implements IKnowledgeService {
...
@@ -226,7 +229,7 @@ class KnowledgeServiceImpl implements IKnowledgeService {
esOpUtil
.
docEditByEntity
(
Constants
.
ES_DATA_FOR_KNOWLEDGE
,
knowledge
.
getId
(),
knowledge
);
esOpUtil
.
docEditByEntity
(
Constants
.
ES_DATA_FOR_KNOWLEDGE
,
knowledge
.
getId
(),
knowledge
);
}
}
//删除
//删除
deleteForPython
(
knowledge
.
getId
());
deleteForPython
(
knowledge
.
getId
()
,
knowledge
.
getKnowledgeProjectId
()
);
BeanUtils
.
copyProperties
(
knowledge
,
knowledgeMessage
);
BeanUtils
.
copyProperties
(
knowledge
,
knowledgeMessage
);
knowledgeMessage
.
setType
(
knowledge
.
getTypeId
());
knowledgeMessage
.
setType
(
knowledge
.
getTypeId
());
...
@@ -240,7 +243,7 @@ class KnowledgeServiceImpl implements IKnowledgeService {
...
@@ -240,7 +243,7 @@ class KnowledgeServiceImpl implements IKnowledgeService {
}
}
@Override
@Override
public
void
deleteKnowledge
(
String
ids
)
{
public
void
deleteKnowledge
(
String
ids
,
String
knowledgeBaseId
)
{
String
[]
split
=
ids
.
split
(
","
);
String
[]
split
=
ids
.
split
(
","
);
for
(
String
id
:
split
)
{
for
(
String
id
:
split
)
{
esOpUtil
.
docDeleteById
(
Constants
.
ES_DATA_FOR_KNOWLEDGE
,
id
);
esOpUtil
.
docDeleteById
(
Constants
.
ES_DATA_FOR_KNOWLEDGE
,
id
);
...
@@ -248,16 +251,19 @@ class KnowledgeServiceImpl implements IKnowledgeService {
...
@@ -248,16 +251,19 @@ class KnowledgeServiceImpl implements IKnowledgeService {
CompletableFuture
.
runAsync
(()
->
{
CompletableFuture
.
runAsync
(()
->
{
for
(
String
id
:
split
)
{
for
(
String
id
:
split
)
{
deleteForPython
(
id
);
deleteForPython
(
id
,
knowledgeBaseId
);
}
}
});
});
}
}
public
void
deleteForPython
(
String
id
)
{
public
void
deleteForPython
(
String
id
,
String
knowledgeBaseId
)
{
JSONObject
params
=
new
JSONObject
();
JSONObject
params
=
new
JSONObject
();
params
.
put
(
"knowledge_base_id"
,
id
);
List
<
String
>
lidList
=
new
ArrayList
<>();
lidList
.
add
(
id
);
params
.
put
(
"id"
,
lidList
);
params
.
put
(
"knowledge_base_id"
,
knowledgeBaseId
);
try
{
try
{
HttpUtil
.
doPost
(
deleteUrl
,
params
,
120000
);
HttpUtil
.
doPost
(
deleteUrl
,
params
,
120000
);
}
catch
(
IOException
e
)
{
}
catch
(
IOException
e
)
{
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论