适配器模式应用~获取IP地址时想起了适配器

获取IP地址信息时，一般我们需要一个HttpServletRequest对象，然后从请求头里获取x-forwarded-for的值，而当我们使用dubbo+netty开发rest接口时，如果希望获取IP地址，HttpServletRequest是用不了的，你需要使用netty中的NettyRequestFacade对象，这时，你之前的获取IP地址的方法需要扩展，加一个只有入参不一样，代码逻辑完成一样的丑陋的，重复的代码；这无疑是一种代码的坏味道，应该想办法解决它。
适配器（包装器）

这种常用的设计模式在spring框架里用的非常多，如ServletRequestAttributes,它包装了HttpServletRequest和HttpServletResponse，用来添加额外的功能，如获取请求参数、session等信息。
为获取IP地址添加包装

自己的http请求包装器接口RequestWrapper

1. public interface RequestWrapper {
3.         String getHeader(String name);
5.         String getRemoteAddr();
7. }

复制代码

HttpServletRequest的包装

1. public class HttpRequestWrapper implements RequestWrapper {
3.         private final HttpServletRequest request;
5.         public HttpRequestWrapper(HttpServletRequest request) {
6.                 this.request = request;
7.         }
9.         @Override
10.         public String getHeader(String name) {
11.                 return request.getHeader(name);
12.         }
14.         @Override
15.         public String getRemoteAddr() {
16.                 return request.getRemoteAddr();
17.         }
19. }

复制代码

NettyRequestFacade的包装

1. public class NettyRequestWrapper implements RequestWrapper {
3.         private final NettyRequestFacade request;
5.         public NettyRequestWrapper(NettyRequestFacade request) {
6.                 this.request = request;
7.         }
8.         @Override
9.         public String getHeader(String name) {
10.                 return request.getHeader(name);
11.         }
13.         @Override
14.         public String getRemoteAddr() {
15.                 return request.getRemoteAddr();
16.         }
18. }

复制代码

获取IP地址方法，入参是一个http包装器

1. public static String getIpAddr(RequestWrapper request) {
2.                 String ip = request.getHeader("x-forwarded-for");
3.                 if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
4.                         ip = request.getHeader("Proxy-Client-IP");
5.                 }
6.                 if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
7.                         ip = request.getHeader("WL-Proxy-Client-IP");
8.                 }
9.                 if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
10.                         ip = request.getRemoteAddr();
11.                         if (ip.equals("127.0.0.1")) {
12.                                 // 根据网卡取本机配置的IP
13.                                 InetAddress inet = null;
14.                                 try {
15.                                         inet = InetAddress.getLocalHost();
16.                                 }
17.                                 catch (UnknownHostException e) {
18.                                         e.printStackTrace();
19.                                 }
20.                                 ip = inet != null ? inet.getHostAddress() : ip;
21.                         }
22.                 }
23.                 // 对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照','分割
24.                 if (ip != null && ip.length() > 15) {
25.                         if (ip.indexOf(",") > 0) {
26.                                 ip = ip.substring(0, ip.indexOf(","));
27.                         }
28.                 }
29.                 if ("0:0:0:0:0:0:0:1".equals(ip)) {
30.                         ip = "127.0.0.1";
31.                 }
32.                 return ip;
33.         }

复制代码

调用示例：

1. // 使用HttpServletRequest
2. public String getIpFromHttp(HttpServletRequest request) {
3.     return getIpAddr(new HttpRequestWrapper(request));
4. }
6. // 使用NettyRequestFacade
7. public String getIpFromNetty(NettyRequestFacade request) {
8.     return getIpAddr(new NettyRequestWrapper(request));
9. }

复制代码

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！