Initial pcu support implementation

astra-db-java/pom.xml

@@ -71,6 +71,7 @@
         <dependency>
             <groupId>org.junit.platform</groupId>
             <artifactId>junit-platform-launcher</artifactId>
+            <version>${junit.platform.version}</version>
             <scope>test</scope>
         </dependency>
         <dependency>

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/AstraOpsClient.java

@@ -6,6 +6,7 @@
 import com.dtsx.astra.sdk.org.TokensClient;
 import com.dtsx.astra.sdk.org.UsersClient;
 import com.dtsx.astra.sdk.org.domain.*;
+import com.dtsx.astra.sdk.pcu.PcuGroupsClient;
 import com.dtsx.astra.sdk.streaming.AstraStreamingClient;
 import com.dtsx.astra.sdk.utils.ApiLocator;
 import com.dtsx.astra.sdk.utils.ApiResponseHttp;
@@ -166,4 +167,17 @@ public TokensClient tokens() {
         return new TokensClient(token, environment);
     }
 
+    // ------------------------------------------------------
+    //                 WORKING WITH PCU GROUPS
+    // ------------------------------------------------------
+
+    /**
+     * Work with PCU groups.
+     *
+     * @return
+     *      pcu groups client
+     */
+    public PcuGroupsClient pcuGroups() { // TODO `pcu()` or `pcuGroups()`?
+        return new PcuGroupsClient(token, environment);
+    }
 }

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/db/domain/AccessListAddress.java

@@ -7,7 +7,7 @@
 /**
  * Nested Address
  */
