socket/
socket.rs

1extern crate shrust;
2use shrust::{Shell, ShellIO};
3use std::io::prelude::*;
4
5use std::sync::{Arc, Mutex};
6use std::thread;
7
8use std::collections::HashMap;
9use std::str::FromStr;
10
11use std::net::TcpListener;
12
13fn main() {
14    let map = Arc::new(Mutex::new(HashMap::new()));
15
16    let mut shell = Shell::new(map);
17
18    shell.new_command(&"put", &"Insert a value", 2, |_, map, args| {
19        map.lock().unwrap().insert(usize::from_str(args[0])?, args[1].to_string());
20        Ok(())
21    });
22    shell.new_command("get", "Get a value", 1, |io, map, args| {
23        match map.lock().unwrap().get(&usize::from_str(args[0])?) {
24            Some(val) => writeln!(io, "{}", val).unwrap(),
25            None => writeln!(io, "Not found").unwrap()
26        };
27        Ok(())
28    });
29    shell.new_command("remove", "Remove a value", 1, |_, map, args| {
30        map.lock().unwrap().remove(&usize::from_str(args[0])?);
31        Ok(())
32    });
33    shell.new_command("list", "List all values", 0, |io, map, _| {
34        for (k, v) in &*map.lock().unwrap() {
35            writeln!(io, "{} = {}", k, v).unwrap();
36        }
37        Ok(())
38    });
39    shell.new_command_noargs("clear", "Clear all values", |_, map| {
40        map.lock().unwrap().clear();
41        Ok(())
42    });
43
44    let serv = TcpListener::bind("0.0.0.0:1234").expect("Cannot open socket");
45    for sock in serv.incoming() {
46        let sock = sock.unwrap();
47        let mut shell = shell.clone();
48        let mut io = ShellIO::new_io(sock);
49        thread::spawn(move || shell.run_loop(&mut io));
50    }
51}
socket/socket.rs

socket/
socket.rs
