Netty 学习（2）

package yang.netty.test;

import static org.jboss.netty.buffer.ChannelBuffers.dynamicBuffer;

import java.net.InetSocketAddress;
import java.util.concurrent.Executors;

import org.jboss.netty.bootstrap.ClientBootstrap;
import org.jboss.netty.bootstrap.ServerBootstrap;
import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelFactory;
import org.jboss.netty.channel.ChannelFuture;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelPipelineFactory;
import org.jboss.netty.channel.ChannelStateEvent;
import org.jboss.netty.channel.Channels;
import org.jboss.netty.channel.ExceptionEvent;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler;
import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;
import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;
import org.jboss.netty.handler.codec.frame.FrameDecoder;

class Persons {
	private String name;
	private int age;
	private double salary;

	public Persons(String name, int age, double salary) {
		this.name = name;
		this.age = age;
		this.salary = salary;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public int getAge() {
		return age;
	}

	public void setAge(int age) {
		this.age = age;
	}

	public double getSalary() {
		return salary;
	}

	public void setSalary(double salary) {
		this.salary = salary;
	}

	@Override
	public String toString() {
		return "Persons [name=" + name + ", age=" + age + ", salary=" + salary
				+ "]";
	}
}

class TimeClientHandler extends SimpleChannelHandler {

	@Override
	public void channelConnected(ChannelHandlerContext ctx, ChannelStateEvent e)
			throws Exception {
		Persons person = new Persons("杨少飞", 24, 8888.88);
		e.getChannel().write(person);
		System.out.println("Send Message: " + person);
	}
	
	public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) {
		System.out.println("Unexpected exception from downstream."
				+ e.getCause());
		e.getChannel().close();
	}
}

class  TimeServerHandler extends SimpleChannelHandler {

	@Override
	public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)
			throws Exception {
		Persons person = (Persons) e.getMessage();
		System.out.println("Recevied Message: " + person);
//		System.out.println("e.message: " + e.getMessage());
//		System.out.println("e.channel: " + e.getChannel());
//		System.out.println("e.RemoteAddress: " + e.getRemoteAddress());
	}
	
	public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) {
		System.out.println("Unexpected exception from upstream."
				+ e.getCause());
		e.getChannel().close();
	}
}

class TimeDecoder extends FrameDecoder {
	private final ChannelBuffer buffer = dynamicBuffer();//默认256

	@Override
	protected Object decode(ChannelHandlerContext ctx, Channel channel,
			ChannelBuffer channelBuffer) throws Exception {
		
		 System.out.println("Decode Method");
//		System.out.println("channelBuffer bytes: " + channelBuffer.readableBytes());
		if (channelBuffer.readable()) {
			//将channelBuffer中的数据转移的buffer中
			channelBuffer.readBytes(buffer, channelBuffer.readableBytes());
		}
//		System.out.println("readable bytes: " + buffer.readableBytes( ));
//        System.out.println("readable index: " + buffer.readerIndex( ));
//        System.out.println("writable bytes: " + buffer.writableBytes( ));
//        System.out.println("writable index: " + buffer.writerIndex( ));
        
		int namelength = buffer.readInt();
		String name = new String(buffer.readBytes(namelength).array(), "GBK");
		int age = buffer.readInt();
		double salary = buffer.readDouble();
		Persons person = new Persons(name, age, salary);
		return person;
	}

}

class TimeEncoder extends SimpleChannelHandler {
	private final ChannelBuffer buffer = dynamicBuffer();

	@Override
	public void writeRequested(ChannelHandlerContext ctx, MessageEvent e)
			throws Exception {
		 System.out.println("Encode Method");
		 
		Persons person = (Persons) e.getMessage();
		buffer.writeInt(person.getName().getBytes("GBK").length);//4
		buffer.writeBytes(person.getName().getBytes("GBK"));//6
		buffer.writeInt(person.getAge());//4
		buffer.writeDouble(person.getSalary());//8
		Channels.write(ctx, e.getFuture(), buffer);
	}
}

public class TestCase2 {
	public static void testServer() {
		ChannelFactory factory = new NioServerSocketChannelFactory(
				Executors.newCachedThreadPool(),
				Executors.newCachedThreadPool());
		ServerBootstrap bootstrap = new ServerBootstrap(factory);
		bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
			public ChannelPipeline getPipeline() throws Exception {
				ChannelPipeline pipeline = Channels.pipeline();
				pipeline.addLast("decoder", new TimeDecoder());
				pipeline.addLast("encoder", new TimeEncoder());
				pipeline.addLast("handler", new TimeServerHandler());
				return pipeline;
			}
		});
		bootstrap.setOption("child.tcpNoDelay", true);
		bootstrap.setOption("child.keepAlive", true);
		bootstrap.bind(new InetSocketAddress("localhost", 3333));
		System.out.println("Server is on ....");
	}

	public static void testClient() {
		// 创建客户端channel的辅助类,发起connection请求
		ClientBootstrap bootstrap = new ClientBootstrap(
				new NioClientSocketChannelFactory(
						Executors.newCachedThreadPool(),
						Executors.newCachedThreadPool()));
		bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
			public ChannelPipeline getPipeline() {
				ChannelPipeline pipeline = Channels.pipeline();
				pipeline.addLast("decoder", new TimeDecoder());
				pipeline.addLast("encoder", new TimeEncoder());
				pipeline.addLast("handler", new TimeClientHandler());
				return pipeline;
			}
		});
		
		/*
		 bbootstrap.setPipelineFactory(new MyPipelineFactory());
		 public class MyPipelineFactory implements ChannelPipelineFactory {
		   public ChannelPipeline getPipeline() throws Exception {
		     // Create and configure a new pipeline for a new channel.
		     ChannelPipeline p = Channels.pipeline();
		     p.addLast("encoder", new EncodingHandler());
		     p.addLast("decoder", new DecodingHandler());
		     p.addLast("logic",   new LogicHandler());
		     return p;
		   }
		 }
		 */
		
		ChannelFuture future = bootstrap.connect(new InetSocketAddress(
				"localhost", 3333));
		System.out.println("Client is on ....");
		future.getChannel().getCloseFuture().awaitUninterruptibly();
		bootstrap.releaseExternalResources();
	}

	public static void main(String[] argv) {
		testServer();
		testClient();
	}
}

运行结果

Server is on ....
Client is on ....
Encode Method
Send Message: Persons [name=杨少飞, age=24, salary=8888.88]
Decode Method
Recevied Message: Persons [name=杨少飞, age=24, salary=8888.88]