程序博客网 > 软件导刊 质量

netty去实现文件传输

来源：互联网 发布：软件导刊 质量 编辑：程序博客网 时间：2024/06/05 01:14

在写出了Netty Hello World 和 netty对象传输之后，又觉得不够，看了官网的例子，所以有了现在的这个文件传输。
顺便说下，netty官网的例子真的好，如果要学习netty，还是看官网例子的好。
不过我英文不太好，刚开始走了绕了好大一圈，但是现在熟悉了之后，回过头来看，还是官网的牛X。
在这里再说下netty的零拷贝，这个零拷贝是netty在3.2版本中新加入的功能。
其主要指的是在进行一些比较大的传输比如对象或者文件传输的时候，通过改变数组索引的方式，将数据传输到特定的channel上。
是的channel之间的转换是零拷贝，例如：ByteBuffer和ChannelBuffer
下面就把文件传输的例子贴出来吧。
1、FileClient.java

[java] view plaincopy在CODE上查看代码片派生到我的代码片
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.channel.Channels.pipeline;  
  4.   
  5. import java.net.InetSocketAddress;  
  6. import java.util.concurrent.Executors;  
  7.   
  8. import org.jboss.netty.bootstrap.ClientBootstrap;  
  9. import org.jboss.netty.channel.ChannelFuture;  
  10. import org.jboss.netty.channel.ChannelPipeline;  
  11. import org.jboss.netty.channel.ChannelPipelineFactory;  
  12. import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;  
  13. import org.jboss.netty.handler.codec.http.DefaultHttpRequest;  
  14. import org.jboss.netty.handler.codec.http.HttpMethod;  
  15. import org.jboss.netty.handler.codec.http.HttpRequest;  
  16. import org.jboss.netty.handler.codec.http.HttpRequestEncoder;  
  17. import org.jboss.netty.handler.codec.http.HttpResponseDecoder;  
  18. import org.jboss.netty.handler.codec.http.HttpVersion;  
  19. import org.jboss.netty.handler.stream.ChunkedWriteHandler;  
  20.   
  21. public class FileClient  
  22. {  
  23.     public static void main(String[] args)  
  24.     {  
  25.   
  26.         ClientBootstrap bootstrap = new ClientBootstrap(  
  27.                 new NioClientSocketChannelFactory(  
  28.                         Executors.newCachedThreadPool(),  
  29.                         Executors.newCachedThreadPool()));  
  30.         bootstrap.setPipelineFactory(new ChannelPipelineFactory()  
  31.         {  
  32.   
  33.             @Override  
  34.             public ChannelPipeline getPipeline() throws Exception  
  35.             {  
  36.                 ChannelPipeline pipeline = pipeline();  
  37.   
  38.                 pipeline.addLast("decoder", new HttpResponseDecoder());  
  39.                   
  40.                 /* 
  41.                  * 不能添加这个，对传输文件 进行了大小的限制。。。。。 
  42.                  */  
  43. //                pipeline.addLast("aggregator", new HttpChunkAggregator(6048576));  
  44.                 pipeline.addLast("encoder", new HttpRequestEncoder());  
  45.                 pipeline.addLast("chunkedWriter", new ChunkedWriteHandler());  
  46.                 pipeline.addLast("handler", new FileClientHandler());  
  47.   
  48.                 return pipeline;  
  49.             }  
  50.   
  51.         });  
  52.   
  53.         ChannelFuture future = bootstrap.connect(new InetSocketAddress(  
  54.                 "localhost", 8080));  
  55.   
  56.         /* 
  57.          * 这里为了保证connect连接，所以才进行了sleep 
  58.          * 当然也可以通过future的connect属性判断 
  59.          */  
  60.         try  
  61.         {  
  62.             Thread.sleep(3000);  
  63.         } catch (InterruptedException e)  
  64.         {  
  65.             e.printStackTrace();  
  66.         }  
  67.         HttpRequest request = new DefaultHttpRequest(HttpVersion.HTTP_1_1,  
  68.                 HttpMethod.GET, "DSC01575.JPG");  
  69.         future.getChannel().write(request);  
  70.   
  71.         // Wait until the connection is closed or the connection attempt fails.  
  72.         future.getChannel().getCloseFuture().awaitUninterruptibly();  
  73.   
  74.         // Shut down thread pools to exit.  
  75.         bootstrap.releaseExternalResources();  
  76.   
  77.     }  
  78. }  
