自定義的投票協議的通信(tcp/ip 與 udp示例)

時間  2019-11-10
標籤 自定義 投票 協議 通信 tcp udp 示例 欄目 系統網絡 简体版
原文   原文鏈接

VoteMsg.java java

package com.tcpip;

/**
 * 投票信息
 * @author Administrator
 *
 */
public class VoteMsg {
	//請求
	private boolean isInquiry ;
	//迴應
	private boolean isResponse;
	//投票ID
	private int candidateID;
	//投票數
	private long voteCount;
	
	public static final int MAX_CANDIDATE_ID=1000;
	
	public VoteMsg(boolean isResponse,boolean isInquiry,int candidateID,long voteCount) throws IllegalArgumentException{
		if(voteCount!=0 && !isResponse){
			throw new IllegalArgumentException("Requset vote count must be zero");
		}
		if(candidateID<0 || candidateID>MAX_CANDIDATE_ID){
			throw new IllegalArgumentException("Bad Candidate ID:"+candidateID);
		}
		if(voteCount<0){
			throw new IllegalArgumentException("Total must be >= zero");
		}
		this.candidateID = candidateID;
		this.isResponse = isResponse;
		this.isInquiry = isInquiry;
		this.voteCount = voteCount;
	}

	public boolean isInquiry() {
		return isInquiry;
	}

	public void setInquiry(boolean isInquiry) {
		this.isInquiry = isInquiry;
	}

	public boolean isResponse() {
		return isResponse;
	}

	public void setResponse(boolean isResponse) {
		this.isResponse = isResponse;
	}

	public int getCandidateID() {
		return candidateID;
	}

	public void setCandidateID(int candidateID) throws IllegalArgumentException{
		if(candidateID<0 || candidateID>MAX_CANDIDATE_ID){
			throw new IllegalArgumentException("Bad Candidate ID:"+candidateID);
		}
		this.candidateID = candidateID;
	}

	public long getVoteCount() {
		return voteCount;
	}

	public void setVoteCount(long voteCount) {
		if((voteCount!=0&&!isResponse) || voteCount<0){
			throw new IllegalArgumentException("Bad vote count");
		}
		this.voteCount = voteCount;
	}
	
	public String toString(){
		String res = (isInquiry ? "inquiry" : "vote")+" for candidate "+candidateID;
		if(isResponse){
			res = "response to" + res +" who now has " + voteCount+ " vote(s)";
		}
		return res;
	}

}

接口類：VoteMsgCoder.java 

package com.tcpip;

import java.io.IOException;

/**
 * 投票消息處理(編碼與解碼)
 * @author Administrator
 *
 */
public interface VoteMsgCoder {
	/**
	 * 將投票消息轉換爲一個字節序列(根據特定的協議)
	 * @param msg
	 * @return
	 * @throws IOException
	 */
	byte[] toWire(VoteMsg msg) throws IOException;
	
	/**
	 * 對給定的字節序列進行解析(根據特定的協議)
	 * @param input
	 * @return
	 * @throws IOException
	 */
	VoteMsg fromWire(byte[] input) throws IOException;
}


實現類：VoteMsgTextCoder.java:服務器

package com.tcpip;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Scanner;

/**
 * 用文本方式對消息進行編碼
 * @author Administrator
 *
 */
public class VoteMsgTextCoder implements VoteMsgCoder {
	
	public static final String MAGIC = "Voting";
	public static final String VOTESTR = "v";
	public static final String INQSTR = "i";
	public static final String RESPONSESTR = "R";
	public static final String CHARSETNAME = "US-ASCII";
	public static final String DELIMSTR = " ";
	public static final int MAX_WIRE_LENGTH = 2000;
	
	/*
	 * 將投票消息轉換爲一個字節序列(根據特定的協議)
	 * @see com.tcpip.VoteMsgCoder#toWire(com.tcpip.VoteMsg)
	 */
	@Override
	public byte[] toWire(VoteMsg msg) throws IOException {
		String msgString = MAGIC + DELIMSTR + (msg.isInquiry() ? INQSTR : VOTESTR)
						 + DELIMSTR + (msg.isResponse() ? RESPONSESTR + DELIMSTR : "")
						 + Integer.toString(msg.getCandidateID()) + DELIMSTR 
						 + Long.toString(msg.getVoteCount());
		
		byte[]  data = msgString.getBytes(CHARSETNAME);
		return data;
	}

