questdb-rs 7.0.0

QuestDB Client Library for Rust
Documentation 
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

/*******************************************************************************
 *     ___                  _   ____  ____
 *    / _ \ _   _  ___  ___| |_|  _ \| __ )
 *   | | | | | | |/ _ \/ __| __| | | |  _ \
 *   | |_| | |_| |  __/\__ \ |_| |_| | |_) |
 *    \__\_\\__,_|\___||___/\__|____/|____/
 *
 *  Copyright (c) 2014-2019 Appsicle
 *  Copyright (c) 2019-2025 QuestDB
 *
 *  Licensed 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.
 *
 ******************************************************************************/

use crate::ingress::{Protocol, SenderBuilder};
use std::io;
use std::net::UdpSocket;
use std::time::Duration;

pub struct QwpUdpMock {
    socket: UdpSocket,
    host: &'static str,
    port: u16,
}

impl QwpUdpMock {
    pub fn new() -> io::Result<Self> {
        let socket = UdpSocket::bind("127.0.0.1:0")?;
        socket.set_read_timeout(Some(Duration::from_millis(500)))?;
        let port = socket.local_addr()?.port();
        Ok(Self {
            socket,
            host: "127.0.0.1",
            port,
        })
    }

    pub fn sender_builder(&self) -> SenderBuilder {
        SenderBuilder::new(Protocol::Udp, self.host, self.port)
    }

    pub fn recv_datagram(&self) -> io::Result<Vec<u8>> {
        let mut buf = vec![0u8; 65_536];
        let (len, _) = self.socket.recv_from(&mut buf)?;
        buf.truncate(len);
        Ok(buf)
    }

    pub fn recv_datagrams(&self, count: usize) -> io::Result<Vec<Vec<u8>>> {
        let mut datagrams = Vec::with_capacity(count);
        for _ in 0..count {
            datagrams.push(self.recv_datagram()?);
        }
        Ok(datagrams)
    }

    pub fn assert_no_datagram(&self) -> io::Result<()> {
        let previous_timeout = self.socket.read_timeout()?;
        self.socket
            .set_read_timeout(Some(Duration::from_millis(50)))?;
        let result = self.recv_datagram();
        self.socket.set_read_timeout(previous_timeout)?;

        match result {
            Ok(datagram) => Err(io::Error::other(format!(
                "unexpected datagram of {} bytes",
                datagram.len()
            ))),
            Err(err)
                if err.kind() == io::ErrorKind::WouldBlock
                    || err.kind() == io::ErrorKind::TimedOut =>
            {
                Ok(())
            }
            Err(err) => Err(err),
        }
    }
}