[java] view plain copy
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.channel.Channels.pipeline;  
  4.   
  5. import java.net.InetSocketAddress;  
  6. import java.util.concurrent.Executors;  
  7.   
  8. import org.jboss.netty.bootstrap.ClientBootstrap;  
  9. import org.jboss.netty.channel.ChannelFuture;  
  10. import org.jboss.netty.channel.ChannelPipeline;  
  11. import org.jboss.netty.channel.ChannelPipelineFactory;  
  12. import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;  
  13. import org.jboss.netty.handler.codec.http.DefaultHttpRequest;  
  14. import org.jboss.netty.handler.codec.http.HttpMethod;  
  15. import org.jboss.netty.handler.codec.http.HttpRequest;  
  16. import org.jboss.netty.handler.codec.http.HttpRequestEncoder;  
  17. import org.jboss.netty.handler.codec.http.HttpResponseDecoder;  
  18. import org.jboss.netty.handler.codec.http.HttpVersion;  
  19. import org.jboss.netty.handler.stream.ChunkedWriteHandler;  
  20.   
  21. public class FileClient  
  22. {  
  23.     public static void main(String[] args)  
  24.     {  
  25.   
  26.         ClientBootstrap bootstrap = new ClientBootstrap(  
  27.                 new NioClientSocketChannelFactory(  
  28.                         Executors.newCachedThreadPool(),  
  29.                         Executors.newCachedThreadPool()));  
  30.         bootstrap.setPipelineFactory(new ChannelPipelineFactory()  
  31.         {  
  32.   
  33.             @Override  
  34.             public ChannelPipeline getPipeline() throws Exception  
  35.             {  
  36.                 ChannelPipeline pipeline = pipeline();  
  37.   
  38.                 pipeline.addLast("decoder", new HttpResponseDecoder());  
  39.                   
  40.                 /* 
  41.                  * 不能添加这个，对传输文件 进行了大小的限制。。。。。 
  42.                  */  
  43. //                pipeline.addLast("aggregator", new HttpChunkAggregator(6048576));  
  44.                 pipeline.addLast("encoder", new HttpRequestEncoder());  
  45.                 pipeline.addLast("chunkedWriter", new ChunkedWriteHandler());  
  46.                 pipeline.addLast("handler", new FileClientHandler());  
  47.   
  48.                 return pipeline;  
  49.             }  
  50.   
  51.         });  
  52.   
  53.         ChannelFuture future = bootstrap.connect(new InetSocketAddress(  
  54.                 "localhost", 8080));  
  55.   
  56.         /* 
  57.          * 这里为了保证connect连接，所以才进行了sleep 
  58.          * 当然也可以通过future的connect属性判断 
  59.          */  
  60.         try  
  61.         {  
  62.             Thread.sleep(3000);  
  63.         } catch (InterruptedException e)  
  64.         {  
  65.             e.printStackTrace();  
  66.         }  
  67.         HttpRequest request = new DefaultHttpRequest(HttpVersion.HTTP_1_1,  
  68.                 HttpMethod.GET, "DSC01575.JPG");  
  69.         future.getChannel().write(request);  
  70.   
  71.         // Wait until the connection is closed or the connection attempt fails.  
  72.         future.getChannel().getCloseFuture().awaitUninterruptibly();  
  73.   
  74.         // Shut down thread pools to exit.  
  75.         bootstrap.releaseExternalResources();  
  76.   
  77.     }  
  78. }  

2、FileClientHandler.java

