feat(llm): 添加服务器主机名称管理功能

- 新增服务器主机名称相关实体、控制器、服务、映射等类
- 实现服务器主机名称的增、删、改、查及分页查询功能
- 添加服务器主机名称列表导出 Excel 功能
- 校验服务器主机名称存在性和 URL
This commit is contained in:
limin 2025-02-18 10:23:54 +08:00
parent b8f8361ab1
commit 31046aa934
11 changed files with 441 additions and 0 deletions

View File

@ -114,4 +114,6 @@ public interface ErrorCodeConstants {
ErrorCode QUESTION_NOT_EXISTS = new ErrorCode(10046, "数据集信息不完整,无法进行评估");
ErrorCode BASE_MODEL_NAME_EXISTS = new ErrorCode(10047, "基座模型名称重复");
ErrorCode SERVER_NAME_NOT_EXISTS = new ErrorCode(10048, "服务器主机名称不存在");
ErrorCode SERVER_NAME_URL_ERROR = new ErrorCode(10049, "主机地址URL格式不正确");
}

View File

@ -0,0 +1,102 @@
package cn.iocoder.yudao.module.llm.controller.admin.servername;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.security.access.prepost.PreAuthorize;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Operation;
import javax.validation.constraints.*;
import javax.validation.*;
import javax.servlet.http.*;
import java.util.*;
import java.io.IOException;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*;
import cn.iocoder.yudao.module.llm.controller.admin.servername.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import cn.iocoder.yudao.module.llm.service.servername.ServerNameService;
@Tag(name = "管理后台 - 服务器主机名称")
@RestController
@RequestMapping("/llm/server-name")
@Validated
public class ServerNameController {
@Resource
private ServerNameService serverNameService;
@PostMapping("/create")
@Operation(summary = "创建服务器主机名称")
// @PreAuthorize("@ss.hasPermission('llm:server-name:create')")
public CommonResult<Long> createServerName(@Valid @RequestBody ServerNameSaveReqVO createReqVO) {
return success(serverNameService.createServerName(createReqVO));
}
@PutMapping("/update")
@Operation(summary = "更新服务器主机名称")
// @PreAuthorize("@ss.hasPermission('llm:server-name:update')")
public CommonResult<Boolean> updateServerName(@Valid @RequestBody ServerNameSaveReqVO updateReqVO) {
serverNameService.updateServerName(updateReqVO);
return success(true);
}
@DeleteMapping("/delete")
@Operation(summary = "删除服务器主机名称")
@Parameter(name = "id", description = "编号", required = true)
// @PreAuthorize("@ss.hasPermission('llm:server-name:delete')")
public CommonResult<Boolean> deleteServerName(@RequestParam("id") Long id) {
serverNameService.deleteServerName(id);
return success(true);
}
@GetMapping("/get")
@Operation(summary = "获得服务器主机名称")
@Parameter(name = "id", description = "编号", required = true, example = "1024")
// @PreAuthorize("@ss.hasPermission('llm:server-name:query')")
public CommonResult<ServerNameRespVO> getServerName(@RequestParam("id") Long id) {
ServerNameDO serverName = serverNameService.getServerName(id);
return success(BeanUtils.toBean(serverName, ServerNameRespVO.class));
}
@GetMapping("/page")
@Operation(summary = "获得服务器主机名称分页")
// @PreAuthorize("@ss.hasPermission('llm:server-name:query')")
public CommonResult<PageResult<ServerNameRespVO>> getServerNamePage(@Valid ServerNamePageReqVO pageReqVO) {
PageResult<ServerNameDO> pageResult = serverNameService.getServerNamePage(pageReqVO);
return success(BeanUtils.toBean(pageResult, ServerNameRespVO.class));
}
@GetMapping("/export-excel")
@Operation(summary = "导出服务器主机名称 Excel")
// @PreAuthorize("@ss.hasPermission('llm:server-name:export')")
@ApiAccessLog(operateType = EXPORT)
public void exportServerNameExcel(@Valid ServerNamePageReqVO pageReqVO,
HttpServletResponse response) throws IOException {
pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
List<ServerNameDO> list = serverNameService.getServerNamePage(pageReqVO).getList();
// 导出 Excel
ExcelUtils.write(response, "服务器主机名称.xls", "数据", ServerNameRespVO.class,
BeanUtils.toBean(list, ServerNameRespVO.class));
}
@GetMapping("/all-list")
@Operation(summary = "获得服务器主机名称分页")
// @PreAuthorize("@ss.hasPermission('llm:server-name:query')")
public CommonResult<List<ServerNameAllVO>> getServerNameAllList() {
List<ServerNameDO> list = serverNameService.getServerNameAllList();
return success(BeanUtils.toBean(list, ServerNameAllVO.class));
}
}

