update...

潘钊
1 parent 4693f0a0
Showing 15 changed files with 430 additions and 30 deletions
pom.xml
src/main/java/com/bsth/CXFConfig.java
src/main/java/com/bsth/server_rs/AuthorizeInterceptor_IN.java
src/main/java/com/bsth/server_rs/adapter/DateAdapter.java
src/main/java/com/bsth/server_rs/base_info/car/CarService.java
src/main/java/com/bsth/server_rs/base_info/line/Line.java
src/main/java/com/bsth/server_rs/base_info/line/LineService.java
src/main/java/com/bsth/server_rs/base_info/line/LineServiceImpl.java
src/main/java/com/bsth/server_rs/base_info/line/buffer/LineBufferData.java
src/main/java/com/bsth/server_rs/base_info/line/buffer/LineRefreshThread.java
src/main/java/com/bsth/server_rs/base_info/person/PersonService.java
src/main/java/com/bsth/server_rs/exception/AesException.java
src/main/java/com/bsth/server_rs/exception/AesExceptionMapper.java
src/main/java/com/bsth/server_rs/exception/ExceptionEntity.java
src/main/java/com/bsth/server_rs/gps/GpsRealService.java
@@ -133,6 +133,18 @@
 			<artifactId>axis</artifactId>
 			<version>1.4</version>
 		</dependency>
+
+		<dependency>
+			<groupId>com.fasterxml.jackson.jaxrs</groupId>
+			<artifactId>jackson-jaxrs-json-provider</artifactId>
+			<version>2.8.7</version>
+		</dependency>
+
+		<dependency>
+			<groupId>org.eclipse.jetty</groupId>
+			<artifactId>jetty-util</artifactId>
+			<version>9.3.8.v20160314</version>
+		</dependency>
 	</dependencies>
 	<build>
 package com.bsth;
+import com.bsth.server_rs.AuthorizeInterceptor_IN;
+import com.bsth.server_rs.exception.AesExceptionMapper;
+import com.bsth.server_rs.base_info.line.LineServiceImpl;
 import com.bsth.server_ws.attendance.AttendanceServiceSoap;
 import com.bsth.server_ws.park_station.CompanyServiceSoap;
 import com.bsth.server_ws.waybill.LD_ServiceSoap;
+import com.fasterxml.jackson.jaxrs.json.JacksonJsonProvider;
 import org.apache.cxf.Bus;
 import org.apache.cxf.bus.spring.SpringBus;
+import org.apache.cxf.endpoint.Server;
+import org.apache.cxf.jaxrs.JAXRSServerFactoryBean;
 import org.apache.cxf.jaxws.EndpointImpl;
 import org.apache.cxf.transport.servlet.CXFServlet;
 import org.springframework.boot.web.servlet.ServletRegistrationBean;
@@ -13,6 +19,7 @@ import org.springframework.context.annotation.Configuration;
 import org.springframework.core.Ordered;
 import javax.xml.ws.Endpoint;
