Update provider endpoints

2 files changed, 14 insertions, 0 deletions

diff --git a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/DefaultZmsClient.java b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/DefaultZmsClient.java
index d8d6d2c3486..6d19b102745 100644
--- a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/DefaultZmsClient.java
+++ b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/DefaultZmsClient.java
@@ -413,6 +413,18 @@ public class DefaultZmsClient extends ClientBase implements ZmsClient {
     }
 
     @Override
+    public void updateProviderEndpoint(AthenzService athenzService, String endpoint) {
+        URI uri = zmsUrl.resolve(String.format("domain/%s/service/%s/meta/system/providerendpoint",
+                                               athenzService.getDomainName(), athenzService.getName()));
+
+        String payload = String.format("{\"providerEndpoint\": \"%s\"}", endpoint);
+        HttpUriRequest request = RequestBuilder.put(uri)
+                .setEntity(new StringEntity(payload, ContentType.APPLICATION_JSON))
+                .build();
+        execute(request, response -> readEntity(response, Void.class));
+    }
+
+    @Override
     public void deleteService(AthenzService athenzService) {
         URI uri = zmsUrl.resolve(String.format("domain/%s/service/%s", athenzService.getDomainName(), athenzService.getName()));
         execute(RequestBuilder.delete(uri).build(), response -> readEntity(response, Void.class));
diff --git a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/ZmsClient.java b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/ZmsClient.java
index 541aa4d9c60..edff7b517c4 100644
--- a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/ZmsClient.java
+++ b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/client/zms/ZmsClient.java
@@ -80,6 +80,8 @@ public interface ZmsClient extends Closeable {
 
     void updateServicePublicKey(AthenzService athenzService, String publicKeyId, PublicKey publicKey);
 
+    void updateProviderEndpoint(AthenzService athenzService, String endpoint);
+
     void deleteService(AthenzService athenzService);
 
     void createRole(AthenzRole role, Map<String, Object> properties);