Skip to content

Commit 21dcfaa

Browse files
committed
feat(backup): add restore options for users/roles
1 parent aab8e4c commit 21dcfaa

2 files changed

Lines changed: 24 additions & 6 deletions

File tree

src/main/java/io/weaviate/client/v1/backup/api/BackupRestorer.java

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,19 @@
11
package io.weaviate.client.v1.backup.api;
22

33
import com.google.gson.annotations.SerializedName;
4-
import io.weaviate.client.v1.backup.model.BackupRestoreResponse;
5-
import io.weaviate.client.v1.backup.model.BackupRestoreStatusResponse;
6-
import io.weaviate.client.v1.backup.model.RestoreStatus;
7-
import lombok.Builder;
8-
import lombok.Getter;
4+
95
import io.weaviate.client.Config;
106
import io.weaviate.client.base.BaseClient;
117
import io.weaviate.client.base.ClientResult;
128
import io.weaviate.client.base.Response;
139
import io.weaviate.client.base.Result;
1410
import io.weaviate.client.base.http.HttpClient;
11+
import io.weaviate.client.v1.backup.model.BackupRestoreResponse;
12+
import io.weaviate.client.v1.backup.model.BackupRestoreStatusResponse;
13+
import io.weaviate.client.v1.backup.model.RbacRestoreOption;
14+
import io.weaviate.client.v1.backup.model.RestoreStatus;
15+
import lombok.Builder;
16+
import lombok.Getter;
1517

1618
public class BackupRestorer extends BaseClient<BackupRestoreResponse> implements ClientResult<BackupRestoreResponse> {
1719

@@ -68,7 +70,6 @@ public BackupRestorer withWaitForCompletion(boolean waitForCompletion) {
6870
@Override
6971
public Result<BackupRestoreResponse> run() {
7072
BackupRestore payload = BackupRestore.builder()
71-
.config(BackupRestoreConfig.builder().build())
7273
.include(includeClassNames)
7374
.exclude(excludeClassNames)
7475
.config(config)
@@ -139,8 +140,11 @@ private Result<BackupRestoreResponse> merge(Response<BackupRestoreStatusResponse
139140
@Getter
140141
@Builder
141142
private static class BackupRestore {
143+
@SerializedName("config")
142144
BackupRestoreConfig config;
145+
@SerializedName("include")
143146
String[] include;
147+
@SerializedName("exclude")
144148
String[] exclude;
145149
}
146150

@@ -153,5 +157,9 @@ public static class BackupRestoreConfig {
153157
String bucket;
154158
@SerializedName("Path")
155159
String path;
160+
@SerializedName("usersOptions")
161+
RbacRestoreOption usersRestore;
162+
@SerializedName("rolesOptions")
163+
RbacRestoreOption rolesRestore;
156164
}
157165
}
Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
package io.weaviate.client.v1.backup.model;
2+
3+
import com.google.gson.annotations.SerializedName;
4+
5+
public enum RbacRestoreOption {
6+
@SerializedName("noRestore")
7+
NO_RESTORE,
8+
@SerializedName("all")
9+
ALL;
10+
}

0 commit comments

Comments
 (0)