mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-11 12:07:42 +00:00
[opt](headless)Deletion of chat model and database should record user.
This commit is contained in:
@@ -46,8 +46,10 @@ public class ChatModelController {
|
||||
}
|
||||
|
||||
@DeleteMapping("/{id}")
|
||||
public boolean deleteModel(@PathVariable("id") Integer id) {
|
||||
chatModelService.deleteChatModel(id);
|
||||
public boolean deleteModel(@PathVariable("id") Integer id,
|
||||
HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse) {
|
||||
User user = UserHolder.findUser(httpServletRequest, httpServletResponse);
|
||||
chatModelService.deleteChatModel(id, user);
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
@@ -64,8 +64,10 @@ public class DatabaseController {
|
||||
}
|
||||
|
||||
@DeleteMapping("/{id}")
|
||||
public boolean deleteDatabase(@PathVariable("id") Long id) {
|
||||
databaseService.deleteDatabase(id);
|
||||
public boolean deleteDatabase(@PathVariable("id") Long id, HttpServletRequest request,
|
||||
HttpServletResponse response) {
|
||||
User user = UserHolder.findUser(request, response);
|
||||
databaseService.deleteDatabase(id, user);
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
@@ -34,7 +34,7 @@ public interface DatabaseService {
|
||||
|
||||
List<DatabaseResp> getDatabaseList(User user);
|
||||
|
||||
void deleteDatabase(Long databaseId);
|
||||
void deleteDatabase(Long databaseId, User user);
|
||||
|
||||
List<String> getCatalogs(Long id) throws SQLException;
|
||||
|
||||
|
||||
@@ -138,7 +138,12 @@ public class DatabaseServiceImpl extends ServiceImpl<DatabaseDOMapper, DatabaseD
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deleteDatabase(Long databaseId) {
|
||||
public void deleteDatabase(Long databaseId, User user) {
|
||||
DatabaseResp databaseResp = getDatabase(databaseId);
|
||||
if (!checkAdminPermission(user, databaseResp)) {
|
||||
throw new RuntimeException("没有权限删除该数据库");
|
||||
}
|
||||
|
||||
ModelFilter modelFilter = new ModelFilter();
|
||||
modelFilter.setDatabaseId(databaseId);
|
||||
modelFilter.setIncludesDetail(false);
|
||||
|
||||
Reference in New Issue
Block a user