首頁(yè) 猿問(wèn) golang：使用 gin...

golang：使用 gin 路由器為 net.Conn 服務(wù)

Go

翻翻過(guò)去那場(chǎng)雪 2022-11-08 16:14:23

我有一個(gè)處理傳入 TCP 連接的函數(shù)：func Handle(conn net.Conn) error { // ...}另外，我有一個(gè)已初始化的帶有已實(shí)現(xiàn)句柄的 gin 路由器：router := gin.New()router.GET(...)router.POST(...)該router.Run(addr)調(diào)用將在addr.有沒(méi)有辦法在Handle不運(yùn)行單獨(dú)的 HTTP 服務(wù)器的情況下使用此路由器處理函數(shù)內(nèi)部的傳入連接？

查看完整描述

2 回答

開(kāi)心每一天1111

TA貢獻(xiàn)1836條經(jīng)驗(yàn) 獲得超13個(gè)贊

創(chuàng)建一個(gè)net.Listener實(shí)現(xiàn)，通過(guò)在通道上接收來(lái)接受連接：

type listener struct {

ch chan net.Conn

addr net.Addr

}

// newListener creates a channel listener. The addr argument

// is the listener's network address.

func newListener(addr net.Addr) *listener {

return &listener{

ch: make(chan net.Conn),

addr: addr,

}

func (l *listener) Accept() (net.Conn, error) {

c, ok := <-l.ch

if !ok {

return nil, errors.New("closed")

}

return c, nil

}

func (l *listener) Close() error { return nil }

func (l *listener) Addr() net.Addr { return l.addr }

通過(guò)發(fā)送到通道來(lái)處理連接：

func (l *listener) Handle(c net.Conn) error {

l.ch <- c

return nil

}

以下是如何將它們聯(lián)系在一起：

創(chuàng)建監(jiān)聽(tīng)器：

s := newListener(someAddr)

像往常一樣配置 Gin 引擎。

router := gin.New()

router.GET(...)

router.POST(...)

使用通道監(jiān)聽(tīng)器和 Gin 引擎在 goroutine 中運(yùn)行 net/http 服務(wù)器：

err := http.Serve(s, router)

if err != nil {

// handle error

}

在您的調(diào)度代碼中，調(diào)用將s.Handle(c)連接傳遞到 net/http 服務(wù)器，然后傳遞到 Gin 引擎。

反對(duì) 回復(fù) 2022-11-08

DIEA

TA貢獻(xiàn)1820條經(jīng)驗(yàn) 獲得超3個(gè)贊

對(duì)于那些有類(lèi)似任務(wù)的人 - 使用單個(gè)路由器處理來(lái)自多個(gè)端口的 TCP 連接，這是我最終找到的解決方法。我沒(méi)有在端口上運(yùn)行 HTTP 服務(wù)器，而是使用 UNIX 套接字運(yùn)行它router.RunUnix(socketName)。完整的解決方案包括三個(gè)步驟：

運(yùn)行 HTTP 服務(wù)器以使用 UNIX 套接字進(jìn)行偵聽(tīng)router.RunUnix(socketName)
在Handle函數(shù)內(nèi)部，從連接中讀取傳入的字節(jié)并將它們發(fā)送到套接字。之后，從套接字讀取 HTTP 響應(yīng)并將其寫(xiě)入連接。
關(guān)閉連接。