demostf/sync-rs
1
0
Fork 0
mirror of https://codeberg.org/demostf/sync.git synced 2026-06-03 16:44:07 +02:00
Code Issues Releases Wiki Activity

switch to zero-copy deserialization for SyncCommand

Browse source
This commit is contained in:
Robin Appelman 2019-10-27 18:18:19 +01:00
commit 744fe16df5
3 changed files with 89 additions and 109 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/client.rs
View file

@ -68,11 +68,19 @@ mod mock {
}
}
pub fn received(&self) -> Vec<SyncCommand> {
RefCell::borrow(&self.received)
pub fn received_count(&self) -> usize {
RefCell::borrow(&self.received).len()
}
pub fn assert_received(&self, expected: Vec<SyncCommand>) {
let map = RefCell::borrow(&self.received);
let received: Vec<_> = map
.iter()
.map(|msg| serde_json::from_str::<SyncCommand>(msg).expect("invalid message"))
.collect()
.collect();
assert_eq!(expected, received);
}
pub fn clear(&self) {

90
src/integration_tests.rs
View file

@ -52,86 +52,76 @@ fn integration_tests() {
send(
&mut owner,
SyncCommand::Create {
session: "foo".to_string(),
token: "bar".to_string(),
session: "foo",
token: "bar",
},
);
send(
&mut owner,
SyncCommand::Tick {
session: "foo".to_string(),
session: "foo",
tick: 99,
},
);
send(
send(&mut client, SyncCommand::Join { session: "foo" });
assert_receive(
&mut client,
SyncCommand::Join {
session: "foo".to_string(),
SyncCommand::Tick {
session: "foo",
tick: 99,
},
);
assert_eq!(
Some(SyncCommand::Tick {
session: "foo".to_string(),
tick: 99,
}),
receive(&mut client)
);
assert_eq!(
Some(SyncCommand::Play {
session: "foo".to_string(),
play: false
}),
receive(&mut client)
assert_receive(
&mut client,
SyncCommand::Play {
session: "foo",
play: false,
},
);
send(
&mut owner,
SyncCommand::Play {
session: "foo".to_string(),
session: "foo",
play: true,
},
);
assert_eq!(
Some(SyncCommand::Play {
session: "foo".to_string(),
play: true
}),
receive(&mut client)
assert_receive(
&mut client,
SyncCommand::Play {
session: "foo",
play: true,
},
);
// should be ignored
send(
&mut client,
SyncCommand::Tick {
session: "foo".to_string(),
session: "foo",
tick: 5,
},
);
let mut client2 = test.get_client();
send(
send(&mut client2, SyncCommand::Join { session: "foo" });
assert_receive(
&mut client2,
SyncCommand::Join {
session: "foo".to_string(),
SyncCommand::Tick {
session: "foo",
tick: 99,
},
);
assert_eq!(
Some(SyncCommand::Tick {
session: "foo".to_string(),
tick: 99,
}),
receive(&mut client2)
);
assert_eq!(
Some(SyncCommand::Play {
session: "foo".to_string(),
play: true
}),
receive(&mut client2)
assert_receive(
&mut client2,
SyncCommand::Play {
session: "foo",
play: true,
},
);
}
@ -142,10 +132,8 @@ fn send<T: std::io::Write>(client: &mut Client<T>, command: SyncCommand) {
sleep(DELAY);
}
fn receive<T: std::io::Read>(client: &mut Client<T>) -> Option<SyncCommand> {
client
.receive()
.unwrap()
.and_then(|message| message.as_text().map(|s| s.to_string()))
.map(|text| serde_json::from_str(&text).unwrap())
fn assert_receive<T: std::io::Read>(client: &mut Client<T>, expected: SyncCommand) {
let message = client.receive().unwrap().unwrap();
let text = message.as_text().unwrap();
assert_eq!(expected, serde_json::from_str(text).unwrap());
}

94
src/main.rs
View file

@ -14,11 +14,11 @@ use std::time::{Duration, Instant};
#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
#[serde(tag = "type")]
#[serde(rename_all = "lowercase")]
pub enum SyncCommand {
Create { session: String, token: String },
Join { session: String },
Tick { session: String, tick: u64 },
Play { session: String, play: bool },
pub enum SyncCommand<'a> {
Create { session: &'a str, token: &'a str },
Join { session: &'a str },
Tick { session: &'a str, tick: u64 },
Play { session: &'a str, play: bool },
}
#[derive(PartialEq, Debug)]
@ -71,19 +71,19 @@ fn handle_command(
SyncCommand::Create { session, token } => {
sessions
.borrow_mut()
.entry(session.clone())
.entry(session.to_string())
.and_modify(|session| {
if token == &session.owner_token {
session.owner = sender.token();
session.owner_left = None;
}
})
.or_insert_with(|| Session::new(sender.token(), token.clone()));
.or_insert_with(|| Session::new(sender.token(), token.to_string()));
gc_sessions(sessions);
}
SyncCommand::Join {
session: session_name,
} => match sessions.borrow_mut().get_mut(session_name) {
} => match sessions.borrow_mut().get_mut(*session_name) {
Some(session) => {
session.join(sender);
session.send_command(&SyncCommand::Tick {
@ -236,8 +236,8 @@ mod tests {
let input = "{\"type\": \"create\", \"session\": \"foo\", \"token\": \"bar\"}";
assert_eq!(
SyncCommand::Create {
session: "foo".to_string(),
token: "bar".to_string(),
session: "foo",
token: "bar",
},
serde_json::from_str(input).unwrap()
);
@ -249,7 +249,7 @@ mod tests {
let sender = Client::mock(1);
let command = SyncCommand::Create {
session: "test".into(),
token: "bar".to_string(),
token: "bar",
};
handle_command(command, &sender, &sessions);
@ -283,7 +283,7 @@ mod tests {
});
let sender = Client::mock(1);
let command = SyncCommand::Play {
session: "test".into(),
session: "test",
play: true,
};
@ -318,7 +318,7 @@ mod tests {
});
let sender = Client::mock(2);
let command = SyncCommand::Play {
session: "test".into(),
session: "test",
play: true,
};
@ -353,7 +353,7 @@ mod tests {
});
let sender = Client::mock(1);
let command = SyncCommand::Tick {
session: "test".into(),
session: "test",
tick: 99,
};
@ -388,7 +388,7 @@ mod tests {
});
let sender = Client::mock(2);
let command = SyncCommand::Tick {
session: "test".into(),
session: "test",
tick: 99,
};
@ -422,9 +422,7 @@ mod tests {
}
});
let sender = Client::mock(2);
let command = SyncCommand::Join {
session: "test".into(),
};
let command = SyncCommand::Join { session: "test" };
handle_command(command, &sender, &sessions);
@ -443,19 +441,16 @@ mod tests {
);
if let Client::Mock(mock) = sender {
assert_eq!(
vec![
SyncCommand::Tick {
session: "test".into(),
tick: 99,
},
SyncCommand::Play {
session: "test".into(),
play: true,
}
],
mock.received()
);
mock.assert_received(vec![
SyncCommand::Tick {
session: "test",
tick: 99,
},
SyncCommand::Play {
session: "test",
play: true,
},
]);
};
}
@ -472,9 +467,7 @@ mod tests {
}
});
let sender = Client::mock(2);
let command = SyncCommand::Join {
session: "test2".into(),
};
let command = SyncCommand::Join { session: "test2" };
handle_command(command, &sender, &sessions);
@ -516,15 +509,11 @@ mod tests {
let owner = Client::mock(1);
let sender1 = Client::mock(2);
let sender2 = Client::mock(3);
let command = SyncCommand::Join {
session: "test".into(),
};
let command = SyncCommand::Join { session: "test" };
handle_command(command, &sender1, &sessions);
let command = SyncCommand::Join {
session: "test2".into(),
};
let command = SyncCommand::Join { session: "test2" };
handle_command(command, &sender2, &sessions);
if let Client::Mock(mock) = &sender1 {
@ -535,23 +524,20 @@ mod tests {
}
let command = SyncCommand::Tick {
session: "test".into(),
session: "test",
tick: 999,
};
handle_command(command, &owner, &sessions);
if let Client::Mock(mock) = sender1 {
assert_eq!(
vec![SyncCommand::Tick {
session: "test".into(),
tick: 999,
}],
mock.received()
);
mock.assert_received(vec![SyncCommand::Tick {
session: "test",
tick: 999,
}]);
};
if let Client::Mock(mock) = sender2 {
assert_eq!(0, mock.received().len());
assert_eq!(0, mock.received_count());
};
}
@ -569,9 +555,7 @@ mod tests {
});
let sender1 = Client::mock(2);
let sender2 = Client::mock(3);
let command = SyncCommand::Join {
session: "test".into(),
};
let command = SyncCommand::Join { session: "test" };
handle_command(command.clone(), &sender1, &sessions);
handle_command(command.clone(), &sender2, &sessions);
@ -584,17 +568,17 @@ mod tests {
}
let command = SyncCommand::Tick {
session: "test".into(),
session: "test",
tick: 999,
};
handle_command(command, &sender1, &sessions);
if let Client::Mock(mock) = sender1 {
assert_eq!(0, mock.received().len());
assert_eq!(0, mock.received_count());
};
if let Client::Mock(mock) = sender2 {
assert_eq!(0, mock.received().len());
assert_eq!(0, mock.received_count());
};
}
}