[java] view plaincopy在CODE上查看代码片派生到我的代码片
  1. package filetrans;  
  2.   
  3. import java.io.File;  
  4. import java.io.FileOutputStream;  
  5.   
  6. import org.jboss.netty.buffer.ChannelBuffer;  
  7. import org.jboss.netty.channel.ChannelHandlerContext;  
  8. import org.jboss.netty.channel.ExceptionEvent;  
  9. import org.jboss.netty.channel.MessageEvent;  
  10. import org.jboss.netty.channel.SimpleChannelUpstreamHandler;  
  11. import org.jboss.netty.handler.codec.http.DefaultHttpResponse;  
  12. import org.jboss.netty.handler.codec.http.HttpChunk;  
  13. import org.jboss.netty.handler.codec.http.HttpResponse;  
  14.   
  15. public class FileClientHandler extends SimpleChannelUpstreamHandler  
  16. {  
  17.     private volatile boolean readingChunks;  
  18.     private File downloadFile;  
  19.     private FileOutputStream fOutputStream = null;  
  20.   
  21.     @Override  
  22.     public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)  
  23.             throws Exception  
  24.     {  
  25.         /* 
  26.          * 按照channle的顺序进行处理 
  27.          * server先发送HttpResponse过来，所以这里先对HttpResponse进行处理，进行文件判断之类 
  28.          * 之后，server发送的都是ChunkedFile了。 
  29.          */  
  30.           
  31.         if (e.getMessage() instanceof HttpResponse)  
  32.         {  
  33.             DefaultHttpResponse httpResponse = (DefaultHttpResponse) e  
  34.                     .getMessage();  
  35.             String fileName = httpResponse.getHeader("fileName");  
  36.             downloadFile = new File(System.getProperty("user.dir")  
  37.                     + File.separator + "recived_" + fileName);  
  38.             readingChunks = httpResponse.isChunked();  
  39.         } else  
  40.         {  
  41.             HttpChunk httpChunk = (HttpChunk) e.getMessage();  
  42.             if (!httpChunk.isLast())  
  43.             {  
  44.                 ChannelBuffer buffer = httpChunk.getContent();  
  45.                 if (fOutputStream == null)  
  46.                 {  
  47.                     fOutputStream = new FileOutputStream(downloadFile);  
  48.                 }  
  49.                 while (buffer.readable())  
  50.                 {  
  51.                     byte[] dst = new byte[buffer.readableBytes()];  
  52.                     buffer.readBytes(dst);  
  53.                     fOutputStream.write(dst);  
  54.                 }  
  55.             } else  
  56.             {  
  57.                 readingChunks = false;  
  58.             }  
  59.             fOutputStream.flush();  
  60.         }  
  61.         if (!readingChunks)  
  62.         {  
  63.             fOutputStream.close();  
  64.             e.getChannel().close();  
  65.         }  
  66.     }  
  67.   
  68.     @Override  
  69.     public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e)  
  70.             throws Exception  
  71.     {  
  72.         System.out.println(e.getCause());  
  73.     }  
  74. }  
[java] view plain copy
  1. package filetrans;  
  2.   
  3. import java.io.File;  
  4. import java.io.FileOutputStream;  
  5.   
  6. import org.jboss.netty.buffer.ChannelBuffer;  
  7. import org.jboss.netty.channel.ChannelHandlerContext;  
  8. import org.jboss.netty.channel.ExceptionEvent;  
  9. import org.jboss.netty.channel.MessageEvent;  
  10. import org.jboss.netty.channel.SimpleChannelUpstreamHandler;  
  11. import org.jboss.netty.handler.codec.http.DefaultHttpResponse;  
  12. import org.jboss.netty.handler.codec.http.HttpChunk;  
  13. import org.jboss.netty.handler.codec.http.HttpResponse;  
  14.   
  15. public class FileClientHandler extends SimpleChannelUpstreamHandler  
  16. {  
  17.     private volatile boolean readingChunks;  
  18.     private File downloadFile;  
  19.     private FileOutputStream fOutputStream = null;  
  20.   
  21.     @Override  
  22.     public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)  
  23.             throws Exception  
  24.     {  
  25.         /* 
  26.          * 按照channle的顺序进行处理 
  27.          * server先发送HttpResponse过来，所以这里先对HttpResponse进行处理，进行文件判断之类 
  28.          * 之后，server发送的都是ChunkedFile了。 
  29.          */  
  30.           
  31.         if (e.getMessage() instanceof HttpResponse)  
  32.         {  
  33.             DefaultHttpResponse httpResponse = (DefaultHttpResponse) e  
  34.                     .getMessage();  
  35.             String fileName = httpResponse.getHeader("fileName");  
  36.             downloadFile = new File(System.getProperty("user.dir")  
  37.                     + File.separator + "recived_" + fileName);  
  38.             readingChunks = httpResponse.isChunked();  
  39.         } else  
  40.         {  
  41.             HttpChunk httpChunk = (HttpChunk) e.getMessage();  
  42.             if (!httpChunk.isLast())  
  43.             {  
  44.                 ChannelBuffer buffer = httpChunk.getContent();  
  45.                 if (fOutputStream == null)  
  46.                 {  
  47.                     fOutputStream = new FileOutputStream(downloadFile);  
  48.                 }  
  49.                 while (buffer.readable())  
  50.                 {  
  51.                     byte[] dst = new byte[buffer.readableBytes()];  
  52.                     buffer.readBytes(dst);  
  53.                     fOutputStream.write(dst);  
  54.                 }  
  55.             } else  
  56.             {  
  57.                 readingChunks = false;  
  58.             }  
  59.             fOutputStream.flush();  
  60.         }  
  61.         if (!readingChunks)  
  62.         {  
  63.             fOutputStream.close();  
  64.             e.getChannel().close();  
  65.         }  
  66.     }  
  67.   
  68.     @Override  
  69.     public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e)  
  70.             throws Exception  
  71.     {  
  72.         System.out.println(e.getCause());  
  73.     }  
  74. }  

