golang 采集图片

前端之家收集整理的这篇文章主要介绍了golang 采集图片前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

帮朋友忙去采集的,看他打开韩国网站 一个一个点开去保存看不下去了速度太慢了,原本想用PHP写的,刚好看了无闻大大的go采集,然后无耻的改了下就用了

package main

import ( "fmt" "io" "io/IoUtil" "log" "net/http" "os" "path" "regexp" "strings" ) type NotFoundError struct { Message string } func (e NotFoundError) Error() string { return e.Message } type RemoteError struct { Host string Err error } func (e *RemoteError) Error() string { return e.Err.Error() } var UserAgent = "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML,like Gecko) Chrome/29.0.1541.0 Safari/537.36" // HttpGet gets the specified resource. ErrNotFound is returned if the // server responds with status 404. func HttpGet(client *http.Client,url string,header http.Header) (io.ReadCloser,error) { req,err := http.NewRequest("GET",url,nil) if err != nil { return nil,err } req.Header.Set("User-Agent",UserAgent) for k,vs := range header { req.Header[k] = vs } resp,err := client.Do(req) if err != nil { return nil,&RemoteError{req.URL.Host,err} } if resp.StatusCode == 200 { return resp.Body,nil } resp.Body.Close() if resp.StatusCode == 404 { // 403 can be rate limit error. || resp.StatusCode == 403 { err = NotFoundError{"Resource not found: " + url} } else { err = &RemoteError{req.URL.Host,fmt.Errorf("get %s -> %d",resp.StatusCode)} } return nil,err } // HttpGetBytes gets the specified resource. ErrNotFound is returned if the server // responds with status 404. func HttpGetBytes(client *http.Client,header http.Header) ([]byte,error) { rc,err := HttpGet(client,header) if err != nil { return nil,err } defer rc.Close() return IoUtil.ReadAll(rc) } // HttpGetToFile gets the specified resource and writes to file. // ErrNotFound is returned if the server responds with status 404. func HttpGetToFile(client *http.Client,header http.Header,fileName string) error { rc,header) if err != nil { return err } defer rc.Close() os.MkdirAll(path.Dir(fileName),os.ModePerm) f,err := os.Create(fileName) if err != nil { return err } defer f.Close() _,err = io.Copy(f,rc) return err } var img = regexp.MustCompile(`href=\"javascript:goView\((\d+)`) var imgPattern = regexp.MustCompile(`id="mainImage" src=\"../upload(.*?).jpg`) var totalTask int func download(url string,num chan bool) { url = strings.TrimPrefix(url,`href="javascript:goView(`) page := "http://www.gdweb.co.kr/main/koreaWebView.asp?idx=%s&url=koreaWeb.asp" t,err := HttpGetBytes(&http.Client{},fmt.Sprintf(page,url),nil) if err != nil { log.Fatalf("获取页面失败:%v",err) } matches := imgPattern.FindAll(t,-1) for _,match := range matches { url = "http://www.gdweb.co.kr" + strings.TrimPrefix(string(match),`id="mainImage" src="..`) log.Printf("正在下载:%s",url) err := HttpGetToFile(&http.Client{},nil,"pics/"+path.Base(url)) if err != nil { log.Printf("图片下载失败(%s):%v",err) } } totalTask-- <-num } func main() { // 控制同时下载数量 num := make(chan bool,5) // 主线程爬取页面,子线程下载图片 //baseUrl := "http://nvmingxing.net/hotness/%d/" //abaseUrl := "http://www.gdweb.co.kr/main/koreaWebView.asp?idx=8200&url=koreaWeb.asp" baseUrl := "http://www.gdweb.co.kr/main/koreaWeb.asp?idx=&url=index.asp&lpage=124&page=%d" for i := 2; i < 124; i++ { log.Printf("抓取页面:%d",totalTask) data,fmt.Sprintf(baseUrl,i+1),nil) if err != nil { log.Fatalf("获取页面失败(%d):%v",i,err) } matches := img.FindAll(data,match := range matches { totalTask++ num <- true go download(string(match),num) } } }

猜你在找的Go相关文章