clickhouse-sys 0.0.1

Static clickhouse-cpp bridge for the clickhouse-cxx workspace
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#pragma once

#include <clickhouse/base/socket.h>
#include <rust/cxx.h>

#include <cstdint>
#include <memory>
#include <string>
#include <vector>

struct TlsSession;

struct TlsConnect;

/// Socket factory whose streams read and write through a Rust rustls session.
class RustlsSocketFactory : public clickhouse::SocketFactory {
public:
    struct Settings {
        bool secure = false;
        std::string sni;
        std::vector<std::uint8_t> ca_pem;
        bool use_native_roots = true;
        bool skip_verification = false;
    };

    explicit RustlsSocketFactory(Settings settings);
    ~RustlsSocketFactory() override;

    std::unique_ptr<clickhouse::SocketBase> connect(
        const clickhouse::ClientOptions& opts,
        const clickhouse::Endpoint& endpoint) override;

private:
    Settings settings_;
};

class RustlsSocket : public clickhouse::SocketBase {
public:
    explicit RustlsSocket(rust::Box<TlsSession> session);
    ~RustlsSocket() override;

    std::unique_ptr<clickhouse::InputStream> makeInputStream() const override;
    std::unique_ptr<clickhouse::OutputStream> makeOutputStream() const override;

private:
    std::shared_ptr<rust::Box<TlsSession>> session_;
};