3、FileServer.java

[java] view plaincopy在CODE上查看代码片派生到我的代码片
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.channel.Channels.pipeline;  
  4.   
  5. import java.net.InetSocketAddress;  
  6. import java.util.concurrent.Executors;  
  7.   
  8. import org.jboss.netty.bootstrap.ServerBootstrap;  
  9. import org.jboss.netty.channel.ChannelPipeline;  
  10. import org.jboss.netty.channel.ChannelPipelineFactory;  
  11. import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;  
  12. import org.jboss.netty.handler.codec.http.HttpChunkAggregator;  
  13. import org.jboss.netty.handler.codec.http.HttpRequestDecoder;  
  14. import org.jboss.netty.handler.codec.http.HttpResponseEncoder;  
  15. import org.jboss.netty.handler.stream.ChunkedWriteHandler;  
  16.   
  17. public class FileServer  
  18. {  
  19.     public static void main(String[] args)  
  20.     {  
  21.   
  22.         ServerBootstrap bootstrap = new ServerBootstrap(  
  23.                 new NioServerSocketChannelFactory(  
  24.                         Executors.newCachedThreadPool(),  
  25.                         Executors.newCachedThreadPool()));  
  26.   
  27.         bootstrap.setPipelineFactory(new ChannelPipelineFactory()  
  28.         {  
  29.   
  30.             @Override  
  31.             public ChannelPipeline getPipeline() throws Exception  
  32.             {  
  33.                 ChannelPipeline pipeline = pipeline();  
  34.                 pipeline.addLast("decoder", new HttpRequestDecoder());  
  35.                 pipeline.addLast("aggregator", new HttpChunkAggregator(65536));  
  36.                 pipeline.addLast("encoder", new HttpResponseEncoder());  
  37.                 pipeline.addLast("chunkedWriter", new ChunkedWriteHandler());  
  38.   
  39.                 pipeline.addLast("handler", new FileServerHandler());  
  40.                 return pipeline;  
  41.             }  
  42.   
  43.         });  
  44.   
  45.         bootstrap.bind(new InetSocketAddress(8080));  
  46.     }  
  47. }  
[java] view plain copy
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.channel.Channels.pipeline;  
  4.   
  5. import java.net.InetSocketAddress;  
  6. import java.util.concurrent.Executors;  
  7.   
  8. import org.jboss.netty.bootstrap.ServerBootstrap;  
  9. import org.jboss.netty.channel.ChannelPipeline;  
  10. import org.jboss.netty.channel.ChannelPipelineFactory;  
  11. import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;  
  12. import org.jboss.netty.handler.codec.http.HttpChunkAggregator;  
  13. import org.jboss.netty.handler.codec.http.HttpRequestDecoder;  
  14. import org.jboss.netty.handler.codec.http.HttpResponseEncoder;  
  15. import org.jboss.netty.handler.stream.ChunkedWriteHandler;  
  16.   
  17. public class FileServer  
  18. {  
  19.     public static void main(String[] args)  
  20.     {  
  21.   
  22.         ServerBootstrap bootstrap = new ServerBootstrap(  
  23.                 new NioServerSocketChannelFactory(  
  24.                         Executors.newCachedThreadPool(),  
  25.                         Executors.newCachedThreadPool()));  
  26.   
  27.         bootstrap.setPipelineFactory(new ChannelPipelineFactory()  
  28.         {  
  29.   
  30.             @Override  
  31.             public ChannelPipeline getPipeline() throws Exception  
  32.             {  
  33.                 ChannelPipeline pipeline = pipeline();  
  34.                 pipeline.addLast("decoder", new HttpRequestDecoder());  
  35.                 pipeline.addLast("aggregator", new HttpChunkAggregator(65536));  
  36.                 pipeline.addLast("encoder", new HttpResponseEncoder());  
  37.                 pipeline.addLast("chunkedWriter", new ChunkedWriteHandler());  
  38.   
  39.                 pipeline.addLast("handler", new FileServerHandler());  
  40.                 return pipeline;  
  41.             }  
  42.   
  43.         });  
  44.   
  45.         bootstrap.bind(new InetSocketAddress(8080));  
  46.     }  
  47. }  

4、FileServerHandler.java

