You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
991 B
Go
56 lines
991 B
Go
4 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/gorilla/websocket"
|
||
|
|
"github.com/jordanorelli/blammo"
|
||
|
|
)
|
||
|
|
|
||
|
|
type client struct {
|
||
|
|
*blammo.Log
|
||
|
|
host string
|
||
|
|
port int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *client) run() {
|
||
|
|
dialer := websocket.Dialer{
|
||
|
|
HandshakeTimeout: 3 * time.Second,
|
||
|
|
ReadBufferSize: 32 * 1024,
|
||
|
|
WriteBufferSize: 32 * 1024,
|
||
|
|
Subprotocols: []string{"astrodomu@v0"},
|
||
|
|
}
|
||
|
|
|
||
|
|
path := url.URL{
|
||
|
|
Host: fmt.Sprintf("%s:%d", c.host, c.port),
|
||
|
|
Scheme: "ws",
|
||
|
|
Path: "/",
|
||
|
|
}
|
||
|
|
|
||
|
|
conn, res, err := dialer.Dial(path.String(), nil)
|
||
|
|
if err != nil {
|
||
|
|
c.Error("dial error: %v", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
c.Debug("dial response status: %d", res.StatusCode)
|
||
|
|
for k, vals := range res.Header {
|
||
|
|
c.Debug("dial response header: %s = %s", k, strings.Join(vals, ","))
|
||
|
|
}
|
||
|
|
|
||
|
|
for {
|
||
|
|
w, err := conn.NextWriter(websocket.TextMessage)
|
||
|
|
if err != nil {
|
||
|
|
c.Error("unable to get a websocket frame writer: %v", err)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
w.Write([]byte("hey"))
|
||
|
|
w.Close()
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
}
|
||
|
|
}
|