22

Go でリスニング サーバーを適切に停止する方法を見つけようとしています。ブロックのためlisten.Accept、リスニング ソケットを閉じて終了を知らせる必要がありますが、関連するエラーがエクスポートされないため、他のエラーとは別にそのエラーを伝えることはできません。

これよりうまくやれるだろうか？FIXME以下のコードを参照してくださいserve()

package main

import (
    "io"
    "log"
    "net"
    "time"
)

// Echo server struct
type EchoServer struct {
    listen net.Listener
    done   chan bool
}

// Respond to incoming connection
//
// Write the address connected to then echo
func (es *EchoServer) respond(remote *net.TCPConn) {
    defer remote.Close()
    _, err := io.Copy(remote, remote)
    if err != nil {
        log.Printf("Error: %s", err)
    }
}

// Listen for incoming connections
func (es *EchoServer) serve() {
    for {
        conn, err := es.listen.Accept()
        // FIXME I'd like to detect "use of closed network connection" here
        // FIXME but it isn't exported from net
        if err != nil {
            log.Printf("Accept failed: %v", err)
            break
        }
        go es.respond(conn.(*net.TCPConn))
    }
    es.done <- true
}

// Stop the server by closing the listening listen
func (es *EchoServer) stop() {
    es.listen.Close()
    <-es.done
}

// Make a new echo server
func NewEchoServer(address string) *EchoServer {
    listen, err := net.Listen("tcp", address)
    if err != nil {
        log.Fatalf("Failed to open listening socket: %s", err)
    }
    es := &EchoServer{
        listen: listen,
        done:   make(chan bool),
    }
    go es.serve()
    return es
}

// Main
func main() {
    log.Println("Starting echo server")
    es := NewEchoServer("127.0.0.1:18081")
    // Run the server for 1 second
    time.Sleep(1 * time.Second)
    // Close the server
    log.Println("Stopping echo server")
    es.stop()
}

これは印刷します

2012/11/16 12:53:35 Starting echo server
2012/11/16 12:53:36 Stopping echo server
2012/11/16 12:53:36 Accept failed: accept tcp 127.0.0.1:18081: use of closed network connection

メッセージを非表示にしたいのですが、報告できるAccept failed他のエラーをマスクしたくないことは明らかです。Acceptもちろん、エラーテストを調べることもできますuse of closed network connectionが、それは本当に醜いでしょう。閉じようとしているというフラグを設定し、それが設定されている場合はエラーを無視することができます-より良い方法はありますか?

4

4 に答える 4

14

接続を閉じる前に es.done を使用して信号を送信することで、これを処理します。次のコードに加えて、make(chan bool, 1) を使用して es.done を作成し、ブロックせずに 1 つの値を入れることができるようにする必要があります。

// Listen for incoming connections
func (es *EchoServer) serve() {
  for {
    conn, err := es.listen.Accept()
    if err != nil {
      select {
      case <-es.done:
        // If we called stop() then there will be a value in es.done, so
        // we'll get here and we can exit without showing the error.
      default:
        log.Printf("Accept failed: %v", err)
      }
      return
    }
    go es.respond(conn.(*net.TCPConn))
  }
}

// Stop the server by closing the listening listen
func (es *EchoServer) stop() {
  es.done <- true   // We can advance past this because we gave it buffer of 1
  es.listen.Close() // Now it the Accept will have an error above
}
于 2012-11-16T15:38:23.897 に答える
8

accept()呼び出しの直後にループ内の「停止する時間です」フラグを確認しmain、それを から反転してから、リッスン ポートに接続して、サーバー ソケットの「スタックを解除」します。これは、古い「セルフパイプ トリック」に非常によく似ています。

于 2012-11-16T13:17:29.943 に答える
2

この場合、これらの行のいずれかが機能する可能性があります。

// Listen for incoming connections
func (es *EchoServer) serve() {
        for {
                conn, err := es.listen.Accept()
                if err != nil {
                    if x, ok := err.(*net.OpError); ok && x.Op == "accept" { // We're done
                            log.Print("Stoping")
                            break
                    }

                    log.Printf("Accept failed: %v", err)
                    continue
                }
                go es.respond(conn.(*net.TCPConn))
        }
        es.done <- true
}
于 2012-11-16T13:17:49.813 に答える
-6

ローカル開発に適した簡単な方法を次に示します。

http://www.sergiotapia.me/how-to-stop-your-go-http-server/

package main

import (  
    "net/http"
    "os"

    "github.com/bmizerany/pat"
)

var mux = pat.New()

func main() {  
    mux.Get("/kill", http.HandlerFunc(kill))
    http.Handle("/", mux)
    http.ListenAndServe(":8080", nil)
}

func kill(w http.ResponseWriter, r *http.Request) {  
    os.Exit(0)
}
于 2014-06-15T01:05:44.107 に答える