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(&mut stream, token, Interest::READABLE | Interest::WRITABLE)?;
49 connections.insert(token, stream);
50 } else {
51 match connections.get_mut(&event.token()) {
52 Some(connection) => {
53 if event.is_writable() {
54 // println!("server writable\n");
55 match connection.write(b"Hello client_from writable") {
56 Err(err) => {
57 println!("1.Send failed {err}");
58 poll.deregister(connection)?;
59 poll.register(connection, event.token(), Interest::READABLE)?;
60 break;
61 }
62 Ok(n) => {
63 println!("1.send len: {n}\n");
64 poll.deregister(connection)?;
65 poll.register(connection, event.token(), Interest::READABLE)?;
66 break;
67 }
68 }
69 } else if event.is_readable() {
70 println!("server readable\n");
71 let mut msg_buf = [0_u8; 100];
72 match connection.read(&mut msg_buf) {
73 Ok(0) => {
74 poll.deregister(connection)?;
75 }
76 Ok(n) => {
77 if let Ok(str_buf) = from_utf8(&msg_buf[0..n]) {
78 println!("recv msg : {str_buf:?}, len : {n}");
79 } else {
80 println!("Received (none UTF-8) data: {:?}", &msg_buf);
81 }
82 }
83 Err(_n) => {
84 poll.deregister(connection)?;
85 break;
86 }
87 }
88 }
89 }
90 None => break,
91 }
92 }
93 }
94 }
95 }
96