diff --git a/src/java/org/apache/cassandra/net/PingMessage.java b/src/java/org/apache/cassandra/net/PingMessage.java index 8eaf23ed93..d3a4322a30 100644 --- a/src/java/org/apache/cassandra/net/PingMessage.java +++ b/src/java/org/apache/cassandra/net/PingMessage.java @@ -38,20 +38,28 @@ public class PingMessage { public static IVersionedSerializer serializer = new PingMessageSerializer(); + private final int connectionType; + + public PingMessage(int connectionType) + { + this.connectionType = connectionType; + } + public static class PingMessageSerializer implements IVersionedSerializer { - public void serialize(PingMessage t, DataOutputPlus out, int version) + @Override + public void serialize(PingMessage t, DataOutputPlus out, int version) throws IOException { - throw new UnsupportedOperationException(); + out.writeByte(t.connectionType); } + @Override public PingMessage deserialize(DataInputPlus in, int version) throws IOException { - // throw away the one byte of the payload - in.readByte(); - return new PingMessage(); + return new PingMessage(in.readByte()); } + @Override public long serializedSize(PingMessage t, int version) { return 1;