我正在嘗試從包含域（未知數量）的文本文件中獲取輸入，然后將每個域用作參數并獲取其服務器類型。正如預期的那樣，這只會返回最后一個域。如何迭代多個返回值？下面是代碼。// 測試包主import ( "bufio" "time" "os" "fmt" "net/http" //"github.com/gocolly/colly")var Domain stringvar Target stringfunc 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}
3 回答
肥皂起泡泡
TA貢獻1829條經驗 獲得超6個贊
如果我理解你的問題，你想讀取文件，以某種方式檢測到該文件被修改，并有一個方法將這些修改發送到客戶端代碼。
這不是文件的工作方式。
你有兩個選擇：
使用一些操作系統特定的 API 監聽文件更改?
使用無限循環讀取文件。讀取文件一次。將副本保存到內存中。在循環中一次又一次地讀取同一個文件，直到新文件與副本不同并計算增量。
檢查是否可以使用它push來代替pull獲取新域。難不成控制文件域名的系統會直接給你推送數據？
如果loop是唯一可能的選項，請在文件讀取之間設置一些暫停時間以減少系統負載。
RISEBY
TA貢獻1856條經驗 獲得超5個贊
如果您想“同時”執行此操作，您將返回一個通道，您將通過該通道發送您想要返回的多個內容：
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
}
隔江千里
//Email
TA貢獻1906條經驗 獲得超10個贊
可能不是最好的解決方案。但是，我決定一起擺脫一個單獨的功能，以覆蓋更多領域。我會在下面發布我期望的代碼。現在，我需要解析域，以便只掃描一次根 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
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
}
- 3 回答
- 0 關注
- 159 瀏覽
添加回答
舉報
0/150
提交
取消