Added StoreType to TableDescription

Author: alex268

table/pom.xml

@@ -47,7 +47,6 @@
             <plugin>
                 <groupId>org.apache.maven.plugins</groupId>
                 <artifactId>maven-surefire-plugin</artifactId>
-                <version>3.1.0</version>
                 <configuration>
                     <environmentVariables>
                         <TESTCONTAINERS_REUSE_ENABLE>true</TESTCONTAINERS_REUSE_ENABLE>

table/src/main/java/tech/ydb/table/description/TableDescription.java

@@ -24,7 +24,12 @@
  * @author Sergey Polovko
  */
 public class TableDescription {
+    public enum StoreType {
+        ROWS,
+        COLUMNS
+    }
 
+    private final StoreType storeType;
     private final List<String> primaryKeys;
     private final List<TableColumn> columns;
     private final List<TableIndex> indexes;
@@ -42,6 +47,7 @@ public class TableDescription {
     private final TableTtl tableTtl;
 
     private TableDescription(Builder builder) {
+        this.storeType = builder.storeType;
         this.primaryKeys = ImmutableList.copyOf(builder.primaryKeys);
         this.columns = builder.buildColumns();
         this.indexes = ImmutableList.copyOf(builder.indexes);
@@ -59,6 +65,10 @@ public static Builder newBuilder() {
         return new Builder();
     }
 
+    public StoreType getStoreType() {
+        return storeType;
+    }
+
     public List<String> getPrimaryKeys() {
         return primaryKeys;
     }
@@ -105,7 +115,7 @@ public List<ChangefeedDescription> getChangefeeds() {
      * BUILDER
      */
     public static class Builder {
-
+        private StoreType storeType = StoreType.ROWS;
         private List<String> primaryKeys = Collections.emptyList();
         private final LinkedHashMap<String, TypeAndFamily> columns = new LinkedHashMap<>();
         private final List<TableIndex> indexes = new ArrayList<>();
@@ -118,6 +128,11 @@ public static class Builder {
         private TableTtl ttlSettings = TableTtl.notSet();
         private final List<ChangefeedDescription> changefeeds = new ArrayList<>();
 
+        public Builder withStoreType(StoreType storeType) {
+            this.storeType = storeType;
+            return this;
+        }
+
         public Builder addNonnullColumn(String name, Type type) {
             return addNonnullColumn(name, type, null);
         }

table/src/main/java/tech/ydb/table/impl/BaseSession.java

@@ -362,6 +362,17 @@ public CompletableFuture<Status> createTable(
                 .setOperationParams(Operation.buildParams(settings.toOperationSettings()))
                 .addAllPrimaryKey(description.getPrimaryKeys());
 
+        switch (description.getStoreType()) {
+            case ROWS:
+                request.setStoreType(YdbTable.StoreType.STORE_TYPE_ROW);
+                break;
+            case COLUMNS:
+                request.setStoreType(YdbTable.StoreType.STORE_TYPE_COLUMN);
+                break;
+            default:
+                break;
+        }
+
         for (ColumnFamily family: description.getColumnFamilies()) {
             request.addColumnFamilies(buildColumnFamity(family));
         }
@@ -726,6 +737,19 @@ private static TableDescription mapDescribeTable(
             DescribeTableSettings describeTableSettings
     ) {
         TableDescription.Builder description = TableDescription.newBuilder();
+        switch (result.getStoreType()) {
+            case STORE_TYPE_ROW:
+                description = description.withStoreType(TableDescription.StoreType.ROWS);
+                break;
+            case STORE_TYPE_COLUMN:
+                description = description.withStoreType(TableDescription.StoreType.COLUMNS);
+                break;
+            case UNRECOGNIZED:
+            case STORE_TYPE_UNSPECIFIED:
+            default:
+                break;
+        }
+
         for (int i = 0; i < result.getColumnsCount(); i++) {
             YdbTable.ColumnMeta column = result.getColumns(i);
             description.addNonnullColumn(column.getName(), ProtoType.fromPb(column.getType()), column.getFamily());