分享基于phantomjs的web截图

main.go

<pre name="code" class="plain">package mainimport (    "encoding/json"    "errors"    "fmt"    "io/ioutil"    "net/http"    "os/exec"    "path"    "runtime"    "strconv"    "strings"    "sync"    "time")type App struct {    Domain string    Post   int    ScreenJS   string    ScreenPath string    Static     string}type Config struct {    Resource Resource    RootDir  string //项目根目录    App      *App    Log      *log.Config}var curDir stringfunc init() {    _, filename, _, _ := runtime.Caller(1)    curDir = path.Dir(filename)}var rootCfg Configfunc conf() Config {    if rootCfg.RootDir == "" {        rootCfg.RootDir = path.Join(curDir, "./")        var f = rootCfg.RootDir + "/dsp_screen.json"        data, err := ioutil.ReadFile(f)        if err != nil {            panic(err)        }        err = json.Unmarshal(data, &rootCfg)        if err != nil {            panic(err)        }    }    return rootCfg}var dbs = make(map[string]*mongo.Mdb)var mutex sync.Mutexconst shortFormat = "20060102"//collectionconst (    ScreenCollection = "Screen")//id_generatorconst (    idGenerator = "id_generator")type Screen struct {    Id         int64  `bson:"_id"`    Url        string `bson:"Url"`    UserAgent  string `bson:"UserAgent"`    Path       string `bson:"Path"`    CreateTime int64  `bson:"CreateTime"`}// func (this *Screen) Init() {//     id := nextId(ScreenCollection)//     this.Id = id//     this.CreateTime = time.Now().Unix()// }// func (this *Screen) IsInit() bool {//     return this.Id != 0// }// func (this *Screen) Save() (err error) {//     return appMdb().Insert(ScreenCollection, this)// }func screen(rw http.ResponseWriter, req *http.Request) {    defer req.Body.Close()    data, err := ioutil.ReadAll(req.Body)    if err != nil {        WriteError(rw, err.Error())        return    }    body := make(map[string]string)    err = json.Unmarshal(data, &body)    if err != nil {        WriteError(rw, err.Error())        return    }    screen := new(Screen)    if url, ok := body["Url"]; ok {        screen.Url = url    }    if data, ok := body["Data"]; ok {        screen.UserAgent = fmt.Sprintf("Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/38.0.2125.111 Chrome/38.0.2125.111 Safari/537.36?%s",            data)    }    t := time.Now()    screen.Path = fmt.Sprintf("%s/%d.png", t.Format(shortFormat), t.UnixNano())    path := fmt.Sprintf("%s/%s", conf().App.ScreenPath, screen.Path)    ph := exec.Command("phantomjs", conf().RootDir+"/"+conf().App.ScreenJS, screen.Url, screen.UserAgent, path)    out, err := ph.Output()    if err != nil {        log.Error(err)        WriteError(rw, err.Error())        return    }    if strings.Contains(string(out), "false") {        log.Error(string(out))        WriteError(rw, "截图失败")        return    }    // screen.Init()    // err = screen.Save()    // if err != nil {    //     log.Error(err)    //     WriteError(rw, err.Error())    //     return    // }    result := make(map[string]string)    result["url"] = conf().App.Domain + "/" + conf().App.Static + "/" + screen.Path    WriteResult(rw, result)}func image(rw http.ResponseWriter, req *http.Request) {    staticHandler := http.FileServer(http.Dir(conf().RootDir))    staticHandler.ServeHTTP(rw, req)    return}func WriteError(rw http.ResponseWriter, msg string) {    rw.Write([]byte(`{"success":false,"message":"` + msg + `"}`))}func WriteResult(rw http.ResponseWriter, result interface{}) {    data, err := json.Marshal(result)    if err != nil {        WriteError(rw, err.Error())    } else {        rw.Write([]byte(fmt.Sprintf(`{"success":false,"result":%s}`, string(data))))    }}func handler(rw http.ResponseWriter, req *http.Request) {    if strings.Contains(req.URL.String(), conf().App.Static) {        image(rw, req)    } else {        screen(rw, req)    }}func main() {    err := log.InitConf(conf().Log)    if err != nil {        panic(err)    }    http.HandleFunc("/", handler)    fmt.Println("监听端口", conf().App.Post)    err = http.ListenAndServe(fmt.Sprintf(":%d", conf().App.Post), nil)    if err != nil {        log.Error(err)        panic(err)    }}

phantomjs.js

<pre name="code" class="javascript">var page = require('webpage').create();page.paperSize = {width:'5in',height:'7in'};//args//code word directoryif (phantom.args.length < 3) {console.log("{'success': false, 'result': '参数不正确'}");phantom.exit();};var args = phantom.args;var url = args[0];var userAgent  = args[1];var path = args[2];page.settings.userAgent = userAgent;page.settings.viewportSize = { width: 1920, height: 1080 };//设置超时// page.settings.resourceTimeout = 10000;page.viewportSize = { width: 1920, height : 1080 };page.open(url, function(status) {if (status !== 'success') {console.log("{'success': false, 'result': '请求出错'}");} else {page.render(path)console.log("{'success': true, 'result': '"+path+"'}");}phantom.exit();});

配置文件

<pre name="code" class="javascript">{    "App": {    "Domain":"http://localhost:9988",    "Post":9988,        "ScreenJS":"phantom.js",        "ScreenPath":"/data/screen/screen",        "Static":"image"    }}

伸手党可能有点烦，因为还是需要改一下才能运行的。主要就是一个网页截图以及展示图片。