1.数知梦接口需要扩展进/external

王通
1 parent 7dd529ff
Showing 5 changed files with 164 additions and 43 deletions
src/main/java/com/bsth/CXFConfig.java
src/main/java/com/bsth/server_rs/AuthorizeInterceptor_IN.java
src/main/java/com/bsth/server_rs/dks/BxRestService.java
src/main/java/com/bsth/server_rs/gps/entity/HistoryArrivalEntity.java
src/main/resources/application.properties
@@ -98,6 +98,8 @@ public class CXFConfig {
     }
     @Autowired
+    private AuthorizeInterceptor_IN authorizeInterceptorIn;
+    @Autowired
     ScheduleRealService scheduleRealService;
     @Autowired
     StationRestService stationRestService;
@@ -166,7 +168,7 @@ public class CXFConfig {
                 bxRestService));
         endpoint.setProviders(Arrays.asList(new JacksonJsonProvider(), new AesExceptionMapper()));
         //endpoint.setFeatures(Arrays.asList(new Swagger2Feature()));
-        endpoint.getInInterceptors().add(new AuthorizeInterceptor_IN());
+        endpoint.getInInterceptors().add(authorizeInterceptorIn);
         return endpoint.create();
     }
@@ -5,6 +5,7 @@ import com.bsth.entity.Resource;
 import com.bsth.server_rs.exception.AesException;
 import com.bsth.service.SystemParamService;
 import com.bsth.service.UserService;
+import com.bsth.util.ThreadLocalUtils;
 import org.apache.commons.lang3.StringEscapeUtils;
 import org.apache.cxf.interceptor.Fault;
 import org.apache.cxf.message.Message;
@@ -15,17 +16,20 @@ import org.eclipse.jetty.util.UrlEncoded;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.ApplicationContextAware;
 import org.springframework.stereotype.Component;
 import org.springframework.util.AntPathMatcher;
 import org.springframework.util.PathMatcher;
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
 import java.security.MessageDigest;
