bevy_octopus_websocket 0.2.1

ECS based networking library for Bevy
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

#![allow(dead_code)]

use bevy::{
    log::{Level, LogPlugin},
    prelude::*,
};
use serde::{Deserialize, Serialize};

use bevy_octopus::prelude::*;
use bevy_octopus_websocket::{WebsocketAddress, WebsocketPlugin};

/// shared app setup
pub fn shared_setup(app: &mut App) {
    app.add_plugins((
        MinimalPlugins.set(bevy::app::ScheduleRunnerPlugin::run_loop(
            std::time::Duration::from_secs_f64(1.0 / 60.0),
        )),
        LogPlugin {
            level: Level::INFO,
            filter: "bevy_octopus=debug".to_string(),
            ..default()
        },
    ))
    .add_plugins(OctopusPlugin)
    .add_plugins(WebsocketPlugin)
    .add_observer(on_node_event);
}

/// this channel is sending and receiving raw packet
pub const RAW_CHANNEL: ChannelId = ChannelId("raw channel");

/// this channel is sending and receiving json packet
pub const JSON_CHANNEL: ChannelId = ChannelId("json channel");

/// this channel is sending and receiving bincode packet
pub const BINCODE_CHANNEL: ChannelId = ChannelId("bincode channel");

#[derive(Serialize, Deserialize, Debug)]
pub struct PlayerInformation {
    pub health: usize,
    pub position: (u32, u32, u32),
}

pub fn on_node_event(on: On<NodeEvent>) {
    let e = on.event();
    info!("{:?} trigger {:?}", e.entity, e.event);
}

pub fn handle_message_events(
    mut ev_channels: MessageReader<ReceiveChannelMessage<PlayerInformation>>,
) {
    for event in ev_channels.read() {
        info!("{} Received: {:?}", event.channel_id, &event.message);
    }
}

pub fn handle_raw_packet(q_server: Query<(&ChannelId, &NetworkNode)>) {
    for (channel_id, net_node) in q_server.iter() {
        while let Ok(Some(packet)) = net_node.recv_message_channel.receiver.try_recv() {
            info!("{} Received: {:?}", channel_id, packet.bytes);
        }
    }
}

pub fn send_json_message(
    mut channel_messages: MessageWriter<SendChannelMessage<PlayerInformation>>,
) {
    channel_messages.write(SendChannelMessage {
        channel_id: JSON_CHANNEL,
        message: PlayerInformation {
            health: 100,
            position: (1, 2, 3),
        },
    });
}

/// send bincode message
pub fn send_bincode_message(
    mut channel_messages: MessageWriter<SendChannelMessage<PlayerInformation>>,
) {
    channel_messages.write(SendChannelMessage {
        channel_id: BINCODE_CHANNEL,
        message: PlayerInformation {
            health: 300,
            position: (4, 5, 6),
        },
    });
}

/// send raw message to server
pub fn send_raw_message_to_channel(
    q_client: Query<(&NetworkNode, &ChannelId), With<ClientNode<WebsocketAddress>>>,
) {
    for (node, channel_id) in q_client.iter() {
        if channel_id == &RAW_CHANNEL {
            node.send_bytes(format!("raw packet to {}", channel_id).as_bytes());
        }
    }
}