mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-02 17:50:39 +00:00
ac9c011475
- Add store of pipes in the app - Add pipe type, handling impedance mismatch, used in app and probe. - App <-> Probe pipes have their own websockets. - Add pipe websocket endpoint in app. - Pipe IDs are strings, lose the request/response IDs, and give the json encoder lowercase field names. - Add simple golang ws client, for testing. - Pipe lifecycle plumbing. - Ref count and timeout both ends of pipes in the app - Deal with POST /api/pipe/:pid?_method=delete - Add end-to-end unit test for pipes. - Add test for timing out pipes. - Update go-docker client to tomwilkie/go-dockerclient - Backend work for non-raw ttys - Close pipes when they close themselves in the probe - Ensure all http connections are done before returning from client.Stop()
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/gorilla/websocket"
|
|
//"golang.org/x/crypto/ssh/terminal"
|
|
"github.com/davecgh/go-spew/spew"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
log.Fatal("must specify url")
|
|
}
|
|
url := os.Args[1]
|
|
log.Printf("Connecting to %s", url)
|
|
|
|
//oldState, err := terminal.MakeRaw(0)
|
|
//if err != nil {
|
|
// panic(err)
|
|
//}
|
|
//defer terminal.Restore(0, oldState)
|
|
|
|
dialer := websocket.Dialer{}
|
|
conn, _, err := dialer.Dial(url, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer conn.Close()
|
|
readQuit := make(chan struct{})
|
|
writeQuit := make(chan struct{})
|
|
|
|
// Read-from-UI loop
|
|
go func() {
|
|
defer close(readQuit)
|
|
for {
|
|
_, buf, err := conn.ReadMessage() // TODO type should be binary message
|
|
if err != nil {
|
|
log.Printf("Error reading websocket: %v", err)
|
|
return
|
|
}
|
|
|
|
spew.Dump(buf)
|
|
|
|
if _, err := os.Stdout.Write(buf); err != nil {
|
|
log.Printf("Error writing stdout: %v", err)
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Write-to-UI loop
|
|
go func() {
|
|
defer close(writeQuit)
|
|
buf := make([]byte, 1024)
|
|
for {
|
|
n, err := os.Stdin.Read(buf)
|
|
if err != nil {
|
|
log.Printf("Error reading stdin: %v", err)
|
|
return
|
|
}
|
|
|
|
if err := conn.WriteMessage(websocket.BinaryMessage, buf[:n]); err != nil {
|
|
log.Printf("Error writing websocket: %v", err)
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
// block until one of the goroutines exits
|
|
// this convoluted mechanism is to ensure we only close the websocket once.
|
|
select {
|
|
case <-readQuit:
|
|
case <-writeQuit:
|
|
}
|
|
}
|