module main import ( ws eventbus time readline term benchmark ) const ( eb = eventbus.new() ) #flag -I $PWD #include "utf8.h" fn C.utf8_validate_str() bool fn main(){ //println(sss) /* for sss in 0..10 { mut bm := benchmark.new_benchmark() for i in 0..10000 { for a, t in tests { ss := ws.utf8_validate(t.str, t.len) if !ss { panic("failed") } //println("${a}:${ss}") } } bm.stop() println( bm.total_message('remarks about the benchmark') ) } */ mut websocket := ws.new("ws://localhost:9001/getCaseCount") //defer { } websocket.subscriber.subscribe("on_open", on_open) websocket.subscriber.subscribe("on_message", on_message) websocket.subscriber.subscribe("on_error", on_error) websocket.subscriber.subscribe("on_close", on_close) //go websocket.connect() websocket.read() //time.usleep(2000000) //go websocket.listen() //term.erase_clear() /* text := read_line("[client]:") if text == "close" { ws.close(1005, "done") time.usleep(1000000) exit(0) } ws.write(text, .text_frame) */ /* time.usleep(1000000) ws.read() */ //ws.close(1005, "done") /* */ //websocket.close(1005, "done") //read_line("wait") } fn read_line(text string) string { mut r := readline.Readline{} mut output := r.read_line(text + " ") or { panic(err) } output = output.replace("\n","") if output.len <= 0 { return "" } return output } fn on_open(params eventbus.Params){ println("Websocket opened.") } fn on_message(params eventbus.Params){ println("Message recieved. Sending it back.") typ := params.get_string("type") len := params.get_int("len") mut ws := params.get_caller(ws.Client{}) if typ == "string" { message := params.get_string("payload") if ws.uri.ends_with("getCaseCount") { num := message.int() ws.close(1005, "done") start_tests(mut ws, num) return } //println("Message: " + message) ws.write(message.str, len, .text_frame) } else { println("Binary message.") message := params.get_raw("payload") ws.write(message, len, .binary_frame) } } fn start_tests(websocket mut ws.Client, num int) { for i := 1; i < num; i++ { println("Running test: " + i.str()) websocket.uri = "ws://localhost:9001/runCase?case=${i.str()}&agent=vws/1.0a" if websocket.connect() >= 0 { websocket.listen() } } println("Done!") websocket.uri = "ws://localhost:9001/updateReports?agent=vws/1.0a" if websocket.connect() >= 0 { websocket.read() websocket.close(1000, "disconnecting...") } exit(0) } fn on_close(params eventbus.Params){ println("Websocket closed.") } fn on_error(params eventbus.Params){ println("we have an error.") }