|
@@ -41,7 +41,7 @@ import org.apache.hadoop.yarn.proto.YarnProtos.PlacementConstraintTargetProto;
|
|
|
import org.apache.hadoop.yarn.proto.YarnProtos.SimplePlacementConstraintProto;
|
|
|
import org.apache.hadoop.yarn.proto.YarnProtos.TimedPlacementConstraintProto;
|
|
|
|
|
|
-import com.google.protobuf.GeneratedMessage;
|
|
|
+import com.google.protobuf.GeneratedMessageV3;
|
|
|
|
|
|
/**
|
|
|
* {@code PlacementConstraintToProtoConverter} generates a
|
|
@@ -50,7 +50,7 @@ import com.google.protobuf.GeneratedMessage;
|
|
|
*/
|
|
|
@Private
|
|
|
public class PlacementConstraintToProtoConverter
|
|
|
- implements PlacementConstraint.Visitor<GeneratedMessage> {
|
|
|
+ implements PlacementConstraint.Visitor<GeneratedMessageV3> {
|
|
|
|
|
|
private PlacementConstraint placementConstraint;
|
|
|
|
|
@@ -65,7 +65,7 @@ public class PlacementConstraintToProtoConverter
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(SingleConstraint constraint) {
|
|
|
+ public GeneratedMessageV3 visit(SingleConstraint constraint) {
|
|
|
SimplePlacementConstraintProto.Builder sb =
|
|
|
SimplePlacementConstraintProto.newBuilder();
|
|
|
|
|
@@ -94,7 +94,7 @@ public class PlacementConstraintToProtoConverter
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(TargetExpression target) {
|
|
|
+ public GeneratedMessageV3 visit(TargetExpression target) {
|
|
|
PlacementConstraintTargetProto.Builder tb =
|
|
|
PlacementConstraintTargetProto.newBuilder();
|
|
|
|
|
@@ -109,16 +109,16 @@ public class PlacementConstraintToProtoConverter
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(TargetConstraint constraint) {
|
|
|
+ public GeneratedMessageV3 visit(TargetConstraint constraint) {
|
|
|
throw new YarnRuntimeException("Unexpected TargetConstraint found.");
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(CardinalityConstraint constraint) {
|
|
|
+ public GeneratedMessageV3 visit(CardinalityConstraint constraint) {
|
|
|
throw new YarnRuntimeException("Unexpected CardinalityConstraint found.");
|
|
|
}
|
|
|
|
|
|
- private GeneratedMessage visitAndOr(
|
|
|
+ private GeneratedMessageV3 visitAndOr(
|
|
|
CompositeConstraint<AbstractConstraint> composite, CompositeType type) {
|
|
|
CompositePlacementConstraintProto.Builder cb =
|
|
|
CompositePlacementConstraintProto.newBuilder();
|
|
@@ -137,17 +137,17 @@ public class PlacementConstraintToProtoConverter
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(And constraint) {
|
|
|
+ public GeneratedMessageV3 visit(And constraint) {
|
|
|
return visitAndOr(constraint, CompositeType.AND);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(Or constraint) {
|
|
|
+ public GeneratedMessageV3 visit(Or constraint) {
|
|
|
return visitAndOr(constraint, CompositeType.OR);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(DelayedOr constraint) {
|
|
|
+ public GeneratedMessageV3 visit(DelayedOr constraint) {
|
|
|
CompositePlacementConstraintProto.Builder cb =
|
|
|
CompositePlacementConstraintProto.newBuilder();
|
|
|
|
|
@@ -166,7 +166,7 @@ public class PlacementConstraintToProtoConverter
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public GeneratedMessage visit(TimedPlacementConstraint constraint) {
|
|
|
+ public GeneratedMessageV3 visit(TimedPlacementConstraint constraint) {
|
|
|
TimedPlacementConstraintProto.Builder tb =
|
|
|
TimedPlacementConstraintProto.newBuilder();
|
|
|
|