lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-02-20 23:47:21 -05:00
Code Issues Releases Wiki Activity

Feat: vformat supports multi-core processing (#996)

Browse Source
This commit is contained in:
Loyalsoldier 2021-05-11 09:05:43 +08:00 committed by GitHub
parent a771e3401a
commit 218f27c9c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
infra/vformat/main.go
View File

@ -77,31 +77,31 @@ func GetGOBIN() string {
return GOBIN return GOBIN
} }
func Run(binary string, args []string) (string, error) { func Run(binary string, args []string) ([]byte, error) {
cmd := exec.Command(binary, args...) cmd := exec.Command(binary, args...)
cmd.Env = append(cmd.Env, os.Environ()...) cmd.Env = append(cmd.Env, os.Environ()...)
output, cmdErr := cmd.CombinedOutput() output, cmdErr := cmd.CombinedOutput()
if cmdErr != nil { if cmdErr != nil {
return "", cmdErr return nil, cmdErr
} }
if len(output) > 0 { return output, nil
return string(output), nil
}
return "", nil
} }
func RunMany(binary string, args, files []string) { func RunMany(binary string, args, files []string) {
fmt.Println("Processing...") fmt.Println("Processing...")
maxTasks := make(chan struct{}, runtime.NumCPU())
for _, file := range files { for _, file := range files {
args := append(args, file) maxTasks <- struct{}{}
output, err := Run(binary, args) go func(file string) {
output, err := Run(binary, append(args, file))
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
continue } else if len(output) > 0 {
} fmt.Println(string(output))
if len(output) > 0 {
fmt.Println(output)
} }
<-maxTasks
}(file)
} }
} }
@ -139,7 +139,7 @@ func main() {
goimports = goimportsPath goimports = goimportsPath
} }
rawFilesSlice := make([]string, 0) rawFilesSlice := make([]string, 0, 1000)
walkErr := filepath.Walk("./", func(path string, info os.FileInfo, err error) error { walkErr := filepath.Walk("./", func(path string, info os.FileInfo, err error) error {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)