迭代golang中的多個返回

Question

我試圖從包含域（未知數量）的文本文件中獲取輸入，然后將每個域用作參數並獲取其服務器類型。 如預期的那樣，這僅返回最后一個域。 如何迭代多個返回值？ 下面是代碼。 //測試包主要

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
}

Answer 1

如果您想“同時”執行此操作，則將返回一個通道，通過該通道將發送您要返回的多個內容：

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
}

Answer 2

如果我理解您的問題，那么您想讀取該文件，以某種方式檢測到該文件已被修改，並且有一種方法可以對客戶端代碼進行這些修改。

那不是文件的工作方式。

您有兩種選擇：

1. 使用某些特定於操作系統的API監聽文件更改-https: //www.linuxjournal.com/content/linux-filesystem-events-inotify

2. 使用無限循環讀取文件。 一次讀取文件。 將副本保存到內存中。 一次又一次地循環讀取同一文件，直到新文件不同於復制並計算增量。

檢查是否有可能使用push而不是pull獲取新域。 控制文件中域名的系統是否可能將數據直接推送給您？

如果loop是唯一可能的選擇，請在兩次文件讀取之間設置一些暫停時間以減少系統負載。

當您設法獲取新域並需要同時處理它們時，請使用@dave 建議的通道。

Answer 3

可能不是最佳解決方案。 但是，我決定擺脫一個單獨的功能，而只是覆蓋更多的領域。 我將在下面的代碼中發布期望的代碼。 現在，我需要解析域，以便根URL和子域僅被掃描一次。

// Main
package main

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

//var Domain string
var Target string


func main() {
    c := colly.NewCollector()

    c.OnError(func(r *colly.Response, err error) {
        fmt.Println("Request URL:", r.Request.URL, "\n Failed with response:", r.StatusCode)
    })

    // Find and visit all links
    c.OnHTML("a", func(e *colly.HTMLElement) {
        e.Request.Visit(e.Attr("href"))
    })

    c.OnRequest(func(r *colly.Request) {
        Domain := r.URL.String()
        Target := BannerGrab(Domain)
        fmt.Println(Domain)
        fmt.Println(Target)
        fmt.Println("Dropping By.. ", r.URL)
        time.Sleep(1000 * time.Millisecond)
    })

    c.Visit("http://www.milliondollarhomepage.com/")
    }

    //CheckDB if not listed else add
    //RiskDB
    //Email


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

}