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
|
use std::env;
use std::net::SocketAddr;
use std::process::exit;
use log::info;
mod logger;
use tonic::{transport::Server, Request, Response, Status};
pub mod clawflake {
tonic::include_proto!("clawflake");
}
use clawflake::clawflake_server::{Clawflake, ClawflakeServer};
use clawflake::{IdReply, IdRequest};
mod id_worker;
use id_worker::IdWorker;
#[derive(Debug, Default)]
pub struct MyClawflakeService {
}
#[tonic::async_trait]
impl Clawflake for MyClawflakeService {
async fn get_id(
&self,
_: Request<IdRequest>,
) -> Result<Response<IdReply>, Status> {
info!("request on GetID");
let mut worker: IdWorker = IdWorker::new(
env::var("CLAWFLAKE_EPOCH").expect("Missing env `CLAWFLAKE_EPOCH`").parse::<i64>().unwrap(),
env::var("CLAWFLAKE_WORKER_ID").expect("Missing env `CLAWFLAKE_WORKER_ID`").parse::<i64>().unwrap(),
env::var("CLAWFLAKE_DATACENTER_ID").expect("Missing env `CLAWFLAKE_DATACENTER_ID`").parse::<i64>().unwrap()
);
let reply: IdReply = clawflake::IdReply {
id: format!("{}", worker.next_id()).into(),
};
Ok(Response::new(reply))
}
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
// init logger
match logger::init() {
Err(e) => {
eprintln!("failed to init logger: {}", &e);
exit(1);
}
_ => {}
}
// init tonic_health
let (mut health_reporter, health_service) = tonic_health::server::health_reporter();
health_reporter.set_serving::<ClawflakeServer<MyClawflakeService>>().await;
// init tonic and IdWorker
let addr: SocketAddr = "[::1]:50051".parse()?;
let srv: MyClawflakeService = MyClawflakeService::default();
println!("Service listening on {}", addr);
Server::builder()
.add_service(health_service)
.add_service(ClawflakeServer::new(srv))
.serve(addr)
.await?;
Ok(())
}
|
