添加测试代码

server/src/main/java/com/pica/cloud/account/account/server/controller/AutoCodeController.java

@@ -33,6 +33,7 @@ import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Enumeration;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -61,8 +62,61 @@ public class AutoCodeController extends AccountBaseController {
 
 
     @GetMapping("/ip")
-    public PicaResponse getIp(HttpServletRequest request) {
-        return PicaResponse.toResponse(IPUtil.getIpAdrress(request));
+    public PicaResponse getIp(javax.servlet.http.HttpServletRequest request) {
+        //return PicaResponse.toResponse(IPUtil.getIpAdrress(request));
+
+        String ip = null;
+
+        //X-Forwarded-For：Squid 服务代理
+        String ipAddresses = request.getHeader("X-Forwarded-For");
+        Enumeration<String> headerNames = request.getHeaderNames();
+        while (headerNames.hasMoreElements()) {
+            //打印所有头信息
+            String s = headerNames.nextElement();
+            String header = request.getHeader(s);
+//            System.out.println(s+"::::"+header);
+        }
+
+//        System.out.println("headerNames:"+ JSON.toJSONString(headerNames));
+//        System.out.println("RemoteHost:"+request.getRemoteHost());
+//        System.out.println("RemoteAddr:"+request.getRemoteAddr());
+
+        String unknown = "unknown";
+        if (ipAddresses == null || ipAddresses.length() == 0 || unknown.equalsIgnoreCase(ipAddresses)) {
+            //Proxy-Client-IP：apache 服务代理
+            ipAddresses = request.getHeader("Proxy-Client-IP");
+        }
+
+        if (ipAddresses == null || ipAddresses.length() == 0 || unknown.equalsIgnoreCase(ipAddresses)) {
+            //WL-Proxy-Client-IP：weblogic 服务代理
+            ipAddresses = request.getHeader("WL-Proxy-Client-IP");
+        }
+
+        if (ipAddresses == null || ipAddresses.length() == 0 || unknown.equalsIgnoreCase(ipAddresses)) {
+            //HTTP_CLIENT_IP：有些代理服务器
+            ipAddresses = request.getHeader("HTTP_CLIENT_IP");
+        }
+
+        if (ipAddresses == null || ipAddresses.length() == 0 || unknown.equalsIgnoreCase(ipAddresses)) {
+            //X-Real-IP：nginx服务代理
+            ipAddresses = request.getHeader("X-Real-IP");
+        }
+
+        //有些网络通过多层代理，那么获取到的ip就会有多个，一般都是通过逗号（,）分割开来，并且第一个ip为客户端的真实IP
+        if (ipAddresses != null && ipAddresses.length() != 0) {
+            ip = ipAddresses.split(",")[0];
+        }
+
+        //还是不能获取到，最后再通过request.getRemoteAddr();获取
+        if (ip == null || ip.length() == 0 || unknown.equalsIgnoreCase(ipAddresses)) {
+            ip = request.getRemoteAddr();
+        }
+
+        if("0:0:0:0:0:0:0:1".equalsIgnoreCase(ip)){
+            ip = "127.0.0.1";
+        }
+
+        return PicaResponse.toResponse(ip);
     }
 
     /*@ApiOperation("数据加密-方便测试")

server/test/main/java/com/pica/cloud/account/account/server/mapper/AutoCodeControllerTest.java

+// Copyright 2016-2101 Pica.
+package com.pica.cloud.account.account.server.mapper;
+
+import com.alibaba.fastjson.JSON;
+import com.pica.cloud.account.account.server.controller.AutoCodeController;
+import com.pica.cloud.account.account.server.entity.EncryptEntity;
+import com.pica.cloud.account.account.server.req.BaseRequest;
+import com.pica.cloud.account.account.server.util.AESUtil;
+import com.pica.cloud.account.account.server.util.RSAUtil;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * @ClassName AutoCodeControllerTest
+ * @Description
+ * @Author Chongwen.jiang
+ * @Date 2020/4/3 14:31
+ * @ModifyDate 2020/4/3 14:31
+ * @Version 1.0
+ */
+@RunWith(SpringRunner.class)
+@SpringBootTest
+public class AutoCodeControllerTest {
+
+    /*@Autowired
+    private AutoCodeController autoCodeController;
+    @Test
+    public void getH5AuthCodeTest() throws Exception {
+        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.currentRequestAttributes();
+        HttpServletRequest request = servletRequestAttributes.getRequest();
+
+        String publicKey = "1234567890123456";
+        BaseRequest req= new BaseRequest();
+        req.setFlag(4);
+        req.setMobile("13517477655");
+        req.setCapCode(String.valueOf(System.currentTimeMillis()));
+        req.setxPos(126);
+        req.setDevice_token("000000");
+        req.setSourceType(4);
+        String content = AESUtil.aesEncrypt(JSON.toJSONString(req), publicKey);
+        String encryptKey = RSAUtil.encrypt(publicKey);
+        EncryptEntity entity = new EncryptEntity();
+        entity.setKey(encryptKey);
+        entity.setContent(content);
+
+        autoCodeController.getH5AuthCode(entity, request);
+
+    }*/
+
+
+    
+}