使用Golang遍历多个返回值

在 Go 语言中，遍历多个返回值时，如果需要同时执行，可以利用通道机制。通道是一个并发原语，它允许协程之间安全地交换数据。通过创建通道并将其作为返回值，可以将多个值同时返回给调用方。 在提供的代码中，`DomainGrab` 函数使用通道返回从文件中读取的域名。该函数启动一个协程来读取文件并通过通道发送域名。调用方可以使用范围遍历从通道中接收域名，从而实现遍历多个返回值的目的。

我试图从包含域（未知数量）的文本文件中获取输入，然后使用每个域作为参数并获取它们的服务器类型。正如预期的那样，这仅返回最后一个域。如何迭代多个返回值？ 下面是代码。 // 测试 包主要

import (
    "bufio"
    "time"
    "os"
    "fmt"
    "net/http"
    //"github.com/gocolly/colly"
)

var Domain string
var Target string

func main() {
    Domain := DomainGrab()
    Target := BannerGrab(Domain)
    //CheckDB if not listed then add else skip
    //RiskDB
    //Email
    fmt.Println(Domain)
    fmt.Println(Target)
}

func BannerGrab(s string) string {

    client := &http.Client{}
    req, err := http.NewRequest("GET", s, nil)
    if err != nil {
    log.Fatalln(err)
    }
    req.Header.Set("User-Agent", "Authac/0.1")
    resp, _ := client.Do(req)
    serverEntry := resp.Header.Get("Server")
    return serverEntry

}
func DomainGrab() string {

    //c := colly.NewCollector()
// Open the file.
    f, _ := os.Open("domains.txt")
    defer f.Close()
    // Create a new Scanner for the file.
    scanner := bufio.NewScanner(f)
    // Loop over all lines in the file and print them.
    for scanner.Scan() {
        line := scanner.Text()
        time.Sleep(2 * time.Second)
        //fmt.Println(line)
        return line
    }
    return Domain
}

解决方案

如果您想“同时”执行此操作，您将返回一个通道，您将通过该通道发送您想要返回的多个内容：

https://play.golang.org/p/iYBGPwfYLYR

func DomainGrab() <-chan string {
    ch := make(chan string, 1)
    f, _ := os.Open("domains.txt")
    defer f.Close()
    scanner := bufio.NewScanner(f)
    go func() {
    // Loop over all lines in the file and print them.
        for scanner.Scan() {
            line := scanner.Text()
            time.Sleep(2 * time.Second)
            ch <- line
        }
        close(ch)
    }()
    return ch
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~