Skip to content

Commit

Permalink
request: 猬嗭笍 Content-Encoding [ci skip]
Browse files Browse the repository at this point in the history
  • Loading branch information
iawia002 committed May 17, 2018
1 parent 9ed0424 commit 79d959d
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions request/request.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package request

import (
"compress/gzip"
"compress/zlib"
"fmt"
"io"
"io/ioutil"
Expand Down Expand Up @@ -123,13 +124,17 @@ func Get(url, refer string) string {
headers["Referer"] = refer
}
res := Request("GET", url, nil, headers)
defer res.Body.Close()
var reader io.ReadCloser
if res.Header.Get("Content-Encoding") == "gzip" {
switch res.Header.Get("Content-Encoding") {
case "gzip":
reader, _ = gzip.NewReader(res.Body)
} else {
case "deflate":
reader, _ = zlib.NewReader(res.Body)
default:
reader = res.Body
}
defer res.Body.Close()
defer reader.Close()
body, _ := ioutil.ReadAll(reader)
return string(body)
}
Expand All @@ -140,7 +145,6 @@ func Headers(url, refer string) http.Header {
"Referer": refer,
}
res := Request("GET", url, nil, headers)
defer res.Body.Close()
return res.Header
}

Expand Down

0 comments on commit 79d959d

Please sign in to comment.