initial dts support changes

Author: YunchuWang

client/build.gradle

@@ -34,6 +34,8 @@ dependencies {
     implementation "com.fasterxml.jackson.core:jackson-annotations:${jacksonVersion}"
     implementation "com.fasterxml.jackson.datatype:jackson-datatype-jsr310:${jacksonVersion}"
 
+    implementation 'com.azure:azure-identity:1.15.0'
+
     testImplementation(platform('org.junit:junit-bom:5.7.2'))
     testImplementation('org.junit.jupiter:junit-jupiter')
 }

client/src/main/java/com/microsoft/durabletask/AccessTokenCache.java

@@ -0,0 +1,32 @@
+package com.microsoft.durabletask;
+
+import com.azure.core.credential.TokenCredential;
+import com.azure.core.credential.AccessToken;
+import com.azure.core.credential.TokenRequestContext;
+
+import java.time.Duration;
+import java.time.OffsetDateTime;
+
+public final class AccessTokenCache {
+    private final TokenCredential credential;
+    private final TokenRequestContext context;
+    private final Duration margin;
+    private AccessToken cachedToken;
+
+    public AccessTokenCache(TokenCredential credential, TokenRequestContext context, Duration margin) {
+        this.credential = credential;
+        this.context = context;
+        this.margin = margin;
+    }
+
+    public AccessToken getToken() {
+        OffsetDateTime nowWithMargin = OffsetDateTime.now().plus(margin);
+
+        if (cachedToken == null 
+            || cachedToken.getExpiresAt().isBefore(nowWithMargin)) {
+            this.cachedToken = credential.getToken(context).block();
+        }
+
+        return cachedToken;
+    }
+}

client/src/main/java/com/microsoft/durabletask/DurableTaskGrpcWorker.java

@@ -140,6 +140,7 @@ public void startAndBlock() {
                                 .setInstanceId(orchestratorRequest.getInstanceId())
                                 .addAllActions(taskOrchestratorResult.getActions())
                                 .setCustomStatus(StringValue.of(taskOrchestratorResult.getCustomStatus()))
+                                .setCompletionToken(workItem.getCompletionToken())
                                 .build();
 
                         this.sidecarClient.completeOrchestratorTask(response);
@@ -164,7 +165,8 @@ public void startAndBlock() {
 
                         ActivityResponse.Builder responseBuilder = ActivityResponse.newBuilder()
                                 .setInstanceId(activityRequest.getOrchestrationInstance().getInstanceId())
-                                .setTaskId(activityRequest.getTaskId());
+                                .setTaskId(activityRequest.getTaskId())
+                                .setCompletionToken(workItem.getCompletionToken());
 
                         if (output != null) {
                             responseBuilder.setResult(StringValue.of(output));

client/src/main/java/com/microsoft/durabletask/OrchestrationRunner.java

@@ -140,6 +140,7 @@ public TaskOrchestration create() {
                 .setInstanceId(orchestratorRequest.getInstanceId())
                 .addAllActions(taskOrchestratorResult.getActions())
                 .setCustomStatus(StringValue.of(taskOrchestratorResult.getCustomStatus()))
+                .setCompletionToken(null)
                 .build();
         return response.toByteArray();
     }

samples/build.gradle

@@ -6,12 +6,27 @@ plugins {
 
 group 'io.durabletask'
 version = '0.1.0'
+def grpcVersion = '1.59.0'
 archivesBaseName = 'durabletask-samples'
 
+application {
+    mainClass = 'io.durabletask.samples.SpringDurableTaskSample'
+}
+
 dependencies {
     implementation project(':client')
 
     implementation 'org.springframework.boot:spring-boot-starter-web'
     implementation platform("org.springframework.boot:spring-boot-dependencies:2.5.2")
     implementation 'org.springframework.boot:spring-boot-starter'
+
+    // https://github.com/grpc/grpc-java#download
+    implementation "io.grpc:grpc-protobuf:${grpcVersion}"
+    implementation "io.grpc:grpc-stub:${grpcVersion}"
+    runtimeOnly "io.grpc:grpc-netty-shaded:${grpcVersion}"
+    implementation 'com.azure:azure-identity:1.15.0'
+
+    // install lombok
+    annotationProcessor 'org.projectlombok:lombok:1.18.22'
+    compileOnly 'org.projectlombok:lombok:1.18.22'
 }

samples/src/main/java/io/durabletask/samples/OrchestrationController.java

@@ -1,44 +1,44 @@
-// Copyright (c) Microsoft Corporation. All rights reserved.
-// Licensed under the MIT License.
-package io.durabletask.samples;
-
-import com.microsoft.durabletask.*;
-
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
-
-@RestController
-public class OrchestrationController {
-
-    final DurableTaskClient client;
-
-    public OrchestrationController() {
-        this.client = new DurableTaskGrpcClientBuilder().build();
-    }
-
-    @GetMapping("/hello")
-    public String greeting(@RequestParam(value = "name", defaultValue = "World") String name) {
-        return String.format("Hello, %s!", name);
-    }
-
-    @GetMapping("/placeOrder")
-    public NewOrderResponse placeOrder(@RequestParam(value = "item") String item) {
-        String instanceId = this.client.scheduleNewOrchestrationInstance(
-            "ProcessOrderOrchestration",
-            new NewOrchestrationInstanceOptions().setInput(item));
-        return new NewOrderResponse(instanceId);
-    }
-
-    private class NewOrderResponse {
-        private final String instanceId;
-
-        public NewOrderResponse(String instanceId) {
-            this.instanceId = instanceId;
-        }
-
-        public String getInstanceId() {
-            return this.instanceId;
-        }
-    }
-}
+// // Copyright (c) Microsoft Corporation. All rights reserved.
+// // Licensed under the MIT License.
+// package io.durabletask.samples;
+
+// import com.microsoft.durabletask.*;
+
+// import org.springframework.web.bind.annotation.GetMapping;
+// import org.springframework.web.bind.annotation.RequestParam;
+// import org.springframework.web.bind.annotation.RestController;
+
+// @RestController
+// public class OrchestrationController {
+
+//     final DurableTaskClient client;
+
+//     public OrchestrationController() {
+//         this.client = new DurableTaskGrpcClientBuilder().build();
+//     }
+
+//     @GetMapping("/hello")
+//     public String greeting(@RequestParam(value = "name", defaultValue = "World") String name) {
+//         return String.format("Hello, %s!", name);
+//     }
+
+//     @GetMapping("/placeOrder")
+//     public NewOrderResponse placeOrder(@RequestParam(value = "item") String item) {
+//         String instanceId = this.client.scheduleNewOrchestrationInstance(
+//             "ProcessOrderOrchestration",
+//             new NewOrchestrationInstanceOptions().setInput(item));
+//         return new NewOrderResponse(instanceId);
+//     }
+
+//     private class NewOrderResponse {
+//         private final String instanceId;
+
+//         public NewOrderResponse(String instanceId) {
+//             this.instanceId = instanceId;
+//         }
+
+//         public String getInstanceId() {
+//             return this.instanceId;
+//         }
+//     }
+// }

samples/src/main/java/io/durabletask/samples/ReadinessController.java

@@ -0,0 +1,15 @@
+package io.durabletask.samples;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/ready")
+public class ReadinessController {
+    @GetMapping
+    public ResponseEntity<String> readiness() {
+        return ResponseEntity.ok("Application is ready");
+    }
+}