63 lines
3.8 KiB
Java
63 lines
3.8 KiB
Java
package com.bruce.common.enumconfig;
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
import com.bruce.common.domain.entity.SysEnum;
|
|
import com.bruce.common.enums.CommonStatusEnum;
|
|
import com.bruce.common.enums.EnableStatusEnum;
|
|
import com.bruce.common.service.ISysEnumService;
|
|
import com.bruce.rag.enums.RagIndexStatusEnum;
|
|
import com.bruce.rag.enums.RagParseStatusEnum;
|
|
import org.junit.jupiter.api.Test;
|
|
import org.junit.jupiter.api.condition.EnabledIfSystemProperty;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
|
|
@SpringBootTest
|
|
@EnabledIfSystemProperty(named = "runEnumInit", matches = "true")
|
|
class SysEnumDataInitTests {
|
|
|
|
@Autowired
|
|
private ISysEnumService sysEnumService;
|
|
|
|
@Test
|
|
public void initDefaultEnums() {
|
|
saveOrUpdate("common", "enable_status", EnableStatusEnum.DISABLED.getLabel(), EnableStatusEnum.DISABLED.getValue(), 0, "通用启用状态");
|
|
saveOrUpdate("common", "enable_status", EnableStatusEnum.ENABLED.getLabel(), EnableStatusEnum.ENABLED.getValue(), 1, "通用启用状态");
|
|
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.DISABLED.getLabel(), CommonStatusEnum.DISABLED.getValue(), 0, "通用状态");
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.ENABLED.getLabel(), CommonStatusEnum.ENABLED.getValue(), 1, "通用状态");
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.DRAFT.getLabel(), CommonStatusEnum.DRAFT.getValue(), 2, "通用状态");
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.PROCESSING.getLabel(), CommonStatusEnum.PROCESSING.getValue(), 3, "通用状态");
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.COMPLETED.getLabel(), CommonStatusEnum.COMPLETED.getValue(), 4, "通用状态");
|
|
saveOrUpdate("common", "common_status", CommonStatusEnum.FAILED.getLabel(), CommonStatusEnum.FAILED.getValue(), 5, "通用状态");
|
|
|
|
saveOrUpdate("rag", "parse_status", RagParseStatusEnum.UPLOADED.getLabel(), RagParseStatusEnum.UPLOADED.getValue(), 1, "RAG文档解析状态");
|
|
saveOrUpdate("rag", "parse_status", RagParseStatusEnum.PARSING.getLabel(), RagParseStatusEnum.PARSING.getValue(), 2, "RAG文档解析状态");
|
|
saveOrUpdate("rag", "parse_status", RagParseStatusEnum.PARSED.getLabel(), RagParseStatusEnum.PARSED.getValue(), 3, "RAG文档解析状态");
|
|
saveOrUpdate("rag", "parse_status", RagParseStatusEnum.FAILED.getLabel(), RagParseStatusEnum.FAILED.getValue(), 4, "RAG文档解析状态");
|
|
|
|
saveOrUpdate("rag", "index_status", RagIndexStatusEnum.PENDING.getLabel(), RagIndexStatusEnum.PENDING.getValue(), 1, "RAG文档索引状态");
|
|
saveOrUpdate("rag", "index_status", RagIndexStatusEnum.INDEXING.getLabel(), RagIndexStatusEnum.INDEXING.getValue(), 2, "RAG文档索引状态");
|
|
saveOrUpdate("rag", "index_status", RagIndexStatusEnum.INDEXED.getLabel(), RagIndexStatusEnum.INDEXED.getValue(), 3, "RAG文档索引状态");
|
|
saveOrUpdate("rag", "index_status", RagIndexStatusEnum.FAILED.getLabel(), RagIndexStatusEnum.FAILED.getValue(), 4, "RAG文档索引状态");
|
|
}
|
|
|
|
private void saveOrUpdate(String catalog, String type, String name, Integer value, Integer sort, String remark) {
|
|
SysEnum sysEnum = sysEnumService.getOne(new LambdaQueryWrapper<SysEnum>()
|
|
.eq(SysEnum::getCatalog, catalog)
|
|
.eq(SysEnum::getType, type)
|
|
.eq(SysEnum::getName, name));
|
|
if (sysEnum == null) {
|
|
sysEnum = new SysEnum();
|
|
}
|
|
sysEnum.setCatalog(catalog);
|
|
sysEnum.setType(type);
|
|
sysEnum.setName(name);
|
|
sysEnum.setValue(value);
|
|
sysEnum.setStrvalue(null);
|
|
sysEnum.setSort(sort);
|
|
sysEnum.setRemark(remark);
|
|
sysEnumService.saveOrUpdate(sysEnum);
|
|
}
|
|
}
|