	/*
	 * 對給定的字節序列進行解析(根據特定的協議)
	 * @see com.tcpip.VoteMsgCoder#fromWire(byte[])
	 */
	@Override
	public VoteMsg fromWire(byte[] message) throws IOException {
		ByteArrayInputStream msgStream = new ByteArrayInputStream(message);
		Scanner s = new Scanner(new InputStreamReader(msgStream,CHARSETNAME));
		
		boolean isInquiry;
		boolean isResponse;
		int candidateID;
		long voteCount;
		String token;
		
		try{
			token = s.next();
			if(!token.equals(MAGIC)){
				throw new IOException("Bad magic string:"+token);
			}
			token = s.next();
			if(token.equals(VOTESTR)){
				isInquiry = false;
			}else if(!token.equals(INQSTR)){
				throw new IOException("Bad vote/inq indicator: "+token);
			}else {
				isInquiry = true;
			}
			token = s.next();
			if(token.equals(RESPONSESTR)){
				isResponse = true;
				token = s.next();
			}else{
				isResponse = false;
			}
			
			candidateID = Integer.parseInt(token);
			if(isResponse){
				token  = s.next();
				voteCount = Long.parseLong(token);
			}else{
				voteCount = 0;
			}
		}catch(IOException ioe){
			throw new IOException("Parse error...");
		}
		return new VoteMsg(isResponse,isInquiry,candidateID,voteCount);
	}

}


實現類：VoteMsgBinCoder.javatcp

package com.tcpip;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

/**
 * 用二進制表示方法對消息進行編碼
 * @author Administrator
 *
 */
public class VoteMsgBinCoder implements VoteMsgCoder {
	
	public static final int MIN_WIRE_LENGTH = 4;
	public static final int MAX_WIRE_LENGTH = 16;
	public static final int MAGIC = 0x5400;
	public static final int MAGIC_MASK = 0xfc00;
	public static final int MAGIC_SHIFT = 8;
	public static final int RESPONSE_FLAG = 0x0200;
	public static final int INQUIRY_FLAG = 0x0100;
	

	/*
	 * 將投票消息轉換爲一個字節序列(根據特定的協議)
	 * @see com.tcpip.VoteMsgCoder#toWire(com.tcpip.VoteMsg)
	 */
	@Override
	public byte[] toWire(VoteMsg msg) throws IOException {
		ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
		DataOutputStream out = new DataOutputStream(byteStream);
		
		short magicAndFlags = MAGIC;
		if(msg.isInquiry()){
			magicAndFlags |= INQUIRY_FLAG;
		}
		if(msg.isResponse()){
			magicAndFlags |= RESPONSE_FLAG;
		}
		out.writeShort(magicAndFlags);
		out.writeShort((short)msg.getCandidateID());
		if(msg.isResponse()){
			out.writeLong(msg.getVoteCount());
		}
		out.flush();
		byte[] data = byteStream.toByteArray();
		return data;
	}

	/*
	 * 對給定的字節序列進行解析(根據特定的協議)
	 * @see com.tcpip.VoteMsgCoder#fromWire(byte[])
	 */
	@Override
	public VoteMsg fromWire(byte[] input) throws IOException {
		if(input.length<MIN_WIRE_LENGTH){
			throw new IOException("Runt message");
		}
		ByteArrayInputStream bs = new ByteArrayInputStream(input);
		DataInputStream in = new DataInputStream(bs);
		
		int magic = in.readShort();
		if((magic&MAGIC_MASK) != MAGIC){
			throw new IOException("Bad Magic #:"+((magic&MAGIC_MASK)>>MAGIC_SHIFT));
		}
		boolean resp = ((magic&RESPONSE_FLAG)!=0);
		boolean inq = ((magic&INQUIRY_FLAG)!=0);
		int candidateID = in.readShort();
		if(candidateID<0 || candidateID>1000){
			throw new IOException("Bad candidate ID:"+candidateID);
		}
		long count = 0;
		if(resp){
			count = in.readLong();
			if(count<0){
				throw new IOException("Bad vote count: "+count);
			}
		}
		
		return new VoteMsg(resp,inq,candidateID,count);
	}

}

VoteService.java 

package com.tcpip;

import java.util.HashMap;
import java.util.Map;

/**
 * 投票服務
 * @author Administrator
 *
 */
public class VoteService {
	private Map<Integer, Long> results = new HashMap<Integer, Long>();
	
	/**
	 * 處理投票請求，返回投票的信息
	 * @param msg
	 * @return
	 */
	public VoteMsg handleRequest(VoteMsg msg){
		if(msg.isResponse()){
			return msg;
		}
		msg.setResponse(true);
		int candidateID = msg.getCandidateID();
		Long count = results.get(candidateID);
		if(count==null){
			count = 0L;
		}
		if(!msg.isInquiry()){
			results.put(candidateID, ++count);
		}
		msg.setVoteCount(count);
		return msg;
	}

}


tcp客戶端：ide

VoteClientTCP.java:
ui

package com.tcpip;

import java.io.OutputStream;
import java.net.Socket;


public class VoteClientTCP {
	
