Query multiple tables using one entity

ydb-platform · Dec 27, 2024 · 5816833 · 5816833
1 parent 286b5b9
commit 5816833
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 10 deletions.
diff --git a/databind/src/main/java/tech/ydb/yoj/ExperimentalApi.java b/databind/src/main/java/tech/ydb/yoj/ExperimentalApi.java
@@ -4,6 +4,7 @@
 import java.lang.annotation.Target;
 
 import static java.lang.annotation.ElementType.ANNOTATION_TYPE;
+import static java.lang.annotation.ElementType.CONSTRUCTOR;
 import static java.lang.annotation.ElementType.FIELD;
 import static java.lang.annotation.ElementType.METHOD;
 import static java.lang.annotation.ElementType.PARAMETER;
@@ -14,7 +15,7 @@
  * Annotates <em>experimental features</em>. These features are not part of the stable YOJ API: they can change incompatibly,
  * or even disappear entirely <em>in any release</em>.
  */
-@Target({TYPE, FIELD, METHOD, PARAMETER, ANNOTATION_TYPE})
+@Target({TYPE, FIELD, METHOD, PARAMETER, CONSTRUCTOR, ANNOTATION_TYPE})
 @Retention(SOURCE)
 public @interface ExperimentalApi {
     /**

diff --git a/...rc/main/java/tech/ydb/yoj/repository/ydb/compatibility/YdbSchemaCompatibilityChecker.java b/...rc/main/java/tech/ydb/yoj/repository/ydb/compatibility/YdbSchemaCompatibilityChecker.java
@@ -5,7 +5,6 @@
 import lombok.AccessLevel;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
-import lombok.SneakyThrows;
 import lombok.Value;
 import lombok.With;
 import org.apache.commons.text.StringEscapeUtils;
@@ -16,6 +15,7 @@
 import tech.ydb.yoj.databind.DbType;
 import tech.ydb.yoj.repository.db.Entity;
 import tech.ydb.yoj.repository.db.EntitySchema;
+import tech.ydb.yoj.repository.db.TableDescriptor;
 import tech.ydb.yoj.repository.ydb.YdbConfig;
 import tech.ydb.yoj.repository.ydb.YdbRepository;
 import tech.ydb.yoj.repository.ydb.client.YdbPaths;
@@ -42,7 +42,7 @@
 public final class YdbSchemaCompatibilityChecker {
     private static final Logger log = LoggerFactory.getLogger(YdbSchemaCompatibilityChecker.class);
 
-    private final List<Class<? extends Entity>> entities;
+    private final List<TableDescriptor<?>> descriptors;
     private final Config config;
     private final YdbRepository repository;
     private final YdbConfig repositoryConfig;
@@ -51,12 +51,20 @@ public final class YdbSchemaCompatibilityChecker {
     private final List<String> canExecuteMessages = new ArrayList<>();
     private final List<String> incompatibleMessages = new ArrayList<>();
 
+    public YdbSchemaCompatibilityChecker(YdbRepository repository, List<TableDescriptor<?>> descriptors) {
+        this(repository, descriptors, Config.DEFAULT);
+    }
+
     public YdbSchemaCompatibilityChecker(List<Class<? extends Entity>> entities, YdbRepository repository) {
         this(entities, repository, Config.DEFAULT);
     }
 
     public YdbSchemaCompatibilityChecker(List<Class<? extends Entity>> entities, YdbRepository repository, Config config) {
-        this.entities = entities;
+        this(repository, toDescriptors(entities), config);
+    }
+
+    public YdbSchemaCompatibilityChecker(YdbRepository repository, List<TableDescriptor<?>> descriptors, Config config) {
+        this.descriptors = descriptors;
         this.config = config;
         this.repository = repository;
         this.repositoryConfig = this.repository.getConfig();
@@ -145,18 +153,16 @@ private String getEndpoint() {
                 .orElse(Objects.toString(repositoryConfig.getHostAndPort()));
     }
 
-    @SneakyThrows
     private Map<String, YdbSchemaOperations.Table> generateSchemeFromCode() {
-        return entities.stream()
+        return descriptors.stream()
                 .map(this::tableForEntity)
                 .collect(toMap(YdbSchemaOperations.Table::getName, Function.identity()));
     }
 
-    @SuppressWarnings("unchecked")
-    private YdbSchemaOperations.Table tableForEntity(Class<? extends Entity> c) {
-        EntitySchema<?> schema = EntitySchema.of(c);
+    private YdbSchemaOperations.Table tableForEntity(TableDescriptor<?> c) {
+        EntitySchema<?> schema = EntitySchema.of(c.entityType());
         return repository.getSchemaOperations()
-                .describeTable(schema.getName(), schema.flattenFields(), schema.flattenId(),
+                .describeTable(c.tableName(), schema.flattenFields(), schema.flattenId(),
                         schema.getGlobalIndexes(), schema.getTtlModifier());
     }
 
@@ -466,6 +472,12 @@ private boolean containsPrefix(String globalName, Set<String> prefixes) {
                 .anyMatch(realName::startsWith);
     }
 
+    private static List<TableDescriptor<?>> toDescriptors(List<Class<? extends Entity>> entities) {
+        List<TableDescriptor<?>> descriptors = new ArrayList<>();
+        entities.forEach(e -> descriptors.add(TableDescriptor.from(EntitySchema.of(e))));
+        return descriptors;
+    }
+
     @Value
     @Builder
     @With

diff --git a/repository/src/main/java/tech/ydb/yoj/repository/db/AbstractDelegatingTable.java b/repository/src/main/java/tech/ydb/yoj/repository/db/AbstractDelegatingTable.java
@@ -3,6 +3,8 @@
 import com.google.common.reflect.TypeToken;
 import lombok.AccessLevel;
 import lombok.Getter;
+import lombok.NonNull;
+import tech.ydb.yoj.ExperimentalApi;
 import tech.ydb.yoj.databind.expression.FilterExpression;
 import tech.ydb.yoj.databind.expression.OrderExpression;
 import tech.ydb.yoj.repository.BaseDb;
@@ -26,6 +28,12 @@ protected AbstractDelegatingTable() {
         this.target = BaseDb.current(BaseDb.class).table(resolveEntityType());
     }
 
+    @ExperimentalApi(issue = "https://github.com/ydb-platform/yoj-project/issues/32")
+    protected AbstractDelegatingTable(@NonNull String tableName) {
+        EntitySchema<T> schema = EntitySchema.of(resolveEntityType());
+        this.target = BaseDb.current(BaseDb.class).table(new TableDescriptor<T>(schema.getType(), tableName));
+    }
+
     @SuppressWarnings("unchecked")
     private Class<T> resolveEntityType() {
         return (Class<T>) (new TypeToken<T>(getClass()) {