56470a4569
* Java Docs: use instanceof with name binding * remove changes in generated code * Update docs/src/test/java/jdocs/ddata/protobuf/TwoPhaseSetSerializer2.java Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * merge issue --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
91 lines
2.8 KiB
Java
91 lines
2.8 KiB
Java
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
* license agreements; and to You under the Apache License, version 2.0:
|
|
*
|
|
* https://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* This file is part of the Apache Pekko project, which was derived from Akka.
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2015-2022 Lightbend Inc. <https://www.lightbend.com>
|
|
*/
|
|
|
|
package jdocs.ddata.protobuf;
|
|
|
|
// #serializer
|
|
import docs.ddata.protobuf.msg.TwoPhaseSetMessages;
|
|
import docs.ddata.protobuf.msg.TwoPhaseSetMessages.TwoPhaseSet2.Builder;
|
|
import jdocs.ddata.TwoPhaseSet;
|
|
import org.apache.pekko.actor.ExtendedActorSystem;
|
|
import org.apache.pekko.cluster.ddata.GSet;
|
|
import org.apache.pekko.cluster.ddata.protobuf.AbstractSerializationSupport;
|
|
import org.apache.pekko.cluster.ddata.protobuf.ReplicatedDataSerializer;
|
|
|
|
public class TwoPhaseSetSerializer2 extends AbstractSerializationSupport {
|
|
|
|
private final ExtendedActorSystem system;
|
|
private final ReplicatedDataSerializer replicatedDataSerializer;
|
|
|
|
public TwoPhaseSetSerializer2(ExtendedActorSystem system) {
|
|
this.system = system;
|
|
this.replicatedDataSerializer = new ReplicatedDataSerializer(system);
|
|
}
|
|
|
|
@Override
|
|
public ExtendedActorSystem system() {
|
|
return this.system;
|
|
}
|
|
|
|
@Override
|
|
public boolean includeManifest() {
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public int identifier() {
|
|
return 99998;
|
|
}
|
|
|
|
@Override
|
|
public byte[] toBinary(Object obj) {
|
|
if (obj instanceof TwoPhaseSet tps) {
|
|
return twoPhaseSetToProto(tps).toByteArray();
|
|
} else {
|
|
throw new IllegalArgumentException("Can't serialize object of type " + obj.getClass());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Object fromBinaryJava(byte[] bytes, Class<?> manifest) {
|
|
return twoPhaseSetFromBinary(bytes);
|
|
}
|
|
|
|
protected TwoPhaseSetMessages.TwoPhaseSet2 twoPhaseSetToProto(TwoPhaseSet twoPhaseSet) {
|
|
Builder b = TwoPhaseSetMessages.TwoPhaseSet2.newBuilder();
|
|
if (!twoPhaseSet.adds.isEmpty())
|
|
b.setAdds(otherMessageToProto(twoPhaseSet.adds).toByteString());
|
|
if (!twoPhaseSet.removals.isEmpty())
|
|
b.setRemovals(otherMessageToProto(twoPhaseSet.removals).toByteString());
|
|
return b.build();
|
|
}
|
|
|
|
@SuppressWarnings("unchecked")
|
|
protected TwoPhaseSet twoPhaseSetFromBinary(byte[] bytes) {
|
|
try {
|
|
TwoPhaseSetMessages.TwoPhaseSet2 msg = TwoPhaseSetMessages.TwoPhaseSet2.parseFrom(bytes);
|
|
|
|
GSet<String> adds = GSet.create();
|
|
if (msg.hasAdds()) adds = (GSet<String>) otherMessageFromBinary(msg.getAdds().toByteArray());
|
|
|
|
GSet<String> removals = GSet.create();
|
|
if (msg.hasRemovals())
|
|
adds = (GSet<String>) otherMessageFromBinary(msg.getRemovals().toByteArray());
|
|
|
|
return new TwoPhaseSet(adds, removals);
|
|
} catch (Exception e) {
|
|
throw new RuntimeException(e.getMessage(), e);
|
|
}
|
|
}
|
|
}
|
|
// #serializer
|