+import java.util.Arrays;
 /**
  * Created by panzhao on 2017/3/9.
@@ -25,21 +32,9 @@ public class CXFConfig {
         return new SpringBus();
     }
-    public CompanyServiceSoap companyService(){
-        return new CompanyServiceSoap();
-    }
-
-    public LD_ServiceSoap ldServiceSoap(){
-        return new LD_ServiceSoap();
-    }
-
-    public AttendanceServiceSoap attendanceService(){
-        return new AttendanceServiceSoap();
-    }
-
     @Bean
     public Endpoint companyEndpoint() {
-        EndpointImpl endpoint = new EndpointImpl(springBus(), companyService());
+        EndpointImpl endpoint = new EndpointImpl(springBus(), new CompanyServiceSoap());
         endpoint.publish("/CompanyService");
         //endpoint.getInInterceptors().add(new AuthInterceptor());
         return endpoint;
@@ -47,7 +42,7 @@ public class CXFConfig {
    @Bean
     public Endpoint ldServiceEndpoint() {
-        EndpointImpl endpoint = new EndpointImpl(springBus(), ldServiceSoap());
+        EndpointImpl endpoint = new EndpointImpl(springBus(), new LD_ServiceSoap());
         endpoint.publish("/LD_Service");
         //endpoint.getInInterceptors().add(new AuthInterceptor());
         return endpoint;
@@ -55,12 +50,33 @@ public class CXFConfig {
     @Bean
     public Endpoint attendanceServiceEndpoint() {
-        EndpointImpl endpoint = new EndpointImpl(springBus(), attendanceService());
+        EndpointImpl endpoint = new EndpointImpl(springBus(), new AttendanceServiceSoap());
         endpoint.publish("/attendanceService");
         //endpoint.getInInterceptors().add(new AuthInterceptor());
         return endpoint;
     }
+
+/*    @Bean
+    public Endpoint LineServiceEndpoint(){
+        EndpointImpl endpoint = new EndpointImpl(springBus(), lineService());
+        endpoint.publish("/data_line");
+        //endpoint.getInInterceptors().add(new AuthInterceptor());
+        return endpoint;
+    }*/
+
+    @Bean
+    public Server rsServer() {
+        JAXRSServerFactoryBean endpoint = new JAXRSServerFactoryBean();
+        endpoint.setBus(springBus());
+        endpoint.setAddress("/rest");
+        endpoint.setServiceBeans(Arrays.<Object>asList(new LineServiceImpl()));
+        endpoint.setProviders(Arrays.asList(new JacksonJsonProvider(), new AesExceptionMapper()));
+        //endpoint.setFeatures(Arrays.asList(new Swagger2Feature()));
+        endpoint.getInInterceptors().add(new AuthorizeInterceptor_IN());
+        return endpoint.create();
+    }
+
     @Bean
     public ServletRegistrationBean servletRegistrationBean() {
         ServletRegistrationBean bean = new ServletRegistrationBean(new CXFServlet(), "/webservice/*");
+package com.bsth.server_rs;
+
+import com.bsth.server_rs.exception.AesException;
+import com.bsth.service.UserService;
+import org.apache.commons.lang3.StringEscapeUtils;
+import org.apache.cxf.interceptor.Fault;
+import org.apache.cxf.message.Message;
+import org.apache.cxf.phase.AbstractPhaseInterceptor;
+import org.apache.cxf.phase.Phase;
+import org.eclipse.jetty.util.MultiMap;
+import org.eclipse.jetty.util.UrlEncoded;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.BeansException;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
+import org.springframework.stereotype.Component;
+
+import java.security.MessageDigest;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * rest 接口授权校验（IN 输入拦截）
+ * Created by panzhao on 2017/3/28.
+ */
+@Component
+public class AuthorizeInterceptor_IN extends AbstractPhaseInterceptor<Message> implements ApplicationContextAware {
+
+    private static final String SIGN = "sign";
+    private static final String TIMESTAMP = "timestamp";
+    private static final String NONCE = "nonce";
+    private static final String PASSWORD = "password";
+
+    static UserService userService;
+
+    static Logger logger = LoggerFactory.getLogger(AuthorizeInterceptor_IN.class);
+
+    public AuthorizeInterceptor_IN() {
+        super(Phase.RECEIVE);
+    }
+
+    @Override
+    public void handleMessage(Message message) throws Fault {
+
+        if (message.get(Message.QUERY_STRING) == null) {
+            throw new AesException(AesException.MISS_SIGN);
+        }
+
+        //放行wadl
+        if(message.get(Message.QUERY_STRING).equals("_wadl")
+                && message.get(Message.PATH_INFO).equals("/webservice/rest")){
+            return ;
+        }
+
+        //获取参数，不包括 url 路径参数  ?号之后的
+        String queryString = StringEscapeUtils.unescapeHtml4(message.get(Message.QUERY_STRING).toString());
+        MultiMap<String> params = new MultiMap<>();
+        UrlEncoded.decodeTo(queryString, params, "utf-8");
+        Map<String, String> map = multi2One(params);
+
+        if (!map.containsKey(SIGN)) {
+            throw new AesException(AesException.MISS_SIGN);
+        }
+        if (!map.containsKey(TIMESTAMP)) {
+            throw new AesException(AesException.MISS_TIMESTAMP);
+        }
+        if (!map.containsKey(NONCE)) {
+            throw new AesException(AesException.MISS_NONCE);
+        }
+        if (!map.containsKey(PASSWORD)) {
+            throw new AesException(AesException.MISS_PWD);
+        }
+
+        if (userService.get(map.get(PASSWORD)) == null) {
+            throw new AesException(AesException.INVALID_PWD);
+        }
+
+        String sign = map.get(SIGN);
+        map.remove(SIGN);
+        String sh1 = "";
+        try {
+            sh1 = getSHA1(map);
+        } catch (Exception e) {
+            throw new AesException(AesException.SIGN_CHECK_ERROR);
+        }
+
+        if (!sign.equals(sh1)) {
+            throw new AesException(AesException.SIGN_CHECK_FAIL);
+        }
+    }
+
+    public static Map<String, String> multi2One(MultiMap<String> params) {
+        Map<String, String> map = new HashMap<>();
+        Set<String> ks = params.keySet();
+        for (String k : ks) {
+            map.put(k, params.getString(k));
+        }
+        return map;
+    }
+
+
+    public static String getSHA1(Map<String, String> map) throws Exception {
+
+        try {
+            String[] array = new String[map.size()];
+            map.values().toArray(array);
+            StringBuffer sb = new StringBuffer();
+
+            // 字符串排序
+            Arrays.sort(array);
+            for (int i = 0; i < array.length; i++) {
+                sb.append(array[i]);
+            }
+            String str = sb.toString();
+            // SHA1签名生成
+            MessageDigest md = MessageDigest.getInstance("SHA-1");
+            md.update(str.getBytes());
+            byte[] digest = md.digest();
+
+            StringBuffer hexstr = new StringBuffer();
+            String shaHex = "";
+            for (int i = 0; i < digest.length; i++) {
+                shaHex = Integer.toHexString(digest[i] & 0xFF);
+                if (shaHex.length() < 2) {
+                    hexstr.append(0);
+                }
+                hexstr.append(shaHex);
+            }
+            return hexstr.toString();
+        } catch (Exception e) {
+            logger.error("", e);
+            throw e;
+        }
+    }
+
+    @Override
+    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+        userService = applicationContext.getBean(UserService.class);
+    }
+}
+package com.bsth.server_rs.adapter;
+
+import org.joda.time.format.DateTimeFormat;
+import org.joda.time.format.DateTimeFormatter;
+
+import javax.xml.bind.annotation.adapters.XmlAdapter;
+import java.util.Date;
+
+/**
+ * 时间格式处理 yyyy-MM-dd HH:mm:ss
+ * Created by panzhao on 2017/3/28.
+ */
+public class DateAdapter extends XmlAdapter<String, Date> {
+
+    private static DateTimeFormatter fmtyyyyMMddHHmmss = DateTimeFormat.forPattern("yyyy-MM-dd HH:mm:ss");
+
+    @Override
+    public Date unmarshal(String v) throws Exception {
+        return fmtyyyyMMddHHmmss.parseDateTime(v).toDate();
+    }
+
+    @Override
+    public String marshal(Date v) throws Exception {
+        return fmtyyyyMMddHHmmss.print(v.getTime());
+    }
+}
+package com.bsth.server_rs.base_info.car;
+
+/**
+ * Created by panzhao on 2017/3/28.
+ */
+public class CarService {
+}
 package com.bsth.server_rs.base_info.line;
