package com.tievd.jyz.cache;
|
|
import com.baomidou.mybatisplus.extension.activerecord.Model;
|
import com.baomidou.mybatisplus.extension.service.IService;
|
import com.tievd.jyz.constants.SystemConstant;
|
import com.tievd.jyz.entity.*;
|
import com.tievd.jyz.service.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.PostConstruct;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* Author: wqy
|
* Date: 2023/3/9 9:37
|
*/
|
|
@Component
|
public class AlgorithmCache {
|
|
@Autowired
|
private ISysAlgorithmItemService sysAlgorithmItemService;
|
|
public static final String OIL_ALG_CODE = "C10001,C10003";
|
|
private static Map<String, SysAlgorithmItem> algorithmMap = new HashMap<>();
|
/** 所属阶段 1卸前 2卸中 3卸后 */
|
private static Map<Byte, List<SysAlgorithmItem>> phraseAlgorithmMap = new HashMap<>();
|
|
public static Map<String, SysAlgorithmItem> algorithmMap() {
|
return algorithmMap;
|
}
|
|
/**
|
* 获取卸油算法map
|
* @return
|
*/
|
public static Map<String, SysAlgorithmItem> oiloutAlgMap() {
|
Map<String, SysAlgorithmItem> oiloutMap = new HashMap<>();
|
for (Map.Entry<String, SysAlgorithmItem> entry : algorithmMap.entrySet()) {
|
if (entry.getValue().getRegion() == SystemConstant.INSTALL_ADDRESS_OIL_OUT.byteValue()) {
|
oiloutMap.put(entry.getKey(), entry.getValue());
|
}
|
}
|
return oiloutMap;
|
}
|
|
/**
|
*获取阶段算法
|
* @param phrase
|
* @return
|
*/
|
public static List<SysAlgorithmItem> getPhraseAlgorithmMap(Byte phrase) {
|
return phraseAlgorithmMap.get(phrase);
|
}
|
|
/**
|
* 初始化
|
*/
|
@PostConstruct
|
private void init(){
|
algorithmMap.clear();
|
phraseAlgorithmMap.clear();
|
List<SysAlgorithmItem> algorithmItems = sysAlgorithmItemService.list();
|
for(SysAlgorithmItem algorithmItem:algorithmItems){
|
algorithmMap.put(algorithmItem.getAlgorithmCode(),algorithmItem);
|
List<SysAlgorithmItem> sysAlgorithmItems = phraseAlgorithmMap.get(algorithmItem.getPhrase());
|
if(sysAlgorithmItems == null){
|
sysAlgorithmItems = new ArrayList<>();
|
phraseAlgorithmMap.put(algorithmItem.getPhrase(),sysAlgorithmItems);
|
}
|
sysAlgorithmItems.add(algorithmItem);
|
}
|
}
|
|
/**
|
* 获取算法行为
|
* @param algCode
|
* @return
|
*/
|
public static SysAlgorithmItem getAlgorithmMap(String algCode) {
|
return algorithmMap.get(algCode);
|
}
|
|
|
public static Model getEntity(String algCode) {
|
int type = algorithmMap.get(algCode).getAlgorithmType();
|
switch(type) {
|
case 1 : return new OilEvent();
|
case 2 : return new OiloutEvent();
|
case 3 : return new PatrolEvent();
|
case 4 : return new SpotcheckEvent();
|
default: return new OilEvent();
|
}
|
}
|
|
public static Class<? extends IService> getService(String algCode) {
|
int type = algorithmMap.get(algCode).getAlgorithmType();
|
switch(type) {
|
case 1 : return IOilEventService.class;
|
case 2 : return IOiloutEventService.class;
|
case 3 : return IPatrolEventService.class;
|
case 4 : return ISpotcheckEventService.class;
|
default: return IOilEventService.class;
|
}
|
}
|
|
}
|
