diff --git a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java index 4c14b13b1..035b54fdb 100644 --- a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java +++ b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java @@ -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格式不正确"); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/ServerNameController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/ServerNameController.java new file mode 100644 index 000000000..b0f7814b6 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/ServerNameController.java @@ -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 createServerName(@Valid @RequestBody ServerNameSaveReqVO createReqVO) { + return success(serverNameService.createServerName(createReqVO)); + } + + @PutMapping("/update") + @Operation(summary = "更新服务器主机名称") +// @PreAuthorize("@ss.hasPermission('llm:server-name:update')") + public CommonResult 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 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 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> getServerNamePage(@Valid ServerNamePageReqVO pageReqVO) { + PageResult 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 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> getServerNameAllList() { + List list = serverNameService.getServerNameAllList(); + return success(BeanUtils.toBean(list, ServerNameAllVO.class)); + } + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameAllVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameAllVO.java new file mode 100644 index 000000000..d4383ccbb --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameAllVO.java @@ -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; +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNamePageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNamePageReqVO.java new file mode 100644 index 000000000..c487ea7ba --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNamePageReqVO.java @@ -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; + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameRespVO.java new file mode 100644 index 000000000..955d686a4 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameRespVO.java @@ -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; + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameSaveReqVO.java new file mode 100644 index 000000000..1ed1ce9f5 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/servername/vo/ServerNameSaveReqVO.java @@ -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; + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/servername/ServerNameDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/servername/ServerNameDO.java new file mode 100644 index 000000000..9475b8e51 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/servername/ServerNameDO.java @@ -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") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 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; + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/servername/ServerNameMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/servername/ServerNameMapper.java new file mode 100644 index 000000000..714fb378a --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/servername/ServerNameMapper.java @@ -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 { + + default PageResult selectPage(ServerNamePageReqVO reqVO) { + return selectPage(reqVO, new LambdaQueryWrapperX() + .likeIfPresent(ServerNameDO::getCardServerName, reqVO.getCardServerName()) + .eqIfPresent(ServerNameDO::getHost, reqVO.getHost()) + .betweenIfPresent(ServerNameDO::getCreateTime, reqVO.getCreateTime()) + .orderByDesc(ServerNameDO::getId)); + } + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameService.java new file mode 100644 index 000000000..696ddc009 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameService.java @@ -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 getServerNamePage(ServerNamePageReqVO pageReqVO); + + /** + * 获取所有主机 + * @return + */ + List getServerNameAllList(); + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameServiceImpl.java new file mode 100644 index 000000000..0fbf0916e --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/servername/ServerNameServiceImpl.java @@ -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 getServerNamePage(ServerNamePageReqVO pageReqVO) { + return serverNameMapper.selectPage(pageReqVO); + } + + @Override + public List getServerNameAllList() { + List serverNameDOS = serverNameMapper.selectList(); + return serverNameDOS; + } + + /** + * 校验url + * @param url + * @return + */ + public static boolean isValidURL(String url) { + return url.matches(URL_PATTERN); + } + +} \ No newline at end of file diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/servername/ServerNameMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/servername/ServerNameMapper.xml new file mode 100644 index 000000000..8a2c232c1 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/servername/ServerNameMapper.xml @@ -0,0 +1,12 @@ + + + + + + + \ No newline at end of file