|
|
@@ -2,9 +2,11 @@ package com.kym.service.utils;
|
|
|
|
|
|
import com.baomidou.dynamic.datasource.annotation.DS;
|
|
|
import com.kym.entity.admin.AdminUserStation;
|
|
|
+import com.kym.entity.admin.EquipmentInfo;
|
|
|
import com.kym.entity.admin.EquipmentRelation;
|
|
|
import com.kym.entity.admin.Station;
|
|
|
import com.kym.service.admin.AdminUserStationService;
|
|
|
+import com.kym.service.admin.EquipmentInfoService;
|
|
|
import com.kym.service.admin.EquipmentRelationService;
|
|
|
import com.kym.service.admin.StationService;
|
|
|
import lombok.Data;
|
|
|
@@ -27,19 +29,24 @@ public class KymCache {
|
|
|
private static ConcurrentHashMap<String, String> CONNECTOR_STATION_MAPPING = new ConcurrentHashMap<>();
|
|
|
private static ConcurrentHashMap<String, String> STATION_MAPPING = new ConcurrentHashMap<>();
|
|
|
private static ConcurrentHashMap<Long, String> ADMIN_USER_STATION = new ConcurrentHashMap<>();
|
|
|
+ private static ConcurrentHashMap<String, String> EQUIPMENT_ID_PARKING_NO_MAPPING = new ConcurrentHashMap<>();
|
|
|
private final EquipmentRelationService relationService;
|
|
|
private final StationService stationService;
|
|
|
private final AdminUserStationService adminUserStationService;
|
|
|
+ private final EquipmentInfoService equipmentInfoService;
|
|
|
|
|
|
- public KymCache(EquipmentRelationService relationService, StationService stationService, AdminUserStationService adminUserStationService) {
|
|
|
+ public KymCache(EquipmentRelationService relationService, StationService stationService, AdminUserStationService adminUserStationService, EquipmentInfoService equipmentInfoService) {
|
|
|
this.relationService = relationService;
|
|
|
this.stationService = stationService;
|
|
|
this.adminUserStationService = adminUserStationService;
|
|
|
+ this.equipmentInfoService = equipmentInfoService;
|
|
|
+
|
|
|
+ SHORT_ID_MAPPING.putAll( relationService.list().stream().collect(Collectors.toMap(EquipmentRelation::getEquipmentId, EquipmentRelation::getShortId)));
|
|
|
+ CONNECTOR_STATION_MAPPING.putAll( relationService.list().stream().collect(Collectors.toMap(EquipmentRelation::getConnectorId, EquipmentRelation::getStationId)));
|
|
|
|
|
|
- SHORT_ID_MAPPING.putAll(relationService.list().stream().collect(Collectors.toMap(EquipmentRelation::getEquipmentId, EquipmentRelation::getShortId)));
|
|
|
- CONNECTOR_STATION_MAPPING.putAll(relationService.list().stream().collect(Collectors.toMap(EquipmentRelation::getConnectorId, EquipmentRelation::getStationId)));
|
|
|
STATION_MAPPING.putAll(stationService.list().stream().collect(Collectors.toMap(Station::getStationId, Station::getStationName)));
|
|
|
ADMIN_USER_STATION.putAll(adminUserStationService.list().stream().collect(Collectors.toMap(AdminUserStation::getAdminUserId, AdminUserStation::getStationId)));
|
|
|
+ EQUIPMENT_ID_PARKING_NO_MAPPING.putAll(equipmentInfoService.list().stream().collect(Collectors.toMap(EquipmentInfo::getEquipmentId, EquipmentInfo::getParkingNo)));
|
|
|
}
|
|
|
|
|
|
@DS("db-admin")
|
|
|
@@ -115,4 +122,8 @@ public class KymCache {
|
|
|
}
|
|
|
return res;
|
|
|
}
|
|
|
+
|
|
|
+ public String getParkNoByEquipmentId(String equipmentId){
|
|
|
+ return EQUIPMENT_ID_PARKING_NO_MAPPING.get(equipmentId);
|
|
|
+ }
|
|
|
}
|