This is an automated email from the ASF dual-hosted git repository.
JingsongLi pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/paimon.git
The following commit(s) were added to refs/heads/master by this push:
new 1384d724c6 [core][mosaic] Support MOSAIC in FormatTable (#8180)
1384d724c6 is described below
commit 1384d724c6177e53b947d0d8c908744f41a8f005
Author: jianguotian <[email protected]>
AuthorDate: Tue Jun 9 16:29:45 2026 +0800
[core][mosaic] Support MOSAIC in FormatTable (#8180)
---
.../main/java/org/apache/paimon/CoreOptions.java | 2 +
.../java/org/apache/paimon/table/FormatTable.java | 3 +-
.../format/mosaic/FormatTableMosaicReadTest.java | 212 +++++++++++++++++++++
3 files changed, 216 insertions(+), 1 deletion(-)
diff --git a/paimon-api/src/main/java/org/apache/paimon/CoreOptions.java
b/paimon-api/src/main/java/org/apache/paimon/CoreOptions.java
index c1309e9569..4a4b7b5f81 100644
--- a/paimon-api/src/main/java/org/apache/paimon/CoreOptions.java
+++ b/paimon-api/src/main/java/org/apache/paimon/CoreOptions.java
@@ -290,6 +290,7 @@ public class CoreOptions implements Serializable {
public static final String FILE_FORMAT_CSV = "csv";
public static final String FILE_FORMAT_TEXT = "text";
public static final String FILE_FORMAT_JSON = "json";
+ public static final String FILE_FORMAT_MOSAIC = "mosaic";
public static final ConfigOption<String> FILE_FORMAT =
key("file.format")
@@ -2902,6 +2903,7 @@ public class CoreOptions implements Serializable {
return "snappy";
case FILE_FORMAT_AVRO:
case FILE_FORMAT_ORC:
+ case FILE_FORMAT_MOSAIC:
return "zstd";
case FILE_FORMAT_CSV:
case FILE_FORMAT_TEXT:
diff --git a/paimon-core/src/main/java/org/apache/paimon/table/FormatTable.java
b/paimon-core/src/main/java/org/apache/paimon/table/FormatTable.java
index aee60f65ee..4149683d7a 100644
--- a/paimon-core/src/main/java/org/apache/paimon/table/FormatTable.java
+++ b/paimon-core/src/main/java/org/apache/paimon/table/FormatTable.java
@@ -79,7 +79,8 @@ public interface FormatTable extends Table {
PARQUET,
CSV,
TEXT,
- JSON
+ JSON,
+ MOSAIC
}
/** Parses a file format string to a corresponding {@link Format} enum
constant. */
diff --git
a/paimon-mosaic/src/test/java/org/apache/paimon/format/mosaic/FormatTableMosaicReadTest.java
b/paimon-mosaic/src/test/java/org/apache/paimon/format/mosaic/FormatTableMosaicReadTest.java
new file mode 100644
index 0000000000..6891ba9ecf
--- /dev/null
+++
b/paimon-mosaic/src/test/java/org/apache/paimon/format/mosaic/FormatTableMosaicReadTest.java
@@ -0,0 +1,212 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.paimon.format.mosaic;
+
+import org.apache.paimon.catalog.Identifier;
+import org.apache.paimon.data.BinaryString;
+import org.apache.paimon.data.GenericRow;
+import org.apache.paimon.data.InternalRow;
+import org.apache.paimon.data.serializer.InternalRowSerializer;
+import org.apache.paimon.fs.Path;
+import org.apache.paimon.fs.local.LocalFileIO;
+import org.apache.paimon.reader.RecordReader;
+import org.apache.paimon.table.FormatTable;
+import org.apache.paimon.table.sink.BatchTableCommit;
+import org.apache.paimon.table.sink.BatchTableWrite;
+import org.apache.paimon.table.sink.BatchWriteBuilder;
+import org.apache.paimon.table.sink.CommitMessage;
+import org.apache.paimon.table.source.ReadBuilder;
+import org.apache.paimon.table.source.Split;
+import org.apache.paimon.types.DataTypes;
+import org.apache.paimon.types.RowType;
+
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.junit.jupiter.api.Assumptions.assumeTrue;
+
+/** End-to-end tests for a {@link FormatTable} backed by mosaic files. */
+class FormatTableMosaicReadTest {
+
+ @TempDir java.nio.file.Path tempPath;
+
+ @BeforeAll
+ static void checkNativeLibrary() {
+ assumeTrue(isNativeAvailable(), "Mosaic native library not available");
+ }
+
+ @Test
+ void testReadWriteUnpartitioned() throws Exception {
+ RowType rowType =
+ RowType.builder()
+ .field("a", DataTypes.STRING())
+ .field("b", DataTypes.BIGINT())
+ .field("c", DataTypes.DOUBLE())
+ .build();
+ FormatTable table = buildFormatTable(rowType, Collections.emptyList(),
new HashMap<>());
+
+ writeAll(
+ table,
+ Arrays.asList(
+ GenericRow.of(BinaryString.fromString("a1"), 1L, 1.1),
+ GenericRow.of(BinaryString.fromString("a2"), 2L, 2.2),
+ GenericRow.of(BinaryString.fromString("a3"), 3L,
3.3)));
+
+ List<InternalRow> result = readAll(table, rowType);
+
+ assertThat(result).hasSize(3);
+ assertThat(result.get(0).getString(0).toString()).isEqualTo("a1");
+ assertThat(result.get(0).getLong(1)).isEqualTo(1L);
+ assertThat(result.get(0).getDouble(2)).isEqualTo(1.1);
+ assertThat(result.get(2).getString(0).toString()).isEqualTo("a3");
+ }
+
+ @Test
+ void testReadWritePartitioned() throws Exception {
+ RowType rowType =
+ RowType.builder()
+ .field("a", DataTypes.STRING())
+ .field("b", DataTypes.BIGINT())
+ .field("dt", DataTypes.STRING())
+ .build();
+ FormatTable table =
+ buildFormatTable(rowType, Collections.singletonList("dt"), new
HashMap<>());
+
+ writeAll(
+ table,
+ Arrays.asList(
+ GenericRow.of(
+ BinaryString.fromString("a1"),
+ 1L,
+ BinaryString.fromString("20260608")),
+ GenericRow.of(
+ BinaryString.fromString("a2"),
+ 2L,
+ BinaryString.fromString("20260608")),
+ GenericRow.of(
+ BinaryString.fromString("a3"),
+ 3L,
+ BinaryString.fromString("20260609"))));
+
+ LocalFileIO fileIO = LocalFileIO.create();
+ Path tablePath = new Path(table.location());
+ assertThat(fileIO.exists(new Path(tablePath, "dt=20260608"))).isTrue();
+ assertThat(fileIO.exists(new Path(tablePath, "dt=20260609"))).isTrue();
+
+ List<InternalRow> result = readAll(table, rowType);
+
+ assertThat(result).hasSize(3);
+ List<String> aDt = new ArrayList<>();
+ for (InternalRow row : result) {
+ aDt.add(row.getString(0).toString() + "/" +
row.getString(2).toString());
+ }
+ assertThat(aDt).containsExactlyInAnyOrder("a1/20260608",
"a2/20260608", "a3/20260609");
+ }
+
+ @Test
+ void testEmptyDirectoryScansToNoSplits() throws Exception {
+ RowType rowType = RowType.builder().field("a",
DataTypes.STRING()).build();
+ FormatTable table = buildFormatTable(rowType, Collections.emptyList(),
new HashMap<>());
+
+ List<Split> splits = table.newReadBuilder().newScan().plan().splits();
+ assertThat(splits).isEmpty();
+ }
+
+ @Test
+ void testInvalidCompressionPropagatedFromWriter() {
+ RowType rowType = RowType.builder().field("a",
DataTypes.STRING()).build();
+ Map<String, String> options = new HashMap<>();
+ options.put("file.compression", "gzip");
+ FormatTable table = buildFormatTable(rowType, Collections.emptyList(),
options);
+
+ assertThatThrownBy(
+ () ->
+ writeAll(
+ table,
+ Collections.singletonList(
+
GenericRow.of(BinaryString.fromString("a1")))))
+ .isInstanceOf(UnsupportedOperationException.class)
+ .hasMessageContaining("Mosaic format only supports zstd");
+ }
+
+ private FormatTable buildFormatTable(
+ RowType rowType, List<String> partitionKeys, Map<String, String>
options) {
+ String location = new Path(tempPath.toUri()).toString();
+ options.put("file.format", "mosaic");
+ // CoreOptions.path() reads from options; SQL/REST catalogs inject it,
+ // a raw Java-API caller has to do it explicitly.
+ options.put("path", location);
+ return FormatTable.builder()
+ .fileIO(LocalFileIO.create())
+ .identifier(Identifier.create("default", "t"))
+ .rowType(rowType)
+ .partitionKeys(partitionKeys)
+ .location(location)
+ .format(FormatTable.Format.MOSAIC)
+ .options(options)
+ .build();
+ }
+
+ private static void writeAll(FormatTable table, List<InternalRow> rows)
throws Exception {
+ BatchWriteBuilder builder = table.newBatchWriteBuilder();
+ try (BatchTableWrite write = builder.newWrite()) {
+ for (InternalRow row : rows) {
+ write.write(row);
+ }
+ List<CommitMessage> committables = write.prepareCommit();
+ try (BatchTableCommit commit = builder.newCommit()) {
+ commit.commit(committables);
+ }
+ }
+ }
+
+ private static List<InternalRow> readAll(FormatTable table, RowType
rowType) throws Exception {
+ ReadBuilder readBuilder = table.newReadBuilder();
+ List<Split> splits = readBuilder.newScan().plan().splits();
+ assertThat(splits).isNotEmpty();
+
+ List<InternalRow> result = new ArrayList<>();
+ InternalRowSerializer serializer = new InternalRowSerializer(rowType);
+ for (Split split : splits) {
+ try (RecordReader<InternalRow> reader =
readBuilder.newRead().createReader(split)) {
+ reader.forEachRemaining(row ->
result.add(serializer.copy(row)));
+ }
+ }
+ return result;
+ }
+
+ private static boolean isNativeAvailable() {
+ try {
+ Class.forName("org.apache.paimon.mosaic.NativeLib");
+ return true;
+ } catch (Throwable t) {
+ return false;
+ }
+ }
+}