[java] view plaincopy在CODE上查看代码片派生到我的代码片
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.handler.codec.http.HttpHeaders.*;  
  4. import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.*;  
  5. import static org.jboss.netty.handler.codec.http.HttpMethod.*;  
  6. import static org.jboss.netty.handler.codec.http.HttpResponseStatus.*;  
  7. import static org.jboss.netty.handler.codec.http.HttpVersion.*;  
  8.   
  9. import java.io.File;  
  10. import java.io.FileNotFoundException;  
  11. import java.io.RandomAccessFile;  
  12. import java.io.UnsupportedEncodingException;  
  13. import java.net.URLDecoder;  
  14.   
  15. import org.jboss.netty.buffer.ChannelBuffers;  
  16. import org.jboss.netty.channel.Channel;  
  17. import org.jboss.netty.channel.ChannelFuture;  
  18. import org.jboss.netty.channel.ChannelFutureListener;  
  19. import org.jboss.netty.channel.ChannelFutureProgressListener;  
  20. import org.jboss.netty.channel.ChannelHandlerContext;  
  21. import org.jboss.netty.channel.DefaultFileRegion;  
  22. import org.jboss.netty.channel.ExceptionEvent;  
  23. import org.jboss.netty.channel.FileRegion;  
  24. import org.jboss.netty.channel.MessageEvent;  
  25. import org.jboss.netty.channel.SimpleChannelUpstreamHandler;  
  26. import org.jboss.netty.handler.codec.frame.TooLongFrameException;  
  27. import org.jboss.netty.handler.codec.http.DefaultHttpResponse;  
  28. import org.jboss.netty.handler.codec.http.HttpRequest;  
  29. import org.jboss.netty.handler.codec.http.HttpResponse;  
  30. import org.jboss.netty.handler.codec.http.HttpResponseStatus;  
  31. import org.jboss.netty.handler.ssl.SslHandler;  
  32. import org.jboss.netty.handler.stream.ChunkedFile;  
  33. import org.jboss.netty.util.CharsetUtil;  
  34.   
  35. /** 
  36.  * 这里的代码主要来源于官网的例子，http里面有个例子，自己仿照server写了client 
  37.  * @author Ransom 
  38.  */  
  39. public class FileServerHandler extends SimpleChannelUpstreamHandler  
  40. {  
  41.   
  42.     @Override  
  43.     public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)  
  44.             throws Exception  
  45.     {  
  46.         HttpRequest request = (HttpRequest) e.getMessage();  
  47.         if (request.getMethod() != GET)  
  48.         {  
  49.             sendError(ctx, METHOD_NOT_ALLOWED);  
  50.             return;  
  51.         }  
  52.   
  53.         final String path = sanitizeUri(request.getUri());  
  54.         if (path == null)  
  55.         {  
  56.             sendError(ctx, FORBIDDEN);  
  57.             return;  
  58.         }  
  59.   
  60.         File file = new File(path);  
  61.         if (file.isHidden() || !file.exists())  
  62.         {  
  63.             sendError(ctx, NOT_FOUND);  
  64.             return;  
  65.         }  
  66.         if (!file.isFile())  
  67.         {  
  68.             sendError(ctx, FORBIDDEN);  
  69.             return;  
  70.         }  
  71.   
  72.         RandomAccessFile raf;  
  73.         try  
  74.         {  
  75.             raf = new RandomAccessFile(file, "r");  
  76.         } catch (FileNotFoundException fnfe)  
  77.         {  
  78.             sendError(ctx, NOT_FOUND);  
  79.             return;  
  80.         }  
  81.         long fileLength = raf.length();  
  82.   
  83.         HttpResponse response = new DefaultHttpResponse(HTTP_1_1, OK);  
  84.           
  85.         /* 
  86.          * 由于是异步传输，所以不得已加入了一些属性，用来进行文件识别 
  87.          */  
  88.         response.addHeader("fileName", request.getUri());  
  89.           
  90.         setContentLength(response, fileLength);  
  91.   
  92.         Channel ch = e.getChannel();  
  93.   
  94.         // Write the initial line and the header.  
  95.         ch.write(response);  
  96.   
  97.         // Write the content.  
  98.         ChannelFuture writeFuture;  
  99.         if (ch.getPipeline().get(SslHandler.class) != null)  
  100.         {  
  101.             // Cannot use zero-copy with HTTPS.  
  102.             writeFuture = ch.write(new ChunkedFile(raf, 0, fileLength, 8192));  
  103.         } else  
  104.         {  
  105.             // No encryption - use zero-copy.  
  106.             final FileRegion region = new DefaultFileRegion(raf.getChannel(),  
  107.                     0, fileLength);  
  108.             writeFuture = ch.write(region);  
  109.             writeFuture.addListener(new ChannelFutureProgressListener()  
  110.             {  
  111.                 public void operationComplete(ChannelFuture future)  
  112.                 {  
  113.                     region.releaseExternalResources();  
  114.                 }  
  115.   
  116.                 public void operationProgressed(ChannelFuture future,  
  117.                         long amount, long current, long total)  
  118.                 {  
  119.                     System.out.printf("%s: %d / %d (+%d)%n", path, current,  
  120.                             total, amount);  
  121.                 }  
  122.             });  
  123.         }  
  124.   
  125.         // Decide whether to close the connection or not.  
  126.         if (!isKeepAlive(request))  
  127.         {  
  128.             // Close the connection when the whole content is written out.  
  129.             writeFuture.addListener(ChannelFutureListener.CLOSE);  
  130.         }  
  131.     }  
  132.   
  133.     @Override  
  134.     public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e)  
  135.             throws Exception  
  136.     {  
  137.         Channel ch = e.getChannel();  
  138.         Throwable cause = e.getCause();  
  139.         if (cause instanceof TooLongFrameException)  
  140.         {  
  141.             sendError(ctx, BAD_REQUEST);  
  142.             return;  
  143.         }  
  144.   
  145.         cause.printStackTrace();  
  146.         if (ch.isConnected())  
  147.         {  
  148.             sendError(ctx, INTERNAL_SERVER_ERROR);  
  149.         }  
  150.     }  
  151.   
  152.     private String sanitizeUri(String uri)  
  153.     {  
  154.         // Decode the path.  
  155.         try  
  156.         {  
  157.             uri = URLDecoder.decode(uri, "UTF-8");  
  158.         } catch (UnsupportedEncodingException e)  
  159.         {  
  160.             try  
  161.             {  
  162.                 uri = URLDecoder.decode(uri, "ISO-8859-1");  
  163.             } catch (UnsupportedEncodingException e1)  
  164.             {  
  165.                 throw new Error();  
  166.             }  
  167.         }  
  168.   
  169.         // Convert file separators.  
  170.         uri = uri.replace('/', File.separatorChar);  
  171.   
  172.         // Simplistic dumb security check.  
  173.         // You will have to do something serious in the production environment.  
  174.         if (uri.contains(File.separator + ".")  
  175.                 || uri.contains("." + File.separator) || uri.startsWith(".")  
  176.                 || uri.endsWith("."))  
  177.         {  
  178.             return null;  
  179.         }  
  180.   
  181.         // Convert to absolute path.  
  182.         return System.getProperty("user.dir") + File.separator + uri;  
  183.     }  
  184.   
  185.     private void sendError(ChannelHandlerContext ctx, HttpResponseStatus status)  
  186.     {  
  187.         HttpResponse response = new DefaultHttpResponse(HTTP_1_1, status);  
  188.         response.setHeader(CONTENT_TYPE, "text/plain; charset=UTF-8");  
  189.         response.setContent(ChannelBuffers.copiedBuffer(  
  190.                 "Failure: " + status.toString() + "\r\n", CharsetUtil.UTF_8));  
  191.   
  192.         // Close the connection as soon as the error message is sent.  
  193.         ctx.getChannel().write(response)  
  194.                 .addListener(ChannelFutureListener.CLOSE);  
  195.     }  
  196. }  
