|
|
|
|
@ -6,22 +6,29 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
|
|
import com.biutag.supervision.common.UserContextHolder; |
|
|
|
|
import com.biutag.supervision.pojo.Result; |
|
|
|
|
import com.biutag.supervision.pojo.entity.Model; |
|
|
|
|
import com.biutag.supervision.pojo.entity.ModelClass; |
|
|
|
|
import com.biutag.supervision.pojo.model.UserAuth; |
|
|
|
|
import com.biutag.supervision.pojo.param.ModelQueryParam; |
|
|
|
|
import com.biutag.supervision.pojo.vo.ModelTree; |
|
|
|
|
import com.biutag.supervision.service.ModelClassService; |
|
|
|
|
import com.biutag.supervision.service.ModelService; |
|
|
|
|
import lombok.RequiredArgsConstructor; |
|
|
|
|
import org.springframework.web.bind.annotation.*; |
|
|
|
|
|
|
|
|
|
import java.time.LocalDateTime; |
|
|
|
|
import java.util.ArrayList; |
|
|
|
|
import java.util.List; |
|
|
|
|
import java.util.Objects; |
|
|
|
|
|
|
|
|
|
@RequestMapping("model") |
|
|
|
|
@RequestMapping("models") |
|
|
|
|
@RequiredArgsConstructor |
|
|
|
|
@RestController |
|
|
|
|
public class ModelController { |
|
|
|
|
|
|
|
|
|
private final ModelService modelService; |
|
|
|
|
|
|
|
|
|
private final ModelClassService modelClassService; |
|
|
|
|
|
|
|
|
|
@GetMapping |
|
|
|
|
public Result<Page<Model>> list(ModelQueryParam queryParam) { |
|
|
|
|
LambdaQueryWrapper<Model> queryWrapper = new LambdaQueryWrapper<>(); |
|
|
|
|
@ -33,6 +40,24 @@ public class ModelController {
|
|
|
|
|
return Result.success(modelService.page(new Page<>(queryParam.getCurrent(), queryParam.getSize()), queryWrapper)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@GetMapping("tree") |
|
|
|
|
public Result<List<ModelTree>> tree() { |
|
|
|
|
List<ModelClass> modelClasses = modelClassService.list(); |
|
|
|
|
List<Model> models = modelService.list(); |
|
|
|
|
List<ModelTree> list = new ArrayList<>(); |
|
|
|
|
list.addAll(modelClasses.stream().map(item -> { |
|
|
|
|
ModelTree node = new ModelTree(); |
|
|
|
|
node.setValue(item.getId()); |
|
|
|
|
node.setLabel(item.getName()); |
|
|
|
|
node.setType("modelClass"); |
|
|
|
|
List<ModelTree> children = models.stream().filter(model -> item.getId().equals(model.getClassId())).map(ModelTree::of).toList(); |
|
|
|
|
node.setChildren(children); |
|
|
|
|
return node; |
|
|
|
|
}).toList()); |
|
|
|
|
list.addAll(models.stream().filter(model -> Objects.isNull(model.getClassId())).map(ModelTree::of).toList()); |
|
|
|
|
return Result.success(list); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@PostMapping |
|
|
|
|
public Result<Void> add(@RequestBody Model model) { |
|
|
|
|
model.setCreateTime(LocalDateTime.now()); |
|
|
|
|
|