Golang handlefunc with channel

时间:2015-02-09 21:37:27

标签: go

我认为这个问题之前已被问过(可能不止一次),但我找不到它......

我正在学习Go,我想通过向“处理程序”发送一个频道来扩展经典的Web服务器示例。

我有这个标准的东西:

func hello(w http.ResponseWriter, r *http.Request) {
   io.WriteString(w, "Hello world!")
}

func main() {
    http.HandleFunc("/", hello)
    http.ListenAndServe(":8000", nil)
}

现在我希望“hello”功能能够在频道上写东西,让某人消费......我使用“普通”功能的方法是创建一个频道:

c := make(chan string) 

并在函数调用中传递c。类似的东西:

dosomething(c)

但是......如果我想要“hello”访问频道c,我该如何去做呢?

2 个答案:

答案 0 :(得分:16)

还有其他两种方法(除了在上一个答案中导出频道)。

第一种是使用函数返回另一个处理函数。返回函数时,它将在通道周围创建一个闭包。

func makeHello(logger chan string) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        logger <- r.Host
        io.WriteString(w, "Hello world!")
    }
}

第二种是使用一个将通道保存为成员的结构,并使用指针接收器方法来处理请求......

type DataPasser struct {
    logs chan string
}

func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
    p.logs <- r.URL.String()
    io.WriteString(w, "Hello world")
}

这是一个完整的工作示例(只需点击/ 1和/ 2即可查看两个示例)

package main

import (
    "fmt"
    "io"
    "net/http"
)

func main() {
    // METHOD 1
    logs := make(chan string)
    go logLogs(logs)
    handleHello := makeHello(logs)

    // METHOD 2
    passer := &DataPasser{logs: make(chan string)}
    go passer.log()

    http.HandleFunc("/1", handleHello)
    http.HandleFunc("/2", passer.handleHello)
    http.ListenAndServe(":9999", nil)
}

// METHOD 1

func makeHello(logger chan string) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        logger <- r.Host
        io.WriteString(w, "Hello world!")
    }
}

func logLogs(logger chan string) {
    for item := range logger {
        fmt.Println("1. Item", item)
    }
}

// METHOD 2

type DataPasser struct {
    logs chan string
}

func (p *DataPasser) handleHello(w http.ResponseWriter, r *http.Request) {
    p.logs <- r.URL.String()
    io.WriteString(w, "Hello world")
}

func (p *DataPasser) log() {
    for item := range p.logs {
        fmt.Println("2. Item", item)
    }
}

答案 1 :(得分:2)

有几种方法可以解决这个问题,最简单的方法是在一个包中定义一个导出的通道,然后在你想要使用该通道的地方导入所述包。

package mychannel

var Chan = make(chan string)