github/wasm-pong
1
0
Fork 0
mirror of https://github.com/thilo-behnke/wasm-pong.git synced 2026-07-11 02:29:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

why can't the test find the utils method?

Browse Source
This commit is contained in:
Thilo Behnke
2022-05-13 17:50:00 +02:00
parent de3d96a919
commit 44bf7c24f8
7 changed files with 68 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pong/src/event.rs
View File

@@ -67,6 +67,7 @@ pub mod event {
pub trait EventReaderImpl : Send + Sync {
fn read(&mut self) -> Vec<Event>;
fn read_from_topic(&mut self, topic: &str, key: &str) -> Vec<Event>;
}
pub struct EventReader {
@@ -83,5 +84,9 @@ pub mod event {
pub fn read(&mut self) -> Vec<Event> {
self.reader_impl.read()
}
pub fn read_from_topic(&mut self, topic: &str, key: &str) -> Vec<Event> {
self.reader_impl.read_from_topic(topic, key)
}
}
}

3
server/Cargo.toml
View File

@@ -12,3 +12,6 @@ tokio = { version = "1", features = ["full"] }
tokio-stream = {version = "0.1" }
serde_json = { version = "1.0.79" }
pong = { path = "../pong", version = "0.1.0" }
[dev-dependencies]
rstest = "0.12.0"

18
server/src/http.rs
View File

@@ -8,6 +8,7 @@ use kafka::producer::Producer;
use tokio::sync::Mutex;
use pong::event::event::{Event, EventReader, EventWriter};
use crate::kafka::{KafkaEventReaderImpl, KafkaEventWriterImpl};
use crate::utils::http_utils::get_query_params;
pub struct HttpServer {
addr: [u8; 4],
@@ -68,10 +69,19 @@ async fn handle_event_write(event_writer: &Arc<Mutex<EventWriter>>, req: Request
}
async fn handle_event_read(event_reader: &Arc<Mutex<EventReader>>, req: Request<Body>) -> Result<Response<Body>, Infallible> {
let mut locked = event_reader.lock().await;
let events = locked.read();
println!("read {} events", events.len());
Ok(Response::new(format!("{:?}", events).into()))
let query_params = get_query_params(&req);
println!("{:?}", query_params);
let topic = query_params.get("topic");
let key = query_params.get("key");
match (topic, key) {
(Some(topic), Some(key)) => {
let mut locked = event_reader.lock().await;
let events = locked.read();
println!("read {} events", events.len());
Ok(Response::new(format!("{:?}", events).into()))
},
_ => Ok(Response::new(format!("provide topic and key as query params").into())),
}
}

16
server/src/kafka.rs
View File

@@ -57,16 +57,30 @@ impl KafkaEventReaderImpl {
}
impl EventReaderImpl for KafkaEventReaderImpl {
fn read(&mut self) -> Vec<Event> {
self.consume(None, None)
}
fn read_from_topic(&mut self, topic: &str, key: &str) -> Vec<Event> {
self.consume(Some(topic), Some(key))
}
}
impl KafkaEventReaderImpl {
fn consume(&mut self, topic: Option<&str>, key: Option<&str>) -> Vec<Event> {
let polled = self.consumer.poll().unwrap();
let message_sets: Vec<MessageSet<'_>> = polled.iter().collect();
let mut events = vec![];
for ms in message_sets {
let topic = ms.topic();
let partition = ms.partition();
println!("topic={},partition={}", topic, partition);
for m in ms.messages() {
let event = Event {topic: String::from(ms.topic()), key: std::str::from_utf8(m.key).unwrap().parse().unwrap(), msg: std::str::from_utf8(m.value).unwrap().parse().unwrap() };
let event = Event {topic: String::from(topic), key: std::str::from_utf8(m.key).unwrap().parse().unwrap(), msg: std::str::from_utf8(m.value).unwrap().parse().unwrap() };
events.push(event);
}
self.consumer.consume_messageset(ms);
}
self.consumer.commit_consumed().unwrap();
events
}
}

2
server/src/main.rs
View File

@@ -2,6 +2,8 @@ use crate::http::HttpServer;
mod http;
mod kafka;
pub mod utils;
#[tokio::main]
pub async fn main() {

15
server/src/utils.rs Normal file
View File

@@ -0,0 +1,15 @@
pub mod http_utils {
use std::collections::HashMap;
use hyper::{Body, Request};
pub fn get_query_params(req: &Request<Body>) -> HashMap<&str, &str> {
let uri = req.uri();
let query = uri.query();
match query {
None => HashMap::new(),
Some(query) => {
query.split("&").map(|s| s.split_at(query.find("=").unwrap())).collect()
}
}
}
}

14
server/tests/utils_tests.rs Normal file
View File

@@ -0,0 +1,14 @@
mod test {
use std::collections::HashMap;
use rstest::rstest;
#[rstest]
#[case(
"test=abc",
HashMap::from([("test", "abc")])
)]
fn get_query_params_tests(#[case] query_str: &str, #[case] expected: HashMap<&str, &str>) {
let res = get_(query_str);
assert_eq!(res, expected)
}
}