mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
examples: move websocket example from x/
This commit is contained in:
54
examples/websocket/client-server/client.v
Normal file
54
examples/websocket/client-server/client.v
Normal file
@ -0,0 +1,54 @@
|
||||
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
|
||||
}
|
43
examples/websocket/client-server/server.v
Normal file
43
examples/websocket/client-server/server.v
Normal file
@ -0,0 +1,43 @@
|
||||
module main
|
||||
|
||||
import x.websocket
|
||||
import term
|
||||
|
||||
// this server accepts client connections and broadcast all messages to other connected clients
|
||||
fn main() {
|
||||
println('press ctrl-c to quit...')
|
||||
start_server() ?
|
||||
}
|
||||
|
||||
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 {
|
||||
for i, _ in m.clients {
|
||||
mut c := m.clients[i]
|
||||
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()
|
||||
}
|
||||
}
|
96
examples/websocket/ping.v
Normal file
96
examples/websocket/ping.v
Normal file
@ -0,0 +1,96 @@
|
||||
module main
|
||||
|
||||
import time
|
||||
import os
|
||||
import x.websocket
|
||||
|
||||
fn main() {
|
||||
go start_server()
|
||||
time.sleep_ms(100)
|
||||
go start_client()
|
||||
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
|
||||
})?
|
||||
s.on_message(fn (mut ws websocket.Client, msg &websocket.Message) ? {
|
||||
ws.write(msg.payload, msg.opcode) or {
|
||||
panic(err)
|
||||
}
|
||||
})
|
||||
s.on_close(fn (mut ws websocket.Client, code int, reason string) ? {
|
||||
// println('client ($ws.id) closed connection')
|
||||
})
|
||||
s.listen() or {
|
||||
// println('error on server listen: $err')
|
||||
}
|
||||
unsafe {
|
||||
s.free()
|
||||
}
|
||||
}
|
||||
|
||||
fn start_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('open!')
|
||||
})
|
||||
// use on_error_ref if you want to send any reference object
|
||||
ws.on_error(fn (mut ws websocket.Client, err string) ? {
|
||||
println('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('closed')
|
||||
})
|
||||
// use on_message_ref if you want to send any reference object
|
||||
ws.on_message(fn (mut ws websocket.Client, msg &websocket.Message) ? {
|
||||
if msg.payload.len > 0 {
|
||||
message := msg.payload.bytestr()
|
||||
println('client got type: $msg.opcode payload:\n$message')
|
||||
}
|
||||
})
|
||||
// you can add any pointer reference to use in callback
|
||||
// t := TestRef{count: 10}
|
||||
// ws.on_message_ref(fn (mut ws websocket.Client, msg &websocket.Message, r &SomeRef)? {
|
||||
// // println('type: $msg.opcode payload:\n$msg.payload ref: $r')
|
||||
// }, &r)
|
||||
ws.connect() or {
|
||||
println('error on connect: $err')
|
||||
}
|
||||
go write_echo(mut ws) or {
|
||||
println('error on write_echo $err')
|
||||
}
|
||||
ws.listen() or {
|
||||
println('error on listen $err')
|
||||
}
|
||||
unsafe {
|
||||
ws.free()
|
||||
}
|
||||
}
|
||||
|
||||
fn write_echo(mut ws websocket.Client) ? {
|
||||
message := 'echo this'
|
||||
for i := 0; i <= 10; i++ {
|
||||
// Server will send pings every 30 seconds
|
||||
ws.write_str(message) or {
|
||||
println('panicing writing $err')
|
||||
}
|
||||
time.sleep_ms(100)
|
||||
}
|
||||
ws.close(1000, 'normal') or {
|
||||
println('panicing $err')
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user