Mercurial > hg > Database > Alice
view src/main/java/alice/daemon/IncomingTcpConnection.java @ 625:4835788cbb7e dispose
fix TestRemote default arguments
author | Shinji KONO <kono@ie.u-ryukyu.ac.jp> |
---|---|
date | Tue, 22 Aug 2017 15:58:18 +0900 |
parents | ea21af9a4762 |
children | f5365a5e6672 |
line wrap: on
line source
package alice.daemon; import java.io.EOFException; import java.io.IOException; import java.nio.channels.ClosedChannelException; import alice.datasegment.*; import org.msgpack.MessagePack; import org.msgpack.unpacker.Unpacker; import alice.topology.manager.keeparive.RespondData; public class IncomingTcpConnection extends Thread { private Connection connection; protected DataSegmentManager manager; protected String reverseKey; private LocalDataSegmentManager lmanager = DataSegment.getLocal(); private CompressedLocalDataSegmentManager compressedlmanager = DataSegment.getCompressedLocal(); private static final MessagePack packer = new MessagePack(); public IncomingTcpConnection(DataSegmentManager manager) { this.manager = manager; } public IncomingTcpConnection(Connection connection, DataSegmentManager manager, String reverseKey) { this.manager = manager; this.connection = connection; this.reverseKey = reverseKey; } public LocalDataSegmentManager getLocalDataSegmentManager(){ return lmanager; } public CompressedLocalDataSegmentManager getCompressedLocalDataSegmentManager(){ return compressedlmanager; } /** * pipeline thread for receiving */ public void run() { Unpacker unpacker = null; try { unpacker = packer.createUnpacker(connection.socket.getInputStream()); } catch (IOException e) { e.printStackTrace(); } if (unpacker == null) { return; } while (true) { try { Command cmd = null; ReceiveData rData = null; CommandMessage msg = unpacker.read(CommandMessage.class);///read header CommandType type = CommandType.getCommandTypeFromId(msg.type); switch (type) { case UPDATE: case PUT: int dataSize = unpacker.readInt(); rData = new ReceiveData(unpacker.getSerializedByteArray(dataSize), msg.compressed, msg.dataSize); /* rData = new ReceiveData(new byte[dataSize], msg.compressed, msg.dataSize); try { unpacker.readPayload(rData.getMessagePack()); } catch ( ); */ if (msg.setTime) { rData.setTimes(msg.time, true, msg.depth); } if (msg.setZepped){ rData.setZipped(msg.zippedDataSize, true); } else { rData.setZipped(dataSize, true); } cmd = new Command(type, null, null, rData, 0, 0, null, null, reverseKey); cmd.setCompressFlag(msg.compressed); if (rData.compressed()){ compressedlmanager.getDataSegmentKey(msg.key).runCommand(cmd); } else { lmanager.getDataSegmentKey(msg.key).runCommand(cmd); } break; case PEEK: case TAKE: cmd = new Command(type, null, null, null, msg.index, msg.seq, null, null, connection); cmd.setCompressFlag(msg.compressed); if (msg.compressed){ compressedlmanager.getDataSegmentKey(msg.key).runCommand(cmd); } else { lmanager.getDataSegmentKey(msg.key).runCommand(cmd); } break; case REMOVE: cmd = new Command(type, null, null, null, 0, 0, null, null, ""); lmanager.getDataSegmentKey(msg.key).runCommand(cmd);//ToDo:fix break; case REPLY: cmd = manager.getAndRemoveCmd(msg.seq); rData = new ReceiveData(unpacker.getSerializedByteArray(unpacker.readInt()), msg.compressed, msg.dataSize); Command rCmd = new Command(type, null, null, rData, msg.index, msg.seq, null, null, ""); cmd.setCompressFlag(msg.compressed); cmd.cs.ids.reply(cmd.receiver, rCmd); break; case PING: if (DataSegment.contains(reverseKey)) DataSegment.get(reverseKey).response(msg.key); break; case RESPONSE: rData = new ReceiveData(new RespondData(reverseKey, System.currentTimeMillis())); DataSegment.getLocal().put(msg.key, rData, false); break; default: break; } } catch (ClosedChannelException e) { return; } catch (EOFException e) { return; } catch (IOException e) { return; } } } public void setReverseKey(String name) { this.reverseKey = name; setName(name+"-IncomingTcp"); } }