我目前正在研究 vendor go balancer 代碼。我需要刪除 tcp 撥號呼叫并在沒有呼叫的情況下模擬成功的連接。在下面的代碼中,有這一行:ds, err := net.Dial("tcp", backend.String());if err != nil {
log.Printf("failed to dial %s: %s", backend, err)
us.Close()
return}它所做的是撥號到 tcp 服務器,然后返回連接響應ds,它在此處定義:https: //golang.org/pkg/net/#Dialds我需要的是在不做的情況下獲得tcp dialer。我正在嘗試在沒有任何實際 tcp 調用的情況下測試負載平衡器。所以,基本上,當我們輸入handleConnectionwed 在 tcp 撥號之前創建一個網絡連接并使用這個網絡連接時,它應該在 tcp 撥號開始之前模擬 100% 的網絡連接。func copy(wc io.WriteCloser, r io.Reader) { defer wc.Close() io.Copy(wc, r)}func handleConnection(us net.Conn, backend BA.Backend) { if backend == nil { log.Printf("no backend available for connection from %s", us.RemoteAddr()) us.Close() return } host, _, _ := net.SplitHostPort(us.RemoteAddr().String()) _, ok := dbAuthTokenData[host]; if !ok { w := bufio.NewWriter(us) w.WriteString("InvalidCredentials") w.Flush() us.Close() return } ds, err := net.Dial("tcp", backend.String());if err != nil { log.Printf("failed to dial %s: %s", backend, err) us.Close() return } // Ignore errors go copy(ds, us) go copy(us, ds)}func tcpBalance(bind string, backends BA.Backends) error { log.Println("using tcp balancing") ln, err := net.Listen("tcp", bind) if err != nil { return fmt.Errorf("failed to bind: %s", err) } log.Printf("listening on %s, balancing %d backends", bind, backends.Len()) for { conn, err := ln.Accept() if err != nil { log.Printf("failed to accept: %s", err) continue } go handleConnection(conn, backends.Choose()) } return err}我嘗試注釋掉go handleConnection(conn, backends.Choose())但失敗了。
1 回答

倚天杖
TA貢獻1828條經驗 獲得超3個贊
您可以重構代碼的模式是創建一個Dialer接口。在您的代碼示例中,您使用返回ds的io.ReadWriteCloser. 所以你不需要實現整個net.Conn接口。就像net.Conn里面的讀寫方法一樣,一切正常
type Dialer interface{
Dial(network, address string) (io.ReadWriteCloser, error)
}
現在讓我們改變你的功能:
func handleConnection(us net.Conn, backend BA.Backend, d Dialer) {
// ...
// Code here stays
ds, err := d.Dial("tcp", backend.String());if err != nil {
log.Printf("failed to dial %s: %s", backend, err)
us.Close()
return
}
// ...
}
你的生產代碼可以工作,你現在需要定義一個netDialer包裝net.Dial()函數的類型。在您的測試中,您可以使用 atestDialer其中使用 a bytes.Buffer。
這回答了你的問題:
無需輸入網絡撥號即可模擬網絡連接
- 1 回答
- 0 關注
- 135 瀏覽
添加回答
舉報
0/150
提交
取消