commons_go/chromedp
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minor change to chromedp-proxy to remove non-identifier characters in log names

Browse Source
This commit is contained in:
Kenneth Shaw 2017-12-19 11:24:22 +07:00
parent 6afa874714
commit 6d2e39b47b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/chromedp-proxy/main.go
View File

@ -21,6 +21,7 @@ import (
"net/url" "net/url"
"os" "os"
"path" "path"
"regexp"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
) )
@ -170,11 +171,15 @@ func checkVersion() ([]byte, error) {
return body, nil return body, nil
} }
var (
cleanRE = regexp.MustCompile(`[^a-zA-Z0-9_\-\.]`)
)
func createLog(id string) (io.Closer, *log.Logger) { func createLog(id string) (io.Closer, *log.Logger) {
var f io.Closer var f io.Closer
var w io.Writer = os.Stdout var w io.Writer = os.Stdout
if !*flagNoLog && *flagLogMask != "" { if !*flagNoLog && *flagLogMask != "" {
l, err := os.OpenFile(fmt.Sprintf(*flagLogMask, id), os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644) l, err := os.OpenFile(fmt.Sprintf(*flagLogMask, cleanRE.ReplaceAllString(id, "")), os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }