GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-06 16:07:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: add SendWithCheck for adb transport

Browse Source
This commit is contained in:
lilong.129
2023-06-23 13:41:29 +08:00
parent 10b4612afa
commit 65cfb86ac5
4 changed files with 19 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hrp/pkg/gadb/transport.go
+12 -5
View File
@@ -27,11 +27,14 @@ type transport struct {
readTimeout time.Duration
}
// newTransport creates a new tcp socket connection
func newTransport(address string, readTimeout ...time.Duration) (tp transport, err error) {
if len(readTimeout) == 0 {
readTimeout = []time.Duration{DefaultAdbReadTimeout}
}
tp.readTimeout = readTimeout[0]
tp = transport{
readTimeout: readTimeout[0],
}
tp.sock, err = net.Dial("tcp", address)
if err == nil {
// dial success
@@ -139,11 +142,15 @@ func (t transport) Close() (err error) {
return t.sock.Close()
}
func (t transport) CreateSyncTransport() (sTp syncTransport, err error) {
if err = t.Send("sync:"); err != nil {
return syncTransport{}, err
func (t transport) SendWithCheck(command string) (err error) {
if err = t.Send(command); err != nil {
return err
}
if err = t.VerifyResponse(); err != nil {
return t.VerifyResponse()
}
func (t transport) CreateSyncTransport() (sTp syncTransport, err error) {
if err = t.SendWithCheck("sync:"); err != nil {
return syncTransport{}, err
}
sTp = newSyncTransport(t.sock, t.readTimeout)