Add getter for role token domain

1 files changed, 4 insertions, 1 deletions

diff --git a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/api/ZToken.java b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/api/ZToken.java
index 36c06132532..1691704afd6 100644
--- a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/api/ZToken.java
+++ b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/api/ZToken.java
@@ -6,7 +6,6 @@ import com.yahoo.vespa.athenz.utils.AthenzIdentities;
 
 import java.util.List;
 import java.util.Objects;
-import java.util.stream.Collectors;
 
 import static java.util.stream.Collectors.toList;
 
@@ -31,6 +30,10 @@ public class ZToken {
         return AthenzIdentities.from(token.getPrincipal());
     }
 
+    public AthenzDomain getDomain() {
+        return new AthenzDomain(token.getDomain());
+    }
+
     public List<AthenzRole> getRoles() {
         String domain = token.getDomain();
         return token.getRoles().stream()