Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2024-11-16 17:07:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

message-queue RocksDB + fix listening

Browse source
This commit is contained in:
Luke Parker 2023-07-17 00:20:10 -04:00
parent 8543487db2
commit 845c2842b5
No known key found for this signature in database
3 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Cargo.lock generated
View file

@ -8696,6 +8696,7 @@ dependencies = [
"jsonrpsee",
"lazy_static",
"log",
"rocksdb",
"schnorr-signatures",
"serai-db",
"serai-primitives",

3
message-queue/Cargo.toml
View file

@ -32,7 +32,8 @@ schnorr-signatures = { path = "../crypto/schnorr" }
log = "0.4"
tokio = { version = "1", features = ["full"] }
serai-db = { path = "../common/db" }
serai-db = { path = "../common/db", features = ["rocksdb"] }
rocksdb = "0.21"
serai-primitives = { path = "../substrate/primitives" }

15
message-queue/src/main.rs
View file

@ -16,10 +16,12 @@ use messages::*;
mod queue;
use queue::Queue;
type Db = Arc<rocksdb::TransactionDB>;
lazy_static::lazy_static! {
static ref KEYS: Arc<RwLock<HashMap<Service, <Ristretto as Ciphersuite>::G>>> =
Arc::new(RwLock::new(HashMap::new()));
static ref QUEUES: Arc<RwLock<HashMap<Service, RwLock<Queue<serai_db::MemDb>>>>> =
static ref QUEUES: Arc<RwLock<HashMap<Service, RwLock<Queue<Db>>>>> =
Arc::new(RwLock::new(HashMap::new()));
}
@ -97,8 +99,8 @@ fn ack_message(service: Service, id: u64, _signature: SchnorrSignature<Ristretto
#[tokio::main]
async fn main() {
// Open the DB
// TODO
let db = serai_db::MemDb::new();
let db =
Arc::new(rocksdb::TransactionDB::open_default(std::env::var("DB_PATH").unwrap()).unwrap());
let read_key = |str| {
let Ok(key) = std::env::var(str) else { None? };
@ -132,7 +134,8 @@ async fn main() {
// Start server
let builder = ServerBuilder::new();
// TODO: Set max request/response size
let listen_on: &[std::net::SocketAddr] = &["0.0.0.0".parse().unwrap()];
// 5132 ^ ((b'M' << 8) | b'Q')
let listen_on: &[std::net::SocketAddr] = &["0.0.0.0:2287".parse().unwrap()];
let server = builder.build(listen_on).await.unwrap();
let mut module = RpcModule::new(());
@ -165,5 +168,7 @@ async fn main() {
Ok(())
})
.unwrap();
server.start(module).unwrap();
// Run until stopped, which it never will
server.start(module).unwrap().stopped().await;
}