	public static final int CANDIDATEID = 888;
	public static void main(String args[]) throws Exception{
		
		args = new String[]{"127.0.0.1","9999"};
		
		String destAddr= args[0];
		int destPort = Integer.parseInt(args[1]);
		Socket sock = new Socket(destAddr,destPort);
		OutputStream out = sock.getOutputStream();
		
		VoteMsgCoder coder = new VoteMsgBinCoder();
		
		Framer framer = new LengthFramer(sock.getInputStream());
		VoteMsg msg = new VoteMsg(false, true, CANDIDATEID, 0);
		byte[] encodedMsg = coder.toWire(msg);
		
		System.out.println("Sending Inquiry ("+encodedMsg.length+" bytes):");
		System.out.println(msg);
		framer.frameMsg(encodedMsg, out);
		
		msg.setInquiry(false);
		encodedMsg = coder.toWire(msg);
		System.out.println("Sending Vote ("+encodedMsg.length+" bytes):");
		framer.frameMsg(encodedMsg, out);
		
		encodedMsg = framer.nextMsg();
		msg = coder.fromWire(encodedMsg);
		System.out.println("Received Response ("+encodedMsg.length+" bytes):");
		System.out.println(msg);
		
		msg = coder.fromWire(framer.nextMsg());
		System.out.println("Received Response ("+encodedMsg.length+" bytes):");
		System.out.println(msg);
		sock.close();
	}

}


tcp服務器端：VoteServerTCP.java:this

package com.tcpip;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

public class VoteServerTCP {

	public static void main(String[] args) throws IOException {

		args = new String[]{"9999"};
		
		int port = Integer.parseInt(args[0]);
		ServerSocket servSock = new ServerSocket(port);
		
		VoteMsgCoder coder = new VoteMsgBinCoder();
		VoteService service = new VoteService();
		
		while(true){
			Socket clntSock = servSock.accept();
			System.out.println("Handling client at "+clntSock.getRemoteSocketAddress());
			Framer framer = new LengthFramer(clntSock.getInputStream());
			try{
				byte[] req;
				while((req = framer.nextMsg())!=null){
					System.out.println("Received Message ("+req.length+" bytes)");
					VoteMsg responseMsg = service.handleRequest(coder.fromWire(req));
					framer.frameMsg(coder.toWire(responseMsg), clntSock.getOutputStream());
				}
			}catch(IOException ioe){
				System.out.println("Error handling client: "+ioe.getMessage());
			}finally{
				System.out.println("Closing connection");
				clntSock.close();
			}
		}
	}

}



udp客戶端：VoteClientUDP.java:編碼

package com.tcpip;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.util.Arrays;

public class VoteClientUDP {

	public static void main(String[] args) throws IOException{
		
		args = new String[]{"127.0.0.1","9999","111"};
		
		InetAddress destAddr = InetAddress.getByName(args[0]);
		int destPort = Integer.parseInt(args[1]);
		int candidate = Integer.parseInt(args[2]);
		
		DatagramSocket sock = new DatagramSocket();
		sock.connect(destAddr,destPort);
		
		VoteMsg vote = new VoteMsg(false,false,candidate,0);
		
		VoteMsgCoder coder = new VoteMsgTextCoder();
		
		byte[] encodedVote = coder.toWire(vote);
		System.out.println("Sending Text-Encoded Request ("+encodedVote.length+" bytes):");
		System.out.println(vote);
		
		DatagramPacket message = new DatagramPacket(encodedVote, encodedVote.length);
		
		sock.send(message);
		
		message = new DatagramPacket(new byte[VoteMsgTextCoder.MAX_WIRE_LENGTH],VoteMsgTextCoder.MAX_WIRE_LENGTH);
		sock.receive(message);
		encodedVote = Arrays.copyOfRange(message.getData(),0,message.getLength());
		
		System.out.println("Received Text-Encoded Response ("+encodedVote.length + " bytes):");
		vote = coder.fromWire(encodedVote);
		System.out.println(vote);
	}
}


udp服務端：VoteClientUDP.java:.net

package com.tcpip;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.util.Arrays;

public class VoteClientUDP {

	public static void main(String[] args) throws IOException{
		
		args = new String[]{"127.0.0.1","9999","111"};
		
		InetAddress destAddr = InetAddress.getByName(args[0]);
		int destPort = Integer.parseInt(args[1]);
		int candidate = Integer.parseInt(args[2]);
		
		DatagramSocket sock = new DatagramSocket();
		sock.connect(destAddr,destPort);
		
		VoteMsg vote = new VoteMsg(false,false,candidate,0);
		
		VoteMsgCoder coder = new VoteMsgTextCoder();
		
		byte[] encodedVote = coder.toWire(vote);
		System.out.println("Sending Text-Encoded Request ("+encodedVote.length+" bytes):");
		System.out.println(vote);
		
		DatagramPacket message = new DatagramPacket(encodedVote, encodedVote.length);
		
		sock.send(message);
		
		message = new DatagramPacket(new byte[VoteMsgTextCoder.MAX_WIRE_LENGTH],VoteMsgTextCoder.MAX_WIRE_LENGTH);
		sock.receive(message);
		encodedVote = Arrays.copyOfRange(message.getData(),0,message.getLength());
		
		System.out.println("Received Text-Encoded Response ("+encodedVote.length + " bytes):");
		vote = coder.fromWire(encodedVote);
		System.out.println(vote);
	}
}
相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程