Integrating Protobuf with OpenSearch Transport

server/build.gradle

@@ -355,7 +355,13 @@ tasks.named("missingJavadoc").configure {
   javadocMissingIgnore = [
     "org.opensearch.extensions.proto.ExtensionRequestProto",
     "org.opensearch.extensions.proto.ExtensionRequestProto.ExtensionRequestOrBuilder",
-    "org.opensearch.extensions.proto"
+    "org.opensearch.extensions.proto",
+    "org.opensearch.tasks.proto.TaskResourceStatsProto",
+    "org.opensearch.tasks.proto.TaskResourceStatsProto.TaskResourceStatsOrBuilder",
+    "org.opensearch.tasks.proto.TaskResourceStatsProto.TaskResourceStats.TaskResourceUsageOrBuilder",
+    "org.opensearch.tasks.proto.TaskIdProto",
+    "org.opensearch.tasks.proto.TaskIdProto.TaskIdOrBuilder",
+    "org.opensearch.tasks.proto"
   ]
 }
 
@@ -373,6 +379,7 @@ tasks.named("licenseHeaders").configure {
     excludes << 'org/opensearch/client/documentation/placeholder.txt'
     // Ignore for protobuf generated code
     excludes << 'org/opensearch/extensions/proto/*'
+    excludes << 'org/opensearch/tasks/proto/*'
 }
 
 tasks.test {

server/src/main/java/org/opensearch/Build.java

@@ -32,6 +32,8 @@
 
 package org.opensearch;
 
+import com.google.protobuf.CodedInputStream;
+import com.google.protobuf.CodedOutputStream;
 import org.opensearch.common.Booleans;
 import org.opensearch.common.io.FileSystemUtils;
 import org.opensearch.common.io.stream.StreamInput;
@@ -218,6 +220,18 @@ public static Build readBuild(StreamInput in) throws IOException {
         return new Build(type, hash, date, snapshot, version, distribution);
     }
 
+    public static Build readBuildProtobuf(CodedInputStream in) throws IOException {
+        // the following is new for opensearch: we write the distribution to support any "forks"
+        final String distribution = in.readString();
+        // be lenient when reading on the wire, the enumeration values from other versions might be different than what we know
+        final Type type = Type.fromDisplayName(in.readString(), false);
+        String hash = in.readString();
+        String date = in.readString();
+        boolean snapshot = in.readBool();
+        final String version = in.readString();
+        return new Build(type, hash, date, snapshot, version, distribution);
+    }
+
     public static void writeBuild(Build build, StreamOutput out) throws IOException {
         // the following is new for opensearch: we write the distribution name to support any "forks" of the code
         out.writeString(build.distribution);
@@ -230,6 +244,18 @@ public static void writeBuild(Build build, StreamOutput out) throws IOException
         out.writeString(build.getQualifiedVersion());
     }
 
+    public static void writeBuildProtobuf(Build build, CodedOutputStream out) throws IOException {
+        // the following is new for opensearch: we write the distribution name to support any "forks" of the code
+        out.writeStringNoTag(build.distribution);
+
+        final Type buildType = build.type();
+        out.writeStringNoTag(buildType.displayName());
+        out.writeStringNoTag(build.hash());
+        out.writeStringNoTag(build.date());
+        out.writeBoolNoTag(build.isSnapshot());
+        out.writeStringNoTag(build.getQualifiedVersion());
+    }
+
     /**
      * Get the distribution name (expected to be OpenSearch; empty if legacy; something else if forked)
      * @return distribution name as a string