diff --git a/src/main/java/com/iflytop/sgs/app/model/bo/status/device/SolutionModuleState.java b/src/main/java/com/iflytop/sgs/app/model/bo/status/device/SolutionModuleState.java index 512f8ba..f537a23 100644 --- a/src/main/java/com/iflytop/sgs/app/model/bo/status/device/SolutionModuleState.java +++ b/src/main/java/com/iflytop/sgs/app/model/bo/status/device/SolutionModuleState.java @@ -1,5 +1,6 @@ package com.iflytop.sgs.app.model.bo.status.device; +import com.iflytop.sgs.common.enums.ContainerCode; import io.swagger.v3.oas.annotations.media.Schema; import lombok.Data; @@ -24,7 +25,13 @@ public class SolutionModuleState { @Schema(description = "电磁阀状态") private ValveState valveState; + /* + * 根据容器code获取容器状态 + * */ + public SolutionContainerState getContainerStateByCode(ContainerCode containerCode) { + return this.solutionContainer.stream().filter(s -> s.getContainerCode().equals(containerCode)).findFirst().orElse(null); + } }