[java] view plain copy
  1. package filetrans;  
  2.   
  3. import static org.jboss.netty.handler.codec.http.HttpHeaders.*;  
  4. import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.*;  
  5. import static org.jboss.netty.handler.codec.http.HttpMethod.*;  
  6. import static org.jboss.netty.handler.codec.http.HttpResponseStatus.*;  
  7. import static org.jboss.netty.handler.codec.http.HttpVersion.*;  
  8.   
  9. import java.io.File;  
  10. import java.io.FileNotFoundException;  
  11. import java.io.RandomAccessFile;  
  12. import java.io.UnsupportedEncodingException;  
  13. import java.net.URLDecoder;  
  14.   
  15. import org.jboss.netty.buffer.ChannelBuffers;  
  16. import org.jboss.netty.channel.Channel;  
  17. import org.jboss.netty.channel.ChannelFuture;  
  18. import org.jboss.netty.channel.ChannelFutureListener;  
  19. import org.jboss.netty.channel.ChannelFutureProgressListener;  
  20. import org.jboss.netty.channel.ChannelHandlerContext;  
  21. import org.jboss.netty.channel.DefaultFileRegion;  
  22. import org.jboss.netty.channel.ExceptionEvent;  
  23. import org.jboss.netty.channel.FileRegion;  
  24. import org.jboss.netty.channel.MessageEvent;  
  25. import org.jboss.netty.channel.SimpleChannelUpstreamHandler;  
  26. import org.jboss.netty.handler.codec.frame.TooLongFrameException;  
  27. import org.jboss.netty.handler.codec.http.DefaultHttpResponse;  
  28. import org.jboss.netty.handler.codec.http.HttpRequest;  
  29. import org.jboss.netty.handler.codec.http.HttpResponse;  
  30. import org.jboss.netty.handler.codec.http.HttpResponseStatus;  
  31. import org.jboss.netty.handler.ssl.SslHandler;  
  32. import org.jboss.netty.handler.stream.ChunkedFile;  
  33. import org.jboss.netty.util.CharsetUtil;  
  34.   
  35. /** 
  36.  * 这里的代码主要来源于官网的例子，http里面有个例子，自己仿照server写了client 
  37.  * @author Ransom 
  38.  */  
  39. public class FileServerHandler extends SimpleChannelUpstreamHandler  
  40. {  
  41.   
  42.     @Override  
  43.     public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)  
  44.             throws Exception  
  45.     {  
  46.         HttpRequest request = (HttpRequest) e.getMessage();  
  47.         if (request.getMethod() != GET)  
  48.         {  
  49.             sendError(ctx, METHOD_NOT_ALLOWED);  
  50.             return;  
  51.         }  
  52.   
  53.         final String path = sanitizeUri(request.getUri());  
  54.         if (path == null)  
  55.         {  
  56.             sendError(ctx, FORBIDDEN);  
  57.             return;  
  58.         }  
  59.   
  60.         File file = new File(path);  
  61.         if (file.isHidden() || !file.exists())  
  62.         {  
  63.             sendError(ctx, NOT_FOUND);  
  64.             return;  
  65.         }  
  66.         if (!file.isFile())  
  67.         {  
  68.             sendError(ctx, FORBIDDEN);  
  69.             return;  
  70.         }  
  71.   
  72.         RandomAccessFile raf;  
  73.         try  
  74.         {  
  75.             raf = new RandomAccessFile(file, "r");  
  76.         } catch (FileNotFoundException fnfe)  
  77.         {  
  78.             sendError(ctx, NOT_FOUND);  
  79.             return;  
  80.         }  
  81.         long fileLength = raf.length();  
  82.   
  83.         HttpResponse response = new DefaultHttpResponse(HTTP_1_1, OK);  
  84.           
  85.         /* 
  86.          * 由于是异步传输，所以不得已加入了一些属性，用来进行文件识别 
  87.          */  
  88.         response.addHeader("fileName", request.getUri());  
  89.           
  90.         setContentLength(response, fileLength);  
  91.   
  92.         Channel ch = e.getChannel();  
  93.   
  94.         // Write the initial line and the header.  
  95.         ch.write(response);  
  96.   
  97.         // Write the content.  
  98.         ChannelFuture writeFuture;  
  99.         if (ch.getPipeline().get(SslHandler.class) != null)  
  100.         {  
  101.             // Cannot use zero-copy with HTTPS.  
  102.             writeFuture = ch.write(new ChunkedFile(raf, 0, fileLength, 8192));  
  103.         } else  
  104.         {  
  105.             // No encryption - use zero-copy.  
  106.             final FileRegion region = new DefaultFileRegion(raf.getChannel(),  
  107.                     0, fileLength);  
  108.             writeFuture = ch.write(region);  
  109.             writeFuture.addListener(new ChannelFutureProgressListener()  
  110.             {  
  111.                 public void operationComplete(ChannelFuture future)  
  112.                 {  
  113.                     region.releaseExternalResources();  
  114.                 }  
  115.   
  116.                 public void operationProgressed(ChannelFuture future,  
  117.                         long amount, long current, long total)  
  118.                 {  
  119.                     System.out.printf("%s: %d / %d (+%d)%n", path, current,  
  120.                             total, amount);  
  121.                 }  
  122.             });  
  123.         }  
  124.   
  125.         // Decide whether to close the connection or not.  
  126.         if (!isKeepAlive(request))  
  127.         {  
  128.             // Close the connection when the whole content is written out.  
  129.             writeFuture.addListener(ChannelFutureListener.CLOSE);  
  130.         }  
  131.     }  
  132.   
  133.     @Override  
  134.     public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e)  
  135.             throws Exception  
  136.     {  
  137.         Channel ch = e.getChannel();  
  138.         Throwable cause = e.getCause();  
  139.         if (cause instanceof TooLongFrameException)  
  140.         {  
  141.             sendError(ctx, BAD_REQUEST);  
  142.             return;  
  143.         }  
  144.   
  145.         cause.printStackTrace();  
  146.         if (ch.isConnected())  
  147.         {  
  148.             sendError(ctx, INTERNAL_SERVER_ERROR);  
  149.         }  
  150.     }  
  151.   
  152.     private String sanitizeUri(String uri)  
  153.     {  
  154.         // Decode the path.  
  155.         try  
  156.         {  
  157.             uri = URLDecoder.decode(uri, "UTF-8");  
  158.         } catch (UnsupportedEncodingException e)  
  159.         {  
  160.             try  
  161.             {  
  162.                 uri = URLDecoder.decode(uri, "ISO-8859-1");  
  163.             } catch (UnsupportedEncodingException e1)  
  164.             {  
  165.                 throw new Error();  
  166.             }  
  167.         }  
  168.   
  169.         // Convert file separators.  
  170.         uri = uri.replace('/', File.separatorChar);  
  171.   
  172.         // Simplistic dumb security check.  
  173.         // You will have to do something serious in the production environment.  
  174.         if (uri.contains(File.separator + ".")  
  175.                 || uri.contains("." + File.separator) || uri.startsWith(".")  
  176.                 || uri.endsWith("."))  
  177.         {  
  178.             return null;  
  179.         }  
  180.   
  181.         // Convert to absolute path.  
  182.         return System.getProperty("user.dir") + File.separator + uri;  
  183.     }  
  184.   
  185.     private void sendError(ChannelHandlerContext ctx, HttpResponseStatus status)  
  186.     {  
  187.         HttpResponse response = new DefaultHttpResponse(HTTP_1_1, status);  
  188.         response.setHeader(CONTENT_TYPE, "text/plain; charset=UTF-8");  
  189.         response.setContent(ChannelBuffers.copiedBuffer(  
  190.                 "Failure: " + status.toString() + "\r\n", CharsetUtil.UTF_8));  
  191.   
  192.         // Close the connection as soon as the error message is sent.  
  193.         ctx.getChannel().write(response)  
  194.                 .addListener(ChannelFutureListener.CLOSE);  
  195.     }  
  196. }  

