com.clickhouse.r2dbc.ClickHouseBatch Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of clickhouse-r2dbc Show documentation

R2DBC driver for ClickHouse

There is a newer version: 0.8.0

package com.clickhouse.r2dbc;

import com.clickhouse.client.ClickHouseRequest;
import com.clickhouse.data.ClickHouseFormat;
import io.r2dbc.spi.Batch;
import io.r2dbc.spi.Result;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.ArrayList;
import java.util.List;

public class ClickHouseBatch implements Batch {

    private static final ClickHouseFormat PREFERRED_FORMAT = ClickHouseFormat.TabSeparatedWithNamesAndTypes;
    private ClickHouseRequest request;
    final List sqlList;

    public ClickHouseBatch(ClickHouseRequest request) {
        this.request = request;
        this.sqlList = new ArrayList<>();
    }

    @Override
    public Batch add(String sql) {
        sqlList.add(sql);
        return this;
    }

    @Override
    public Publisher execute() {
        return Flux.fromStream(sqlList.stream().map(sql -> {
            request.query(sql).format(PREFERRED_FORMAT);
            return Mono.fromFuture(request::execute); }))
                .flatMap(Mono::flux)
                .map(ClickHouseResult::new);
    }
}