Added capability and workerstatusresponse.

Author: yojagad

src/main/java/com/microsoft/azure/functions/worker/JavaWorkerClient.java

@@ -41,6 +41,7 @@ private void addHandlers() {
         this.handlerSuppliers.put(StreamingMessage.ContentCase.FUNCTION_ENVIRONMENT_RELOAD_REQUEST, () -> new FunctionEnvironmentReloadRequestHandler(broker));
         this.handlerSuppliers.put(StreamingMessage.ContentCase.FUNCTION_LOAD_REQUEST, () -> new FunctionLoadRequestHandler(broker));
         this.handlerSuppliers.put(StreamingMessage.ContentCase.INVOCATION_REQUEST, () -> new InvocationRequestHandler(broker));
+        this.handlerSuppliers.put(StreamingMessage.ContentCase.WORKER_STATUS_REQUEST, WorkerStatusRequestHandler::new);
     }
 
     public Future<Void> listen(String workerId, String requestId) {

src/main/java/com/microsoft/azure/functions/worker/handler/WorkerInitRequestHandler.java

@@ -15,6 +15,7 @@ public WorkerInitRequestHandler() {
     String execute(WorkerInitRequest request, WorkerInitResponse.Builder response) {
         response.setWorkerVersion(Application.version());
         response.putCapabilities("TypedDataCollection", "TypedDataCollection");
+        response.putCapabilities("WorkerStatus", "WorkerStatus");
         return "Worker initialized";
     }
 }

src/main/java/com/microsoft/azure/functions/worker/handler/WorkerStatusRequestHandler.java

@@ -0,0 +1,17 @@
+package com.microsoft.azure.functions.worker.handler;
+import com.microsoft.azure.functions.rpc.messages.*;
+
+public class WorkerStatusRequestHandler extends MessageHandler<WorkerStatusRequest, WorkerStatusResponse.Builder> {
+
+    public WorkerStatusRequestHandler() {
+        super(StreamingMessage::getWorkerStatusRequest,
+              WorkerStatusResponse::newBuilder,
+              null,
+              StreamingMessage.Builder::setWorkerStatusResponse);
+    }
+
+    @Override
+    String execute(WorkerStatusRequest request, WorkerStatusResponse.Builder response) {
+        return String.format("WorkerStatusRequest completed");
+    }
+}