+import com.bsth.server_rs.adapter.DateAdapter;
+
+import javax.xml.bind.annotation.XmlRootElement;
+import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
 import java.io.Serializable;
 import java.util.Date;
@@ -11,7 +15,7 @@ import java.util.Date;
  * @Date 2016-4-11 16:06:17
  * @Version 公交调度系统BS版 0.1
  */
-
+@XmlRootElement
 public class Line implements Serializable {
     /**
@@ -411,6 +415,7 @@ public class Line implements Serializable {
         this.descriptions = descriptions;
     }
+    @XmlJavaTypeAdapter(DateAdapter.class)
     public Date getCreateDate() {
         return createDate;
     }
-package com.bsth.server_rs.base_info.line;
-
-/**
- * 线路 Rest Service
- * Created by panzhao on 2017/3/27.
- */
-public class LineService {
-
-}
+package com.bsth.server_rs.base_info.line;
+
+import com.bsth.server_rs.base_info.line.buffer.LineBufferData;
+
+import javax.ws.rs.GET;
+import javax.ws.rs.Path;
+import javax.ws.rs.PathParam;
+import javax.ws.rs.Produces;
+import javax.ws.rs.core.MediaType;
+import java.util.List;
+
+/**
+ * 线路 Rest Service
+ * Created by panzhao on 2017/3/27.
+ */
+@Path("/line")
+@Produces({MediaType.APPLICATION_JSON, MediaType.APPLICATION_XML})
+public class LineServiceImpl {
+
+    @GET
+    @Path("/company/{companyId}")
+    public List<Line> findAll(@PathParam("companyId") String companyId) {
+        return companyId.equals("-9999") ? LineBufferData.findAll() : LineBufferData.findByCompany(companyId);
+    }
+
+    @GET
+    @Path("/{id}")
+    public Line findOne(@PathParam("id") Integer id) {
+        return LineBufferData.findOne(id);
+    }
+}
 package com.bsth.server_rs.base_info.line.buffer;
+import com.bsth.Application;
 import com.bsth.server_rs.base_info.line.Line;
+import com.google.common.collect.ArrayListMultimap;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.CommandLineRunner;
 import org.springframework.core.annotation.Order;
@@ -9,13 +11,14 @@ import org.springframework.stereotype.Component;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 /**
  * 线路数据缓存（自更新）
  * Created by panzhao on 2017/3/27.
  */
 @Component
-@Order(10)
+@Order(5)
 public class LineBufferData implements CommandLineRunner {
     @Autowired
@@ -23,6 +26,7 @@ public class LineBufferData implements CommandLineRunner {
     private static List<Line> data;
     private static Map<Integer, Line> idMap;
+    private static ArrayListMultimap<String, Line> companyListMap;
     public static List<Line> findAll(){
         return data;
@@ -32,17 +36,26 @@ public class LineBufferData implements CommandLineRunner {
         return idMap.get(id);
     }
+    public static List<Line> findByCompany(String company){
+        return companyListMap.get(company);
+    }
+
     public static void replaceAll(List<Line> newData){
         data = newData;
         Map<Integer, Line> idMapCopy = new HashMap<>();
+        ArrayListMultimap<String, Line> listMap = ArrayListMultimap.create();
+
         for(Line line : data){
             idMapCopy.put(line.getId(), line);
+            listMap.put(line.getCompany(), line);
         }
         idMap = idMapCopy;
+
+        companyListMap = listMap;
     }
     @Override
     public void run(String... strings) throws Exception {
-        //Application.mainServices.scheduleWithFixedDelay(lineRefreshThread, 10, 60 * 60, TimeUnit.SECONDS);
+        Application.mainServices.scheduleWithFixedDelay(lineRefreshThread, 10, 60 * 60, TimeUnit.SECONDS);
     }
 }
 package com.bsth.server_rs.base_info.line.buffer;
 import com.bsth.server_rs.base_info.line.Line;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.jdbc.core.BeanPropertyRowMapper;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.stereotype.Component;
@@ -16,11 +19,17 @@ public class LineRefreshThread extends Thread{
     @Autowired
     JdbcTemplate jdbcTemplate;
+    Logger logger = LoggerFactory.getLogger(this.getClass());
+
     @Override
     public void run() {
-        List<Line> list = jdbcTemplate.queryForList("select * from bsth_c_line", Line.class);
-        if(list != null && list.size() > 0)
-            LineBufferData.replaceAll(list);
+        try {
+            List<Line> list = jdbcTemplate.query("select * from bsth_c_line",BeanPropertyRowMapper.newInstance(Line.class));
+            if(list != null && list.size() > 0)
+                LineBufferData.replaceAll(list);
+        }catch (Exception e){
+            logger.error("", e);
+        }
     }
 }
+package com.bsth.server_rs.base_info.person;
+
+/**
+ * Created by panzhao on 2017/3/28.
+ */
+public class PersonService {
+}
+package com.bsth.server_rs.exception;
+
+/**
+ * Created by panzhao on 2017/3/28.
+ */
+public class AesException extends RuntimeException {
+
+    public final static int OK = 0;
+    public final static int MISS_SIGN = -30001;
+    public final static int MISS_TIMESTAMP = -30002;
+    public final static int MISS_NONCE = -30003;
+    public final static int MISS_PWD = -30004;
+    public final static int SIGN_CHECK_ERROR = -40001;
+    public final static int SIGN_CHECK_FAIL = -40002;
+    public final static int INVALID_PWD = -40003;
+
+    private int code;
+
+    private static String getMessage(int code) {
+        switch (code) {
+            case MISS_SIGN:
+                return "sign参数丢失";
+            case MISS_TIMESTAMP:
+                return "timestamp参数丢失";
+            case MISS_NONCE:
+                return "nonce参数丢失";
+            case MISS_PWD:
+                return "密码参数丢失";
+            case INVALID_PWD:
+                return "无效的密码";
+            case SIGN_CHECK_ERROR:
+                return "签名校验时出现异常";
+            case SIGN_CHECK_FAIL:
+                return "签名校验失败";
+            default:
+                return null;
+        }
+    }
+
+    public int getCode() {
+        return code;
+    }
+
+    public AesException(int code) {
+        super(getMessage(code));
+        this.code = code;
+    }
+}
+package com.bsth.server_rs.exception;
+
+import org.apache.cxf.jaxrs.ext.MessageContext;
+
+import javax.ws.rs.core.Context;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.ext.ExceptionMapper;
+import java.util.Locale;
+
+/**
+ * cxf rest 异常处理
+ * Created by panzhao on 2017/3/28.
+ */
+public class AesExceptionMapper implements ExceptionMapper<AesException> {
+
+    @Context
+    private MessageContext mc;
+
+    @Override
+    public Response toResponse(AesException e) {
+        StackTraceElement[] trace = new StackTraceElement[1];
+        trace[0] = e.getStackTrace()[0];
+        e.setStackTrace(trace);
+
+        Response.ResponseBuilder rb = Response.status(Response.Status.INTERNAL_SERVER_ERROR);
+        ExceptionEntity entity = new ExceptionEntity(e.getCode() + "", e.getMessage());
+        rb.entity(entity);
+
+        String accept = mc.getHttpServletRequest().getHeader("Accept");
+        rb.type(accept);
+        rb.language(Locale.SIMPLIFIED_CHINESE);
+        return rb.build();
+    }
+}
+package com.bsth.server_rs.exception;
+
+import javax.xml.bind.annotation.XmlRootElement;
+import java.io.Serializable;
+
+/**
+ * Created by panzhao on 2017/3/28.
+ */
+@XmlRootElement
+public class ExceptionEntity implements Serializable {
+
+    /*错误码，默认为0*/
+    private int errCode = 0;
+
+    /*错误子码，自定义该值*/
+    private String errSubCode;
+
+    private String message;
+
+    public ExceptionEntity(String errSubCode, String message) {
+        super();
+        this.errSubCode = errSubCode;
+        this.message = message;
+    }
+
+    public ExceptionEntity(){}
+
+    public int getErrCode() {
+        return errCode;
+    }
+
+    public void setErrCode(int errCode) {
+        this.errCode = errCode;
+    }
+
+    public String getErrSubCode() {
+        return errSubCode;
+    }
+
+    public void setErrSubCode(String errSubCode) {
+        this.errSubCode = errSubCode;
+    }
+
+    public String getMessage() {
+        return message;
+    }
+
+    public void setMessage(String message) {
+        this.message = message;
+    }
+}
+package com.bsth.server_rs.gps;
+
+/**
+ * Created by panzhao on 2017/3/28.
+ */
+public class GpsRealService {
+}