diff --git a/src/main/java/com/biutag/supervision/pojo/entity/Model.java b/src/main/java/com/biutag/supervision/pojo/entity/Model.java index 9e11ac8..5e66b23 100644 --- a/src/main/java/com/biutag/supervision/pojo/entity/Model.java +++ b/src/main/java/com/biutag/supervision/pojo/entity/Model.java @@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.annotation.IdType; import com.baomidou.mybatisplus.annotation.TableField; import com.baomidou.mybatisplus.annotation.TableId; import com.fasterxml.jackson.annotation.JsonFormat; +import io.swagger.v3.oas.annotations.media.Schema; import lombok.Getter; import lombok.Setter; @@ -139,4 +140,14 @@ public class Model { // 预警内容生成器 private String thingDescGeneration; + /** + * @see com.biutag.supervision.constants.enums.ProblemSourcesEnum + */ + @Schema(description = "模型问题来源") + @TableField("problem_sources_code") + private String problemSourcesCode; + + @Schema(description = "创建人") + @TableField("create_id") + private String createId; } \ No newline at end of file diff --git a/src/main/java/com/biutag/supervision/service/ModelService.java b/src/main/java/com/biutag/supervision/service/ModelService.java index 11149b2..007b582 100644 --- a/src/main/java/com/biutag/supervision/service/ModelService.java +++ b/src/main/java/com/biutag/supervision/service/ModelService.java @@ -56,6 +56,7 @@ public class ModelService extends ServiceImpl { UserAuth user = UserContextHolder.getCurrentUser(); model.setCreateDepartId(user.getDepartId()); model.setCreateDepartName(user.getDepartName()); + model.setCreateId(user.getUserName()); boolean save = save(model); if ("2".equals(modelDto.getModelingMethod()) && modelDto.getModelGeneration() != null) { modelDto.getModelGeneration().setModelId(model.getId());