View File

@ -0,0 +1,18 @@
package cn.iocoder.yudao.module.llm.controller.admin.servername.vo;
import com.alibaba.excel.annotation.ExcelProperty;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
@Schema(description = "管理后台 - 服务器主机名称 Response VO")
@Data
public class ServerNameAllVO {
@Schema(description = "主机id", requiredMode = Schema.RequiredMode.REQUIRED, example = "8115")
@ExcelProperty("主机id")
private Long id;
@Schema(description = "显卡服务器名", example = "芋艿")
@ExcelProperty("显卡服务器名")
private String cardServerName;
}

View File

@ -0,0 +1,28 @@
package cn.iocoder.yudao.module.llm.controller.admin.servername.vo;
import lombok.*;
import java.util.*;
import io.swagger.v3.oas.annotations.media.Schema;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
@Schema(description = "管理后台 - 服务器主机名称分页 Request VO")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
public class ServerNamePageReqVO extends PageParam {
@Schema(description = "显卡服务器名", example = "芋艿")
private String cardServerName;
@Schema(description = "主机名称")
private String host;
@Schema(description = "创建时间")
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
private LocalDateTime[] createTime;
}

View File

@ -0,0 +1,31 @@
package cn.iocoder.yudao.module.llm.controller.admin.servername.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.*;
import java.util.*;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import com.alibaba.excel.annotation.*;
@Schema(description = "管理后台 - 服务器主机名称 Response VO")
@Data
@ExcelIgnoreUnannotated
public class ServerNameRespVO {
@Schema(description = "主机id", requiredMode = Schema.RequiredMode.REQUIRED, example = "8115")
@ExcelProperty("主机id")
private Long id;
@Schema(description = "显卡服务器名", example = "芋艿")
@ExcelProperty("显卡服务器名")
private String cardServerName;
@Schema(description = "主机名称")
@ExcelProperty("主机名称")
private String host;
@Schema(description = "创建时间")
@ExcelProperty("创建时间")
private LocalDateTime createTime;
}

View File

@ -0,0 +1,21 @@
package cn.iocoder.yudao.module.llm.controller.admin.servername.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.*;
import java.util.*;
import javax.validation.constraints.*;
@Schema(description = "管理后台 - 服务器主机名称新增/修改 Request VO")
@Data
public class ServerNameSaveReqVO {
@Schema(description = "主机id", requiredMode = Schema.RequiredMode.REQUIRED, example = "8115")
private Long id;
@Schema(description = "显卡服务器名", example = "芋艿")
private String cardServerName;
@Schema(description = "主机名称")
private String host;
}

View File

@ -0,0 +1,39 @@
package cn.iocoder.yudao.module.llm.dal.dataobject.servername;
import lombok.*;
import java.util.*;
import java.time.LocalDateTime;
import java.time.LocalDateTime;
import com.baomidou.mybatisplus.annotation.*;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
/**
* 服务器主机名称 DO
*
* @author 华大大模型
*/
@TableName("llm_server_name")
@KeySequence("llm_server_name_seq") // 用于 OraclePostgreSQLKingbaseDB2H2 数据库的主键自增如果是 MySQL 等数据库可不写
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ServerNameDO extends BaseDO {
/**
* 主机id
*/
@TableId
private Long id;
/**
* 显卡服务器名
*/
private String cardServerName;
/**
* 主机名称
*/
private String host;
}

View File

@ -0,0 +1,28 @@
package cn.iocoder.yudao.module.llm.dal.mysql.servername;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import org.apache.ibatis.annotations.Mapper;
import cn.iocoder.yudao.module.llm.controller.admin.servername.vo.*;
/**
* 服务器主机名称 Mapper
*
* @author 华大大模型
*/
@Mapper
public interface ServerNameMapper extends BaseMapperX<ServerNameDO> {
default PageResult<ServerNameDO> selectPage(ServerNamePageReqVO reqVO) {
return selectPage(reqVO, new LambdaQueryWrapperX<ServerNameDO>()
.likeIfPresent(ServerNameDO::getCardServerName, reqVO.getCardServerName())
.eqIfPresent(ServerNameDO::getHost, reqVO.getHost())
.betweenIfPresent(ServerNameDO::getCreateTime, reqVO.getCreateTime())
.orderByDesc(ServerNameDO::getId));
}
}

