如何在 Go 中选择 os.Stdin 和 http？

今天golang学习网给大家带来了《如何在 Go 中选择 os.Stdin 和 http？》，其中涉及到的知识点包括等等，无论你是小白还是老手，都适合看一看哦~有好的建议也欢迎大家在评论留言，若是看完有所收获，也希望大家能多多点赞支持呀！一起加油学习~

假设我想接受一只动物。用户可以在终端上的 What type of Animal? 提示符处设置动物类型，也可以访问 http://localhost:1234/animal?type=kitten

无论她做什么，终端都会读取什么类型的动物？ kitten（假设她选择了一只小猫），然后程序将在终端（并且仅终端）上提示用户 小猫的名字是什么？

我的想法是使用通道来执行例程，但是由于两个例程都会陷入函数调用中（终端为 Scan()，http 为 ListenAndServe() ），所以我不清楚如何停止收到输入后仍处于函数调用中的 go 例程。选择频道的正常方法将不起作用。

在 C 中，我只需对两个相关文件描述符（fd 0 和来自 listen() 的 fd）执行 select(2) ，但我不认为这就是 Go 中的做法。 p>

注意：我在下面回答了这个问题。如果您对这个问题投了反对票，我很想知道为什么，因为我会发现我提出的答案很有用。如果您对答案投反对票，我真的很想知道原因。如果它不是惯用的 Go 或者它有我遗漏的其他问题，我真的很想修复它们。

解决方案

好的，我找到了我想做的事情的解决方案。

但这可能不是最惯用的 go 方式。如果通道采用带有输入源和字符串的结构，我可能会放弃选择和网络输入通道。

package main

import (
    "bufio"
    "context"
    "fmt"
    "io"
    "net/http"
    "os"
    "time"
)

func gethttpanimal(input chan string) *http.server {
    srv := &http.server{addr: ":1234"}

    http.handlefunc("/animal", func(w http.responsewriter, r *http.request) {
        animal, ok := r.url.query()["type"]
        if !ok || len(animal[0]) <= 0 {
            io.writestring(w, "animal not understood.\n")
            return
        }
        io.writestring(w, "animal understood.\n")
        input <- string(animal[0])
    })

    go func() {
        if err := srv.listenandserve(); err != nil {
            if err.error() != "http: server closed" {
                fmt.printf("gethttpanimal error with listenandserve: %s", err)
            }
        }
    }()

    return srv
}

func getterminalinput(input chan string) {
    scanner := bufio.newscanner(os.stdin)
    for {
        scanner.scan()
        input <- scanner.text()
    }
}

func main() {
    input := make(chan string)
    webinput := make(chan string)

    srv := gethttpanimal(webinput)
    fmt.print("enter animal type: ")
    go getterminalinput(input)

    var animal string
    select {
    case ta := <-input:
        animal = ta
    case webta := <-webinput:
        animal = webta
        fmt.printf("%s\n", animal)
    }

    ctx, _ := context.withtimeout(context.background(), 5*time.second)
    srv.shutdown(ctx)
    close(webinput)

    fmt.printf("enter animal name: ")
    name := <-input
    fmt.printf("congrats on getting %s the half-%s\n", name, animal)
}

你可以通过 close statement 来停止 goroutine。

示例：

package main

import "sync"
func main() {
    var wg sync.WaitGroup
    wg.Add(1)

    ch := make(chan int)
    go func() {
        for {
            foo, ok := <- ch
            if !ok {
                println("done")
                wg.Done()
                return
            }
            println(foo)
        }
    }()
    ch <- 1
    ch <- 2
    ch <- 3
    close(ch)

    wg.Wait()
}

终于介绍完啦！小伙伴们，这篇关于《如何在 Go 中选择 os.Stdin 和 http？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！