-public class AccessListAddress {
+public class    AccessListAddress {
 
     /** Address. */
     private String address;

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/pcu/PcuGroupDatacenterAssociationsClient.java

@@ -0,0 +1,118 @@
+package com.dtsx.astra.sdk.pcu;
+
+import com.dtsx.astra.sdk.pcu.domain.PcuGroupDatacenterAssociation;
+import com.dtsx.astra.sdk.pcu.exception.PcuGroupDbAssociationNotFound;
+import com.dtsx.astra.sdk.pcu.exception.PcuGroupNotFoundException;
+import com.dtsx.astra.sdk.AbstractApiClient;
+import com.dtsx.astra.sdk.utils.*;
+import com.fasterxml.jackson.core.type.TypeReference;
+import lombok.Getter;
+import lombok.NonNull;
+import lombok.extern.slf4j.Slf4j;
+import lombok.val;
+
+import java.util.List;
+import java.util.stream.Stream;
+
+@Slf4j
+public class PcuGroupDatacenterAssociationsClient extends AbstractApiClient {
+    private static final TypeReference<List<PcuGroupDatacenterAssociation>> PCU_GROUP_DB_ASSOCIATIONS =
+        new TypeReference<>() {};
+
+    @Getter
+    private final String pcuGroupId;
+
+    public PcuGroupDatacenterAssociationsClient(String token, String pcuGroupId) {
+        this(token, AstraEnvironment.PROD, pcuGroupId);
+    }
+
+    public PcuGroupDatacenterAssociationsClient(String token, AstraEnvironment env, String pcuGroupId) {
+        super(token, env);
+        this.pcuGroupId = pcuGroupId;
+    }
+
+    @Override
+    public String getServiceName() {
+        return "pcu.group.associations.datacenter";
+    }
+
+    // ---------------------------------
+    // ----        CRUD             ----
+    // ---------------------------------
+
+    public boolean exist(@NonNull String datacenterId) {
+        Assert.isDatacenterID(datacenterId, "datacenter id");
+
+        return findAll()
+            .anyMatch((assoc) -> assoc.getDatacenterUUID().equals(datacenterId));
+    }
+
+    public PcuGroupDatacenterAssociation findByDatacenterId(@NonNull String datacenterId) {
+        Assert.isDatacenterID(datacenterId, "datacenter id");
+
+        return findAll()
+            .filter((assoc) -> assoc.getDatacenterUUID().equals(datacenterId))
+            .findFirst()
+            .orElseThrow(() -> new PcuGroupDbAssociationNotFound(pcuGroupId, datacenterId));
+    }
+
+    public Stream<PcuGroupDatacenterAssociation> findAll() {
+        val res = GET(getEndpointPcuAssociations() + "/" + pcuGroupId, getOperationName("findAll"));
+
+        return unmarshallOrThrow(res, PCU_GROUP_DB_ASSOCIATIONS, "get pcu group db associations").stream();
+    }
+
+    public PcuGroupDatacenterAssociation associate(@NonNull String datacenterId) {
+        Assert.isDatacenterID(datacenterId, "datacenter id");
+
+        val res = POST(getEndpointPcuAssociations() + "/" + pcuGroupId + "/" + datacenterId, getOperationName("associate"));
+
+        return unmarshallOrThrow(res, new TypeReference<List<PcuGroupDatacenterAssociation>>() {}, "associate db to pcu group").get(0);
+    }
+
+    private record TransferReqBody(String fromPCUGroupUUID, String toPCUGroupUUID, String datacenterUUID) {}
+
+    public PcuGroupDatacenterAssociation transfer(@NonNull String toPcuGroup, @NonNull String datacenterId) {
+        Assert.isUUID(toPcuGroup, "target pcu group id");
+        Assert.isDatacenterID(datacenterId, "datacenter id");
+
+        val reqBody = JsonUtils.marshall(new TransferReqBody(this.pcuGroupId, toPcuGroup, datacenterId));
+        val res = POST(getEndpointPcuAssociations() + "/transfer/" + pcuGroupId, reqBody, getOperationName("transfer"));
+
+        return unmarshallOrThrow(res, new TypeReference<List<PcuGroupDatacenterAssociation>>() {}, "transfer db to pcu group").get(0);
+    }
+
+    public void dissociate(@NonNull String datacenterId) {
+        Assert.isDatacenterID(datacenterId, "datacenter id");
+        DELETE(getEndpointPcuAssociations() + "/" + pcuGroupId + "/" + datacenterId, getOperationName("dissociate"));
+    }
+
+    // ---------------------------------
+    // ----       Utilities         ----
+    // ---------------------------------
+
+    public String getEndpointPcuAssociations() {
+        return ApiLocator.getApiDevopsEndpoint(environment) + "/pcus/association";
+    }
+
+    private <T> T unmarshallOrThrow(ApiResponseHttp res, TypeReference<T> clazz, String operation) {
+        try {
+            System.out.println(res.getBody());
+            return JsonUtils.unmarshallType(res.getBody(), clazz);
+        } catch (Exception e) {
+            ApiResponseError responseError = null;
+
+            try {
+                responseError = JsonUtils.unmarshallBean(res.getBody(), ApiResponseError.class);
+            } catch (Exception ignored) {}
+
+            if (responseError != null && responseError.getErrors() != null && !responseError.getErrors().isEmpty()) {
+                if (responseError.getErrors().get(0).getId() == 2000367) {
+                    throw PcuGroupNotFoundException.forId(pcuGroupId);
+                }
+            }
+
+            throw new IllegalStateException("Expected code 2xx to " + operation + " but got " + res.getCode() + "body=" + res.getBody());
+        }
+    }
+}

astra-sdk-devops/src/main/java/com/dtsx/astra/sdk/pcu/PcuGroupOpsClient.java

@@ -0,0 +1,112 @@
+package com.dtsx.astra.sdk.pcu;
+
+import com.dtsx.astra.sdk.pcu.domain.PcuGroup;
+import com.dtsx.astra.sdk.pcu.domain.PcuGroupStatusType;
+import com.dtsx.astra.sdk.pcu.domain.PcuGroupUpdateRequest;
+import com.dtsx.astra.sdk.pcu.exception.PcuGroupNotFoundException;
+import com.dtsx.astra.sdk.AbstractApiClient;
+import com.dtsx.astra.sdk.utils.ApiLocator;
+import com.dtsx.astra.sdk.utils.AstraEnvironment;
+import com.dtsx.astra.sdk.utils.JsonUtils;
+import lombok.Getter;
+import lombok.extern.slf4j.Slf4j;
+import lombok.val;
+
+import java.util.List;
+import java.util.Optional;
+
+@Slf4j
+public class PcuGroupOpsClient extends AbstractApiClient {
+    @Getter
+    private final String pcuGroupId;
+
+    public PcuGroupOpsClient(String token, String pcuGroupId) {
+        this(token, AstraEnvironment.PROD, pcuGroupId);
+    }
+
+    public PcuGroupOpsClient(String token, AstraEnvironment env, String pcuGroupId) {
+        super(token, env);
+        this.pcuGroupId = pcuGroupId;
+    }
+
+    @Override
+    public String getServiceName() {
+        return "pcu.group";
+    }
+
+    // ---------------------------------
+    // ----       READ              ----
+    // ---------------------------------
+
+    public Optional<PcuGroup> find() {
+        try {
+            return Optional.of(get());
+        } catch (PcuGroupNotFoundException e) {
+            return Optional.empty();
+        }
+    }
+
+    public PcuGroup get() {
+        return new PcuGroupsClient(token, environment).findById(pcuGroupId).orElseThrow(() -> PcuGroupNotFoundException.forId(pcuGroupId));
+    }
+
+    public boolean exist() {
+        return find().isPresent();
+    }
+
+    public boolean isActive() {
+        return PcuGroupStatusType.ACTIVE == get().getStatus();
+    }
+
+    public boolean isCreatedOrActive() {
+        return PcuGroupStatusType.CREATED == get().getStatus() || isActive();
+    }
+
+    // ---------------------------------
+    // ----       UPDATE            ----
+    // ---------------------------------
+
+    public void update(PcuGroupUpdateRequest req) {
+        val base = get();
+        PUT(getEndpointPcus(), JsonUtils.marshall(List.of(req.withDefaultsAndValidations(base))), getOperationName("update"));
+    }
+
+    // ---------------------------------
+    // ----       MAINTENANCE       ----
+    // ---------------------------------
+
+    public void park() {
+        val res = POST(getEndpointPcus() + "/park/" + pcuGroupId, getOperationName("park"));
+
+        if (res.getCode() >= 300) {
+            throw new IllegalStateException("Expected code 200 to park pcu group but got " + res.getCode() + "body=" + res.getBody());
+        }
+    }
+
+    public void unpark() {
+        val res = POST(getEndpointPcus() + "/unpark/" + pcuGroupId, getOperationName("unpark"));
+
+        if (res.getCode() >= 300) {
+            throw new IllegalStateException("Expected code 200 to unpark pcu group but got " + res.getCode() + "body=" + res.getBody());
+        }
+    }
+
+    public void delete() {
+        if (!exist()) {
+            throw PcuGroupNotFoundException.forId(pcuGroupId);
+        }
+        DELETE(getEndpointPcus() + "/" + pcuGroupId, getOperationName("delete"));
+    }
+
+    // ---------------------------------
+    // ----       Utilities         ----
+    // ---------------------------------
+
+    public PcuGroupDatacenterAssociationsClient datacenterAssociations() {
+        return new PcuGroupDatacenterAssociationsClient(token, environment, pcuGroupId);
+    }
+
+    public String getEndpointPcus() {
+        return ApiLocator.getApiDevopsEndpoint(environment) + "/pcus";
+    }
+}