KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Implement streaming mode for socket IO

Browse Source
This commit is contained in:
Kovid Goyal 2022-08-31 20:25:42 +05:30
parent 364533b1ed
commit bd1b3d9f6e
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tools/cmd/at/socket_io.go
View File

@ -55,19 +55,43 @@ func read_response_from_conn(conn *net.Conn, timeout time.Duration) (serialized_
}
func simple_socket_io(conn *net.Conn, io_data *rc_io_data) (serialized_response []byte, err error) {
const (
BEFORE_FIRST_ESCAPE_CODE_SENT = iota
SENDING
WAITING_FOR_RESPONSE
)
state := BEFORE_FIRST_ESCAPE_CODE_SENT
wants_streaming := io_data.rc.Stream
for {
var chunk []byte
chunk, _, err = io_data.next_chunk()
var one_escape_code_done bool
chunk, one_escape_code_done, err = io_data.next_chunk()
if err != nil {
return
}
if len(chunk) == 0 {
state = WAITING_FOR_RESPONSE
break
}
err = write_all_to_conn(conn, chunk)
if err != nil {
return
}
if state == BEFORE_FIRST_ESCAPE_CODE_SENT && one_escape_code_done {
if wants_streaming {
var streaming_response []byte
streaming_response, err = read_response_from_conn(conn, io_data.timeout)
if err != nil {
return
}
if !is_stream_response(streaming_response) {
err = fmt.Errorf("Did not receive expected streaming response")
return
}
}
state = SENDING
}
}
if io_data.rc.NoResponse {
return