Netty的例子到这里就算贴完了，后面还打算进行一些Netty实现原理，架构，代码解读方面的分析。到时候如果有什么心得，再写出来吧。

 

阅读全文
0 0
软件导刊 质量 软件导刊 质量
原创粉丝点击
热门IT博客
月相手表 知乎月相手表 知乎
淘宝卖家怎么寄快递淘宝卖家怎么寄快递
淘宝秒杀的验证码
淘宝买家怎么换货
网络电影点击率购买
c 网络爬虫 mac
管理系统数据库设计
手机软件恢复软件下载
什么调音软件好
中航prb打印机端口设置
多益网络客户端下载
soundtrack for mac
mac photos 使用教程
全面的财经数据
悟空网络加速器官网
淘宝第三方推广
关于乒乓球赛事的软件
linux 复制到tmp
淘宝截图怎么发给买家
mysql temporary
热门问题 老师的惩罚 人脸识别 我在镇武司摸鱼那些年 重生之率土为王 我在大康的咸鱼生活 盘龙之生命进化 天生仙种 凡人之先天五行 春回大明朝 姑娘不必设防，我是瞎子 公司下午茶 自助下午茶 法式下午茶 人气下午茶 清迈四季酒店下午茶 下半年考试时间 2019下半年放假安排 2019四六级下半年报名时间 2019下半年计算机二级报名时间 2019下半年考试时间 教资考试时间下半年 四级报名时间2019下半年 四级考试时间2019下半年 计算机二级报名时间2019下半年 下臂心肌缺血 失之交臂下一句 半臂纹身图案大全男 半臂纹身手稿图案大全 纹个半臂大概多少钱 冬季半裙 秋冬半裙 长裙半裙 半节裙 优衣库格子半裙 超辣半强迫篇短文集txt下载 半强迫篇短文集 半泽直树有多少集 物流跟单员 跟单员工资怎么那么高 跟单员工资一般多少 为什么跟单员工资高却没人做 派单员 打单员 业务跟单员好做吗 发单员 采购跟单员的工作内容 出单员 封单量小 大单净量是什么意思 外卖如何提高单量 外卖怎样提升单量

程序博客网，程序员的互联网技术博客家园。csdn论坛精品 msdn技术资料都在这里