亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

無需輸入網絡撥號即可模擬網絡連接

無需輸入網絡撥號即可模擬網絡連接

Go
一只斗牛犬 2023-04-24 16:24:29
我目前正在研究 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。


這回答了你的問題:


無需輸入網絡撥號即可模擬網絡連接


查看完整回答
反對 回復 2023-04-24
  • 1 回答
  • 0 關注
  • 135 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號