-import java.util.Arrays;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
+import java.util.concurrent.ConcurrentHashMap;
 /**
  * rest 接口授权校验（IN 输入拦截）
@@ -49,6 +53,15 @@ public class AuthorizeInterceptor_IN extends AbstractPhaseInterceptor&lt;Message&gt; i
     private static PathMatcher matcher = new AntPathMatcher();
+    @Value("${path.limits}")
+    private String pathLimits;
+
+    private Map<String, Set<String>> pwd2lines = new ConcurrentHashMap<>();
+
+    private Map<String, Set<String>> pwd2device = new ConcurrentHashMap<>();
+
+    private ThreadLocal<String> passwords = new ThreadLocal<>();
+
     public AuthorizeInterceptor_IN() {
         super(Phase.RECEIVE);
     }
@@ -121,12 +134,15 @@ public class AuthorizeInterceptor_IN extends AbstractPhaseInterceptor&lt;Message&gt; i
             throw new AesException(AesException.SIGN_CHECK_FAIL);
         }
+        ThreadLocalUtils.setPassword(map.get(PASSWORD));
         validate(map, message);
     }
-    private static void validate(Map<String, String> map, Message message) {
-        PasswordUser user = userService.get(map.get(PASSWORD));
+    private void validate(Map<String, String> map, Message message) {
+        String password = map.get(PASSWORD);
+        PasswordUser user = userService.get(password);
         if (user.getResources().size() > 0) {
+            loadLimit(password);
             boolean isMatch = false;
             String uri = (String) message.get(Message.REQUEST_URI);
             for (Resource resource : user.getResources()) {
@@ -141,6 +157,49 @@ public class AuthorizeInterceptor_IN extends AbstractPhaseInterceptor&lt;Message&gt; i
         }
     }
+    public void loadLimit(String password) {
+        if (!pwd2lines.containsKey(password)) {
+            Set<String> set = new HashSet<>();
+            try {
+                BufferedReader reader = new BufferedReader(new FileReader(pathLimits + String.format("lines-%s.txt", password)));
+                String line = null;
+                while ((line = reader.readLine()) != null) {
+                    set.add(line);
+                }
+            } catch (IOException e) {
+                logger.error("加载受限线路信息异常", e);
+            } finally {
+                pwd2lines.put(password, set);
+            }
+        }
+        if (!pwd2device.containsKey(password)) {
+            Set<String> set = new HashSet<>();
+            try {
+                BufferedReader reader = new BufferedReader(new FileReader(pathLimits + String.format("devices-%s.txt", password)));
+                String line = null;
+                while ((line = reader.readLine()) != null) {
+                    set.add(line);
+                }
+            } catch (IOException e) {
+                logger.error("加载受限设备信息异常", e);
+            } finally {
+                pwd2device.put(password, set);
+            }
+        }
+    }
+
+    public Set<String> getLimitLines(String password) {
+        return pwd2lines.get(password);
+    }
+
+    public Set<String> getLimitDevices(String password) {
+        return pwd2device.get(password);
+    }
+
+    public ThreadLocal<String> getPasswords() {
+        return passwords;
+    }
+
     public static Map<String, String> multi2One(MultiMap<String> params) {
         Map<String, String> map = new HashMap<>();
         Set<String> ks = params.keySet();
 package com.bsth.server_rs.dks;
-import com.bsth.common.SystemParamKeys;
+import com.bsth.server_rs.AuthorizeInterceptor_IN;
 import com.bsth.server_rs.base_info.car.Car;
 import com.bsth.server_rs.base_info.car.buffer.CarBufferData;
 import com.bsth.server_rs.base_info.carpark.Carpark;
@@ -8,26 +8,24 @@ import com.bsth.server_rs.base_info.carpark.buffer.CarparkBufferData;
 import com.bsth.server_rs.base_info.line.Line;
 import com.bsth.server_rs.base_info.line.buffer.LineBufferData;
 import com.bsth.server_rs.base_info.section.buffer.LD_SectionBufferData;
-import com.bsth.server_rs.base_info.section.entity.LD_Section;
 import com.bsth.server_rs.base_info.section.entity.LD_SectionRoute;
 import com.bsth.server_rs.base_info.station.buffer.StationBufferData;
 import com.bsth.server_rs.base_info.station.entity.StationRotue;
 import com.bsth.server_rs.gps.buffer.BasicDataBuffer;
 import com.bsth.server_rs.gps.buffer.GpsRealDataBuffer;
+import com.bsth.server_rs.gps.dao.HistoryGpsDao;
 import com.bsth.server_rs.gps.entity.GpsEntity;
-import com.bsth.server_rs.gps.entity.LineInfo;
-import com.bsth.service.SystemParamService;
+import com.bsth.server_rs.gps.entity.HistoryArrivalEntity;
+import com.bsth.util.ThreadLocalUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
-import javax.ws.rs.GET;
-import javax.ws.rs.Path;
-import javax.ws.rs.Produces;
+import javax.ws.rs.*;
 import javax.ws.rs.core.MediaType;
 import java.util.*;
-import java.util.concurrent.ConcurrentHashMap;
 /**
  * @author Hill
@@ -41,22 +39,26 @@ public class BxRestService {
     private final static Logger log = LoggerFactory.getLogger(BxRestService.class);
     @Autowired
-    private SystemParamService systemParamService;
+    private LD_SectionBufferData ldSectionBufferData;
     @Autowired
-    private LD_SectionBufferData ldSectionBufferData;
+    private HistoryGpsDao historyGpsDao;
+
+    @Autowired
+    private AuthorizeInterceptor_IN authorizeInterceptorIn;
     @GET
     @Path("/line")
     public List<Map<String, Object>> findAllLine() {
         List<Map<String, Object>> result = new ArrayList<>();
-        String limitLines = systemParamService.getValue(SystemParamKeys.LIMIT_LINES);
-        if (limitLines == null) {
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = authorizeInterceptorIn.getLimitLines(password);
+        if (limitLines == null || limitLines.isEmpty()) {
             return result;
         }
         for (Line line : LineBufferData.findAll()) {
             String lineCode = line.getLineCode();
-            if (limitLines.indexOf(String.format(",%s,", lineCode)) > -1) {
+            if (limitLines.contains(lineCode) || limitLines.contains("ALL")) {
                 Map<String, Object> lineMap = new HashMap<>();
                 Collection<LD_SectionRoute> sectionRoutes = ldSectionBufferData.findByLineCode(lineCode).get(String.format("%s_0", lineCode));
                 StringBuilder sb = new StringBuilder();
@@ -68,6 +70,7 @@ public class BxRestService {
                 lineMap.put("branch", BasicDataBuffer.getBusinessByCode(String.format("%s_%s", line.getCompany(), line.getBrancheCompany())));
                 lineMap.put("lineName", line.getName());
                 lineMap.put("lineCode", line.getShanghaiLinecode());
+                lineMap.put("lineCodeExternal", lineCode);
                 lineMap.put("startStation", line.getStartStationName());
                 lineMap.put("endStation", line.getEndStationName());
                 lineMap.put("lineLength", line.getTotalMileage());
@@ -87,16 +90,23 @@ public class BxRestService {
     @Path("/station")
     public List<Map<String, Object>> findAllStation() {
         List<Map<String, Object>> result = new ArrayList<>();
-        String limitLines = systemParamService.getValue(SystemParamKeys.LIMIT_LINES);
-        if (limitLines == null) {
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = new HashSet<>(authorizeInterceptorIn.getLimitLines(password));
+        if (limitLines == null || limitLines.isEmpty()) {
             return result;
         }
-        for (String lineCode : limitLines.split(",")) {
-            if (!"".equals(lineCode)) {
+        if (limitLines.contains("ALL")) {
+            for (Line line : LineBufferData.findAll()) {
+                limitLines.add(line.getLineCode());
+            }
+        }
+        for (String lineCode : limitLines) {
+            if (StringUtils.isNotEmpty(lineCode)) {
                 Map<String, Object> lineMap = new HashMap<>();
                 Line line = LineBufferData.findOne(lineCode);
                 lineMap.put("lineName", line.getName());
                 lineMap.put("lineCode", line.getShanghaiLinecode());
+                lineMap.put("lineCodeExternal", lineCode);
                 int count = 1;
                 List<Map<String, Object>> upStations = new ArrayList<>(), downStations = new ArrayList<>();
@@ -105,7 +115,7 @@ public class BxRestService {
                     stopMap.put("levelNo", count);
                     stopMap.put("levelName", route.getStationName());
                     stopMap.put("levelLonlat", String.format("%f %f", route.getStation().getLon(), route.getStation().getLat()));
-                    stopMap.put("stationCode", null);
+                    stopMap.put("stationCode", route.getStationCode());
                     upStations.add(stopMap);
                     count++;
                 }
@@ -115,7 +125,7 @@ public class BxRestService {
                     stopMap.put("levelNo", count);
                     stopMap.put("levelName", route.getStationName());
                     stopMap.put("levelLonlat", String.format("%f %f", route.getStation().getLon(), route.getStation().getLat()));
-                    stopMap.put("stationCode", null);
+                    stopMap.put("stationCode", route.getStationCode());
                     downStations.add(stopMap);
                     count++;
                 }
@@ -132,22 +142,19 @@ public class BxRestService {
     @GET
     @Path("/vehicle")
     public List<Map<String, Object>> findAllVehicle() {
-        StringBuilder sb = new StringBuilder();
+        List<Map<String, Object>> result = new ArrayList<>();
         Map<String, String> device2line = GpsRealDataBuffer.getDevice2Line();
-        String limitLines = systemParamService.getValue(SystemParamKeys.LIMIT_LINES);
-        for (Map.Entry<String, String> entry : device2line.entrySet()) {
-            String deviceId = entry.getKey(), lineCode = entry.getValue();
-            if (limitLines.indexOf(String.format(",%s,", lineCode)) > -1) {
-                sb.append(deviceId).append(",");
-            }
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = authorizeInterceptorIn.getLimitLines(password);
+        Set<String> limitDevices = authorizeInterceptorIn.getLimitDevices(password);
+        if (limitLines == null || limitLines.isEmpty()) {
+            return result;
         }
-        String limitDevices = sb.toString();
-        List<Map<String, Object>> result = new ArrayList<>();
-        if (limitDevices.equals("")) {
+        if (limitDevices == null || limitDevices.isEmpty()) {
             return result;
         }
         for (Car car : CarBufferData.findAll()) {
-            if (limitDevices.indexOf(String.format("%s,", car.getEquipmentCode())) > -1) {
+            if (limitDevices.contains(car.getEquipmentCode()) || limitDevices.contains("ALL")) {
                 Map<String, Object> carMap = new HashMap<>();
                 Line line = LineBufferData.findOne(device2line.get(car.getEquipmentCode()));
                 if (line == null) {
@@ -155,6 +162,7 @@ public class BxRestService {
                 }
                 carMap.put("lineName", line.getName());
                 carMap.put("lineCode", line.getShanghaiLinecode());
+                carMap.put("lineCodeExternal", line.getLineCode());
                 carMap.put("vehicleId", car.getCarPlate());
                 carMap.put("vehicleNo", car.getNbbm());
                 carMap.put("vehicleType", car.getVehicleType());
@@ -171,14 +179,15 @@ public class BxRestService {
     @Path("/parking")
     public List<Map<String, Object>> findAllParking() {
         List<Map<String, Object>> result = new ArrayList<>();
-        String limitLines = systemParamService.getValue(SystemParamKeys.LIMIT_LINES);
-        if (limitLines == null) {
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = authorizeInterceptorIn.getLimitLines(password);
+        if (limitLines == null || limitLines.isEmpty()) {
             return result;
         }
         StringBuilder sb = new StringBuilder();
         for (Line line : LineBufferData.findAll()) {
             String lineCode = line.getLineCode();
-            if (limitLines.indexOf(String.format(",%s,", lineCode)) > -1) {
+            if (limitLines.contains(lineCode) || limitLines.contains("ALL")) {
                 sb.append(line.getCarPark()).append(",");
             }
         }
@@ -205,13 +214,15 @@ public class BxRestService {
     @Path("/gps/all")
     public List<Map<String, Object>> findAllGps() {
         List<Map<String, Object>> result = new ArrayList<>();
-        String limitLines = systemParamService.getValue(SystemParamKeys.LIMIT_LINES);
-        if (limitLines == null) {
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = authorizeInterceptorIn.getLimitLines(password);
+        Set<String> limitDevices = authorizeInterceptorIn.getLimitDevices(password);
+        if (limitLines == null || limitLines.isEmpty()) {
             return result;
         }
         for (GpsEntity gps : GpsRealDataBuffer.all()) {
             String lineCode = gps.getLineId(), deviceId = gps.getDeviceId();
-            if (limitLines.indexOf(String.format(",%s,", lineCode)) > -1) {
+            if ((limitLines.contains(lineCode) || limitLines.contains("ALL")) && (limitDevices.contains(deviceId) || limitDevices.contains("ALL"))) {
                 Line line = LineBufferData.findOne(lineCode);
                 Car car = CarBufferData.findByDevice(deviceId);
                 if (car == null) {
@@ -220,6 +231,7 @@ public class BxRestService {
                 Map<String, Object> gpsMap = new HashMap<>();
                 gpsMap.put("lineName", line.getName());
                 gpsMap.put("lineCode", line.getShanghaiLinecode());
+                gpsMap.put("lineCodeExternal", line.getLineCode());
                 gpsMap.put("vehicleId", car.getCarPlate());
                 gpsMap.put("gpsTime", gps.getTimestamp());
                 gpsMap.put("gpsLonlat", String.format("%f %f", gps.getLon(), gps.getLat()));
@@ -231,4 +243,39 @@ public class BxRestService {
         return result;
     }
+    @GET
+    @Path("/history-arrival/{line}/{st}/{et}")
+    public List<HistoryArrivalEntity> historyArrival(@PathParam("line") String line, @PathParam("st") String st, @PathParam("et") String et) {
+        List<HistoryArrivalEntity> result = new ArrayList<>();
+        String password = ThreadLocalUtils.getPassword();
+        Set<String> limitLines = authorizeInterceptorIn.getLimitLines(password);
+        Set<String> limitDevices = authorizeInterceptorIn.getLimitDevices(password);
+        if (limitLines == null || limitLines.isEmpty() || !limitLines.contains(line)) {
+            return result;
+        }
+        if (limitDevices == null || limitDevices.isEmpty()) {
+            return result;
+        }
+        List<HistoryArrivalEntity> arrivals = historyGpsDao.queryArrival(line, Long.parseLong(st), Long.parseLong(et));
+        for (HistoryArrivalEntity arrival : arrivals) {
+            String lineCode = String.valueOf(arrival.getLineId()), deviceId = arrival.getDeviceId();
+            Line line1 = LineBufferData.findOne(lineCode);
+            if (line1 == null) {
+                continue;
+            }
+            arrival.setLineName(line1.getName());
+            if ((limitLines.contains(lineCode) || limitLines.contains("ALL")) && (limitDevices.contains(deviceId) || limitDevices.contains("ALL"))) {
+                result.add(arrival);
+            }
+        }
+
+        return result;
+    }
+
+    @GET
+    @Path("/limits/reload")
+    public void reload() {
+        String password = ThreadLocalUtils.getPassword();
+        authorizeInterceptorIn.loadLimit(password);
+    }
 }
@@ -6,6 +6,8 @@ public class HistoryArrivalEntity {
     private int lineId;
+    private String lineName;
+
     private String stopNo;
     private int upDown;
@@ -30,6 +32,14 @@ public class HistoryArrivalEntity {
         this.lineId = lineId;
     }
+    public String getLineName() {
+        return lineName;
+    }
+
+    public void setLineName(String lineName) {
+        this.lineName = lineName;
+    }
+
     public String getStopNo() {
         return stopNo;
     }
@@ -15,3 +15,6 @@ server.compression.mime-types=application/json,application/xml,text/html,text/xm
 #redis
 cache.days=15
+
+#limits
+path.limits=limits/