add examples about goroutine and wait group
This commit is contained in:
parent
f16bfef103
commit
f0c2621469
28
1-echo-text/main.go
Normal file
28
1-echo-text/main.go
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
@Ref: https://livebook.manning.com/book/go-in-practice/chapter-3/65
|
||||||
|
`os.Stdin` and `os.Stdout` will be used as the system input and output. It's great.
|
||||||
|
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
go echo(os.Stdin, os.Stdout)
|
||||||
|
|
||||||
|
time.Sleep(30 * time.Second)
|
||||||
|
|
||||||
|
fmt.Println("Time out.")
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func echo(in io.Reader, out io.Writer) {
|
||||||
|
io.Copy(out, in)
|
||||||
|
}
|
67
2-compress-file/main.go
Normal file
67
2-compress-file/main.go
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"compress/gzip"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
@Ref: https://livebook.manning.com/book/go-in-practice/chapter-3/65
|
||||||
|
1. Practice gzip compress
|
||||||
|
2. How to use wait group
|
||||||
|
3. os.Args[1:] can stream in multiple files to the program without writing a walk function. Command: go run main.go ./testdata/*
|
||||||
|
|
||||||
|
*/
|
||||||
|
func main() {
|
||||||
|
|
||||||
|
// Single thread
|
||||||
|
// for _, file := range os.Args[1:] {
|
||||||
|
|
||||||
|
// err := compress(file)
|
||||||
|
// if err != nil {
|
||||||
|
// os.Exit(0)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
// Multi threads
|
||||||
|
wg := sync.WaitGroup{}
|
||||||
|
var (
|
||||||
|
i int
|
||||||
|
file string
|
||||||
|
)
|
||||||
|
for i, file = range os.Args[1:] {
|
||||||
|
wg.Add(1)
|
||||||
|
go func(filename string) {
|
||||||
|
compress(filename)
|
||||||
|
wg.Done()
|
||||||
|
}(file)
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
fmt.Printf("Compressed %d files\n", i+1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func compress(filename string) error {
|
||||||
|
in, err := os.Open(filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer in.Close()
|
||||||
|
|
||||||
|
out, err := os.Create(filename + ".gz")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer out.Close()
|
||||||
|
|
||||||
|
gzout := gzip.NewWriter(out)
|
||||||
|
_, err = io.Copy(gzout, in)
|
||||||
|
gzout.Close()
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user