1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

x.websocket: server broadcast plus examples (#7922)

This commit is contained in:
Tomas Hellström
2021-01-06 15:43:54 +01:00
committed by GitHub
parent 42e60b8e27
commit f9a873736e
5 changed files with 112 additions and 5 deletions

View File

@ -0,0 +1,60 @@
module main
import os
import x.websocket
import term
// This client should be compiled an run in different konsoles
// it connects to the server who will broadcast your messages
// to all other connected clients
fn main() {
mut ws := start_client() ?
println(term.green('client $ws.id ready'))
println('Write message and enter to send...')
for {
line := os.get_line()
if line == '' {
break
}
ws.write_str(line)
}
ws.close(1000, 'normal') or {
println(term.red('panicing $err'))
}
unsafe {
ws.free()
}
}
fn start_client() ?&websocket.Client {
mut ws := websocket.new_client('ws://localhost:30000')?
// mut ws := websocket.new_client('wss://echo.websocket.org:443')?
// use on_open_ref if you want to send any reference object
ws.on_open(fn (mut ws websocket.Client) ? {
println(term.green('websocket connected to the server and ready to send messages...'))
})
// use on_error_ref if you want to send any reference object
ws.on_error(fn (mut ws websocket.Client, err string) ? {
println(term.red('error: $err'))
})
// use on_close_ref if you want to send any reference object
ws.on_close(fn (mut ws websocket.Client, code int, reason string) ? {
println(term.green('the connection to the server successfully closed'))
})
// on new messages from other clients, display them in blue text
ws.on_message(fn (mut ws websocket.Client, msg &websocket.Message) ? {
if msg.payload.len > 0 {
message := msg.payload.bytestr()
println(term.blue('$message'))
}
})
ws.connect() or {
println(term.red('error on connect: $err'))
}
go ws.listen() or {
println(term.red('error on listen $err'))
}
return ws
}

View File

@ -0,0 +1,48 @@
module main
import os
import x.websocket
import term
// this server accepts client connections and broadcast all messages to other connected clients
fn main() {
go start_server()
println('press enter to quit...')
os.get_line()
}
fn start_server() ? {
mut s := websocket.new_server(30000, '')
// Make that in execution test time give time to execute at least one time
s.ping_interval = 100
s.on_connect(fn (mut s websocket.ServerClient) ?bool {
// Here you can look att the client info and accept or not accept
// just returning a true/false
if s.resource_name != '/' {
return false
}
return true
})?
// on_message_ref, broadcast all incoming messages to all clients except the one sent it
s.on_message_ref(fn (mut ws websocket.Client, msg &websocket.Message, mut m websocket.Server) ? {
for _, cli in m.clients {
mut c := cli
if c.client.state == .open && c.client.id != ws.id {
c.client.write(msg.payload, websocket.OPCode.text_frame) or {
panic(err)
}
}
}
}, s)
s.on_close(fn (mut ws websocket.Client, code int, reason string) ? {
println(term.green('client ($ws.id) closed connection'))
})
s.listen() or {
println(term.red('error on server listen: $err'))
}
unsafe {
s.free()
}
}