1 // Copyright (c) 2023 Huawei Device Co., Ltd.
2 // Licensed under the Apache License, Version 2.0 (the "License");
3 // you may not use this file except in compliance with the License.
4 // You may obtain a copy of the License at
5 //
6 // http://www.apache.org/licenses/LICENSE-2.0
7 //
8 // Unless required by applicable law or agreed to in writing, software
9 // distributed under the License is distributed on an "AS IS" BASIS,
10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 // See the License for the specific language governing permissions and
12 // limitations under the License.
13
14 //! cargo build --example ylong_io_tcp_server --no-default-features
15 //! --features="ylong_tcp" Uses with ylong_io_tcp_server, start
16 //! ylong_io_tcp_server first, then start ylong_io_tcp_client
17
18 use std::collections::HashMap;
19 use std::io;
20 use std::io::{Read, Write};
21 use std::str::from_utf8;
22
23 use ylong_io::{EventTrait, Events, Interest, Poll, TcpListener, Token};
24
25 const SERVER: Token = Token(0);
26
main() -> io::Result<()>27 fn main() -> io::Result<()> {
28 let poll = Poll::new()?;
29 let addr = "127.0.0.1:1234".parse().unwrap();
30 let mut server = TcpListener::bind(addr)?;
31 println!("server socket {server:?}\n");
32
33 poll.register(&mut server, SERVER, Interest::READABLE)?;
34 let mut events = Events::with_capacity(128);
35 // Map of `Token` -> `TcpListener`.
36 let mut connections = HashMap::new();
37 let mut unique_token = Token(SERVER.0 + 1);
38 loop {
39 poll.poll(&mut events, None)?;
40
41 for event in events.iter() {
42 if SERVER == event.token() {
43 let ret = server.accept()?;
44 let (mut stream, addr) = ret;
45 println!("Accept connection addr: {stream:?} {addr:?}");
46 let token = Token(unique_token.0 + 1);
47 unique_token = Token(unique_token.0 + 1);
48 poll.register(
49 &mut stream,
50 token,
51 Interest::READABLE.add(Interest::WRITABLE),
52 )?;
53 connections.insert(token, stream);
54 } else {
55 match connections.get_mut(&event.token()) {
56 Some(connection) => {
57 if event.is_writable() {
58 // println!("server writable\n");
59 match connection.write(b"Hello client_from writable") {
60 Err(err) => {
61 println!("1.Send failed {err}");
62 poll.deregister(connection)?;
63 poll.register(connection, event.token(), Interest::READABLE)?;
64 break;
65 }
66 Ok(n) => {
67 println!("1.send len: {n}\n");
68 poll.deregister(connection)?;
69 poll.register(connection, event.token(), Interest::READABLE)?;
70 break;
71 }
72 }
73 } else if event.is_readable() {
74 println!("server readable\n");
75 let mut msg_buf = [0_u8; 100];
76 match connection.read(&mut msg_buf) {
77 Ok(0) => {
78 poll.deregister(connection)?;
79 }
80 Ok(n) => {
81 if let Ok(str_buf) = from_utf8(&msg_buf[0..n]) {
82 println!("recv msg : {str_buf:?}, len : {n}");
83 } else {
84 println!("Received (none UTF-8) data: {:?}", &msg_buf);
85 }
86 }
87 Err(_n) => {
88 poll.deregister(connection)?;
89 break;
90 }
91 }
92 }
93 }
94 None => break,
95 }
96 }
97 }
98 }
99 }
100