1 files changed, 42 insertions, 0 deletions
diff --git a/container-disc/src/main/java/com/yahoo/container/jdisc/athenz/impl/AthenzService.java b/container-disc/src/main/java/com/yahoo/container/jdisc/athenz/impl/AthenzService.java
new file mode 100644
index 00000000000..69c52e5e639
--- /dev/null
+++ b/container-disc/src/main/java/com/yahoo/container/jdisc/athenz/impl/AthenzService.java
@@ -0,0 +1,42 @@
+package com.yahoo.container.jdisc.athenz.impl;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpUriRequest;
+import org.apache.http.client.methods.RequestBuilder;
+import org.apache.http.entity.ContentType;
+import org.apache.http.entity.StringEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.util.EntityUtils;
+import org.eclipse.jetty.http.HttpStatus;
+
+import java.io.IOException;
+
+/**
+ * @author mortent
+ */
+public class AthenzService {
+
+    /**
+     * Send instance register request to ZTS, get InstanceIdentity
+     */
+     public InstanceIdentity sendInstanceRegisterRequest(InstanceRegisterInformation instanceRegisterInformation, String athenzUrl) {
+        try(CloseableHttpClient client = HttpClientBuilder.create().build()) {
+            ObjectMapper objectMapper = new ObjectMapper();
+            HttpUriRequest postRequest = RequestBuilder.post()
+                    .setUri(athenzUrl + "/instance")
+                    .setEntity(new StringEntity(objectMapper.writeValueAsString(instanceRegisterInformation), ContentType.APPLICATION_JSON))
+                    .build();
+            CloseableHttpResponse response = client.execute(postRequest);
+            if(HttpStatus.isSuccess(response.getStatusLine().getStatusCode())) {
+                return objectMapper.readValue(response.getEntity().getContent(), InstanceIdentity.class);
+            } else {
+                String message = EntityUtils.toString(response.getEntity());
+                throw new RuntimeException(String.format("Unable to get identity. http code/message: %d/%s" + response.getStatusLine().getStatusCode(), message));
+            }
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+}