diff options
Diffstat (limited to 'extras/hs-test/utils.go')
-rw-r--r-- | extras/hs-test/utils.go | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/extras/hs-test/utils.go b/extras/hs-test/utils.go index d250dc64519..b61ac4271d6 100644 --- a/extras/hs-test/utils.go +++ b/extras/hs-test/utils.go @@ -3,6 +3,7 @@ package main import ( "fmt" "io" + "net" "net/http" "os" "strings" @@ -94,3 +95,25 @@ func newHttpClient() *http.Client { }} return client } + +func tcpSendReceive(address, data string) (string, error) { + conn, err := net.DialTimeout("tcp", address, time.Second*30) + if err != nil { + return "", err + } + defer conn.Close() + err = conn.SetDeadline(time.Now().Add(time.Second * 30)) + if err != nil { + return "", err + } + _, err = conn.Write([]byte(data)) + if err != nil { + return "", err + } + reply := make([]byte, 1024) + _, err = conn.Read(reply) + if err != nil { + return "", err + } + return string(reply), nil +} |