go 协程请求数据

The Go Programming Language

Go Projects

This topic created in 1239 days ago, the information mentioned may be changed or developed.

我一个接口里边，需要多次操作数据库，现在想通过协程来实现。下边这个 demo 是一个样例，我在 resp 怎么区分协程的返回值，这里边还有些不太清晰。

func main() {
    var ch = make(chan []byte)
    go request("https://www.baidu.com", ch)
    go request("https://www.taobao.com", ch)
    go request("https://www.google.com", ch)

    i := 0
    timer := time.NewTimer(time.Second * 3)
L:
    for {
        select {
        case <-timer.C:
            break L
        case resp := <-ch:
            fmt.Printf("%d\n", len(resp))
            // 接收 3 个之后关闭 chan
            i++
            if i == 3 {
                close(ch)
                break L
            }
        }
    }
    // more code...
}

resp

request

协程

timer

7 replies • 2023-01-16 15:48:16 +08:00

catVSdog

Jan 15, 2023

你是要区分每个协程的返回值吗？不同的协程对应不同处理方式？

那么再构建一个一个结构体，例如：

type CRes struct {

}

catVSdog

Jan 15, 2023

你是要区分每个协程的返回值吗？不同的协程(或者说：不同的数据库)对应不同处理方式？

那么再构建一个一个结构体，例如：

type CRes struct {
ChannelType int // taobao:1, baidu:2, google:3
Res interfact{}
}

ch := make(chan CRes)

每个协程函数返回 CRes ，然后将 CRes 返回到 channel 中。在接受、处理 channel 的函数中，根据不同的 channelType 进行处理

awanganddong

Jan 15, 2023

明白了，谢谢你。

awanganddong

Jan 15, 2023

我修改后的 demo 发下

package main

import (
"fmt"
"time"
)

type CRes struct {
ChannelType int
Res interface{}
}

func main() {
var ch = make(chan CRes)
go test1(ch)
go test2(ch)
for i := 0; i < 2; i++ {
select {
case <-time.After(time.Second * 5):
fmt.Println("超时操作")
break
case resp := <-ch:
if resp.ChannelType == 1 {
fmt.Printf("这是 1 的数据")
}
if resp.ChannelType == 2 {
fmt.Printf("这是 2 的数据")
}
break
default:
fmt.Println("资源已满，请稍后再试")
}
}
}
func test1(ch chan CRes) {
res := CRes{
ChannelType: 1,
Res: nil,
}
ch <- res
}

func test2(ch chan CRes) {
res := CRes{
ChannelType: 2,
Res: nil,
}
ch <- res
}

jorneyr

Jan 15, 2023

md5(url) 作为任务的 ID ，返回的时候也对应的带上这个 ID 。

morty0

Jan 16, 2023

context 带个 id 进去

ryan961

Jan 16, 2023

没必要这么复杂吧，直接用 errgroup 。最近有个协程的库看起来还可以 https://github.com/sourcegraph/conc