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
use crate::SOCK;
use async_std::{net::SocketAddr, task};
use flate2::{write::ZlibEncoder, Compression};
use myutil::{err::*, *};
use serde::Serialize;
use std::{io::Write, time::Duration};
use ttserver_def::*;
#[macro_export(crate)]
macro_rules! send_ok {
($uuid: expr, $msg: expr, $peeraddr: expr) => {
$crate::util::send_back(
$crate::util::gen_resp_ok($uuid, $msg),
$peeraddr,
)
};
}
pub(crate) fn gen_resp_ok(uuid: u64, msg: impl Serialize) -> Resp {
Resp {
uuid,
status: RetStatus::Success,
msg: info!(serde_json::to_vec(&msg)).unwrap_or_default(),
}
}
#[macro_export(crate)]
macro_rules! send_err {
($uuid: expr, $err: expr, $peeraddr: expr) => {{
let log = genlog($err);
$crate::util::send_back(
$crate::util::gen_resp_err($uuid, &log),
$peeraddr,
)
.c(d!(&log))
.map(|_| p(eg!(log)))
}};
}
pub(crate) fn gen_resp_err(uuid: u64, msg: &str) -> Resp {
Resp {
uuid,
status: RetStatus::Fail,
msg: msg.as_bytes().to_vec(),
}
}
#[inline(always)]
pub(crate) fn send_back(resp: Resp, peeraddr: SocketAddr) -> Result<()> {
serde_json::to_vec(&resp)
.c(d!())
.and_then(|resp| {
let mut en = ZlibEncoder::new(vct![], Compression::default());
en.write_all(&resp[..]).c(d!())?;
en.finish().c(d!())
})
.map(|resp_compressed| {
task::spawn(async move {
info_omit!(SOCK.send_to(&resp_compressed, peeraddr).await);
});
})
}
#[inline(always)]
pub(crate) async fn asleep(secs: u64) {
task::sleep(Duration::from_secs(secs)).await
}