package com.iailab.module.model.mdk.factory; import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity; import com.iailab.module.model.mcs.pre.entity.MmItemTypeEntity; import com.iailab.module.model.mcs.pre.service.MmItemOutputService; import com.iailab.module.model.mcs.pre.service.MmItemTypeService; import com.iailab.module.model.mcs.pre.service.MmPredictItemService; import com.iailab.module.model.mdk.vo.ItemVO; import com.iailab.module.model.mdk.vo.MergeItemVO; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import java.util.HashMap; import java.util.List; import java.util.Map; /** * 创建预测项对象的工厂 */ @Component public class ItemEntityFactory { private HashMap<String, ItemVO> ItemVOHashMap = new HashMap<>(); private Map<String, MmItemTypeEntity> itemTypeMap = new HashMap<>(); private Map<String, MergeItemVO> mergeItemMap = new HashMap<>(); private Map<String, ItemVO> itemNoEntityMap = new HashMap<>(); private Map<String, List<MmItemOutputEntity>> itemOutPutItemMap = new HashMap<>(); @Autowired private MmPredictItemService mmPredictItemService; @Autowired private MmItemOutputService mmItemOutputService; @Autowired private MmItemTypeService mmItemTypeService; /** * 1.æ ¹æ®ITEMID,获å–模型对应的MERGEITEM实体 * * @param itemId * @return */ public MergeItemVO getMergeItem(String itemId) { MergeItemVO mergeItemVO = mmPredictItemService.getMergeItemByItemId(itemId); if (!mergeItemMap.containsKey(itemId)) { if (mergeItemVO != null) { mergeItemMap.put(itemId, mergeItemVO); } else { return null; } } return mergeItemVO; } /** * 2.æ ¹æ®ITEMNO,获å–模型对应的ITEM实体 * * @param itemNo * @return */ public ItemVO getItemByItemNo(String itemNo) { ItemVO ItemVO = mmPredictItemService.getItemByItemNo(itemNo); if (!itemNoEntityMap.containsKey(itemNo)) { if (ItemVO != null) { itemNoEntityMap.put(itemNo, ItemVO); } else { return null; } } return ItemVO; } /** * 3.æ ¹æ®ITEMID,获å–模型对应的ITEM实体 * * @param itemId * @return */ public ItemVO getItemById(String itemId) { ItemVO ItemVO = mmPredictItemService.getItemByIdFromCache(itemId); if (!ItemVOHashMap.containsKey(itemId)) { if (ItemVO != null) { ItemVOHashMap.put(itemId, ItemVO); } else { return null; } } return ItemVO; } /** * 4. æ ¹æ®ITEMID,获å–模型对应的ITEMOUTPUT实体 * * @param itemId * @return */ public List<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity> getOutPutByItemId(String itemId) { List<com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity> list = mmItemOutputService.getByItemid(itemId); if (!itemOutPutItemMap.containsKey(itemId)) { if (list != null) { itemOutPutItemMap.put(itemId, list); } else { return null; } } return list; } }