View File

@ -0,0 +1,61 @@
package cn.iocoder.yudao.module.llm.service.servername;
import java.util.*;
import javax.validation.*;
import cn.iocoder.yudao.module.llm.controller.admin.servername.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
/**
* 服务器主机名称 Service 接口
*
* @author 华大大模型
*/
public interface ServerNameService {
/**
* 创建服务器主机名称
*
* @param createReqVO 创建信息
* @return 编号
*/
Long createServerName(@Valid ServerNameSaveReqVO createReqVO);
/**
* 更新服务器主机名称
*
* @param updateReqVO 更新信息
*/
void updateServerName(@Valid ServerNameSaveReqVO updateReqVO);
/**
* 删除服务器主机名称
*
* @param id 编号
*/
void deleteServerName(Long id);
/**
* 获得服务器主机名称
*
* @param id 编号
* @return 服务器主机名称
*/
ServerNameDO getServerName(Long id);
/**
* 获得服务器主机名称分页
*
* @param pageReqVO 分页查询
* @return 服务器主机名称分页
*/
PageResult<ServerNameDO> getServerNamePage(ServerNamePageReqVO pageReqVO);
/**
* 获取所有主机
* @return
*/
List<ServerNameDO> getServerNameAllList();
}

View File

@ -0,0 +1,99 @@
package cn.iocoder.yudao.module.llm.service.servername;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import cn.iocoder.yudao.module.llm.controller.admin.servername.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.llm.dal.mysql.servername.ServerNameMapper;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.*;
/**
* 服务器主机名称 Service 实现类
*
* @author 华大大模型
*/
@Service
@Validated
public class ServerNameServiceImpl implements ServerNameService {
private static final String URL_PATTERN =
"^(http|https)://([a-zA-Z0-9-]+(\\.[a-zA-Z0-9-]+)+|\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3})(:\\d{1,5})?$";
@Resource
private ServerNameMapper serverNameMapper;
@Override
public Long createServerName(ServerNameSaveReqVO createReqVO) {
// 插入
if (!isValidURL(createReqVO.getHost())) {
throw exception(SERVER_NAME_URL_ERROR);
}
ServerNameDO serverName = BeanUtils.toBean(createReqVO, ServerNameDO.class);
serverNameMapper.insert(serverName);
// 返回
return serverName.getId();
}
@Override
public void updateServerName(ServerNameSaveReqVO updateReqVO) {
// 校验存在
validateServerNameExists(updateReqVO.getId());
if (!isValidURL(updateReqVO.getHost())) {
throw exception(SERVER_NAME_URL_ERROR);
}
// 更新
ServerNameDO updateObj = BeanUtils.toBean(updateReqVO, ServerNameDO.class);
serverNameMapper.updateById(updateObj);
}
@Override
public void deleteServerName(Long id) {
// 校验存在
validateServerNameExists(id);
// 删除
serverNameMapper.deleteById(id);
}
private void validateServerNameExists(Long id) {
if (serverNameMapper.selectById(id) == null) {
throw exception(SERVER_NAME_NOT_EXISTS);
}
}
@Override
public ServerNameDO getServerName(Long id) {
return serverNameMapper.selectById(id);
}
@Override
public PageResult<ServerNameDO> getServerNamePage(ServerNamePageReqVO pageReqVO) {
return serverNameMapper.selectPage(pageReqVO);
}
@Override
public List<ServerNameDO> getServerNameAllList() {
List<ServerNameDO> serverNameDOS = serverNameMapper.selectList();
return serverNameDOS;
}
/**
* 校验url
* @param url
* @return
*/
public static boolean isValidURL(String url) {
return url.matches(URL_PATTERN);
}
}

View File

@ -0,0 +1,12 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.servername.ServerNameMapper">
<!--
一般情况下,尽可能使用 Mapper 进行 CRUD 增删改查即可。
无法满足的场景,例如说多表关联查询,才使用 XML 编写 SQL。
代码生成器暂时只生成 Mapper XML 文件本身,更多推荐 MybatisX 快速开发插件来生成查询。
文档可见https://www.iocoder.cn/MyBatis/x-plugins/
-->
</mapper>