Skip to content

Commit 109b145

Browse files
committed
Support for vector
1 parent 0048a58 commit 109b145

File tree

3 files changed

+28
-7
lines changed

3 files changed

+28
-7
lines changed

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

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -148,9 +148,8 @@ public Optional<Database> findById(String id) {
148148

149149
/**
150150
* Find Databases matching the provided filter.
151-
*
152-
* https://docs.datastax.com/en/astra/docs/_attachments/devopsv1.html#operation/listDatabases
153-
*
151+
* <a href="https://docs.datastax.com/en/astra/docs/_attachments/devopsv1.html#operation/listDatabases">Reference Documentation</a>
152+
*
154153
* @param filter
155154
* filter to search for db
156155
* @return
@@ -164,13 +163,13 @@ public Stream<Database> search(DatabaseFilter filter) {
164163

165164
/**
166165
* Create a database base on some parameters.
167-
*
166+
*
168167
* @param dbCreationRequest
169168
* creation request with tier and capacity unit
170169
* @return
171170
* the new instance id.
172-
*
173-
* https://docs.datastax.com/en/astra/docs/_attachments/devopsv1.html#operation/createDatabase
171+
*
172+
* <a href="https://docs.datastax.com/en/astra/docs/_attachments/devopsv1.html#operation/createDatabase">Reference Documentation</a>
174173
*/
175174
public String create(DatabaseCreationRequest dbCreationRequest) {
176175
Assert.notNull(dbCreationRequest, "Database creation request");

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

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,8 @@ public class DatabaseInfo {
3333

3434
/** Default keyspaces. */
3535
private String keyspace;
36+
37+
private String dbType;
3638

3739
/** Keyspace name in database. */
3840
private Set<String> keyspaces;
@@ -252,4 +254,23 @@ public String getEngineType() {
252254
public void setEngineType(String engineType) {
253255
this.engineType = engineType;
254256
}
257+
258+
/**
259+
* Gets dbType
260+
*
261+
* @return value of dbType
262+
*/
263+
public String getDbType() {
264+
return dbType;
265+
}
266+
267+
/**
268+
* Set value for dbType
269+
*
270+
* @param dbType
271+
* new value for dbType
272+
*/
273+
public void setDbType(String dbType) {
274+
this.dbType = dbType;
275+
}
255276
}

astra-sdk-devops/src/test/java/com/dtsx/astra/sdk/db/DatabasesClientTest.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ public void shouldCreateServerlessDb() {
5757
Assertions.assertEquals(DatabaseStatusType.ACTIVE, db.getStatus());
5858
}
5959

60+
6061
@Test
6162
@Order(3)
6263
@DisplayName("findAll() retrieve some data")
@@ -112,7 +113,7 @@ public void shouldCreateServerlessWithVector() {
112113
Database finalDb = getDatabasesClient().database(dbId).get();
113114
Assertions.assertEquals(DatabaseStatusType.ACTIVE, finalDb.getStatus());
114115
// Engine Type is not populated in details, only in findAll()
115-
//Assertions.assertEquals("cc", finalDb.getInfo().getEngineType());
116+
Assertions.assertEquals("vector", finalDb.getInfo().getDbType());
116117
}
117118

118119
@Test

0 commit comments

Comments
 (0)