Skip to content

Commit ba3412d

Browse files
committed
Adding bitmap serialization tests
1 parent 9812103 commit ba3412d

File tree

2 files changed

+32
-6
lines changed

2 files changed

+32
-6
lines changed

client-v2/pom.xml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,11 @@
123123
<version>2.35.2</version>
124124
<scope>test</scope>
125125
</dependency>
126+
<dependency>
127+
<groupId>org.roaringbitmap</groupId>
128+
<artifactId>RoaringBitmap</artifactId>
129+
<scope>test</scope>
130+
</dependency>
126131
</dependencies>
127132

128133
<build>

client-v2/src/test/java/com/clickhouse/client/insert/SamplePOJO.java

Lines changed: 27 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
package com.clickhouse.client.insert;
22

3-
import com.clickhouse.client.api.metadata.TableSchema;
4-
import com.clickhouse.data.ClickHouseColumn;
5-
import com.clickhouse.data.ClickHouseEnum;
3+
import com.clickhouse.data.value.ClickHouseBitmap;
4+
65
import org.apache.commons.lang3.RandomStringUtils;
76

87
import java.math.BigDecimal;
@@ -14,13 +13,11 @@
1413
import java.time.LocalDateTime;
1514
import java.util.ArrayList;
1615
import java.util.Arrays;
17-
import java.util.Collections;
1816
import java.util.HashMap;
1917
import java.util.List;
2018
import java.util.Map;
2119
import java.util.Random;
2220
import java.util.UUID;
23-
import java.util.stream.IntStream;
2421

2522
public class SamplePOJO {
2623
private int int8;
@@ -76,6 +73,9 @@ public class SamplePOJO {
7673
private List<Integer> nestedInnerInt;
7774
private List<String> nestedInnerString;
7875

76+
private ClickHouseBitmap groupBitmapUint32;
77+
private ClickHouseBitmap groupBitmapUint64;
78+
7979
public SamplePOJO() {
8080
final Random random = new Random();
8181
int8 = random.nextInt(128);
@@ -154,6 +154,9 @@ public SamplePOJO() {
154154
List<String> innerString = new ArrayList<>();
155155
innerString.add(RandomStringUtils.randomAlphabetic(1, 256));
156156
nestedInnerString = innerString;
157+
158+
groupBitmapUint32 = ClickHouseBitmap.wrap(random.ints(5, Integer.MAX_VALUE - 100, Integer.MAX_VALUE).toArray());
159+
groupBitmapUint64 = ClickHouseBitmap.wrap(new Random().longs(5, Long.MAX_VALUE - 100, Long.MAX_VALUE).toArray());
157160
}
158161

159162
public int getInt8() {
@@ -460,6 +463,22 @@ public void setNestedInnerString(List<String> nestedInnerString) {
460463
this.nestedInnerString = nestedInnerString;
461464
}
462465

466+
public ClickHouseBitmap getGroupBitmapUint32() {
467+
return groupBitmapUint32;
468+
}
469+
470+
public void setGroupBitmapUint32(ClickHouseBitmap groupBitmapUint32) {
471+
this.groupBitmapUint32 = groupBitmapUint32;
472+
}
473+
474+
public ClickHouseBitmap getGroupBitmapUint64() {
475+
return groupBitmapUint64;
476+
}
477+
478+
public void setGroupBitmapUint64(ClickHouseBitmap groupBitmapUint64) {
479+
this.groupBitmapUint64 = groupBitmapUint64;
480+
}
481+
463482
public static String generateTableCreateSQL(String tableName) {
464483
return "CREATE TABLE " + tableName + " (" +
465484
"int8 Int8, " +
@@ -501,7 +520,9 @@ public static String generateTableCreateSQL(String tableName) {
501520
"array Array(String), " +
502521
"tuple Tuple(Int32, Int32, Int32, Int32, Int32, Int32, Int32, Int32, Int32, Int32), " +
503522
"map Map(String, Int32), " +
504-
"nested Nested (innerInt Int32, innerString String)" +
523+
"nested Nested (innerInt Int32, innerString String)," +
524+
"groupBitmapUint32 AggregateFunction(groupBitmap, UInt32)," +
525+
"groupBitmapUint64 AggregateFunction(groupBitmap, UInt64)" +
505526
") ENGINE = Memory";
506527
}
507528
}

0 commit comments

Comments
 (0)