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

56 lines
1.5 KiB
V

module main
import net.websocket
import term
fn slog(message string) {
eprintln(term.colorize(term.bright_yellow, message))
}
// this server accepts client connections and broadcast all messages to other connected clients
fn main() {
eprintln('press ctrl-c to quit...')
start_server()!
}
fn start_server() ! {
slog('start_server')
mut s := websocket.new_server(.ip6, 30000, '')
defer {
unsafe {
s.free()
}
}
// 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 {
slog('s.on_connect')
// 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) ! {
slog('s.on_message_ref')
// for _, cli in m.clients {
for i, _ in m.clients {
mut c := m.clients[i] or { continue }
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) ! {
slog('s.on_close')
println(term.green('client (${ws.id}) closed connection'))
})
s.listen() or { println(term.red('error on server listen: ${err}')) }
slog('s.listen done')
}