Skip to content

Instantly share code, notes, and snippets.

@kaneshin
Created January 29, 2017 15:27
Show Gist options
  • Save kaneshin/a398720b8e20722a83bc6903e4017435 to your computer and use it in GitHub Desktop.
Save kaneshin/a398720b8e20722a83bc6903e4017435 to your computer and use it in GitHub Desktop.
package main
import (
"bufio"
"context"
"flag"
"fmt"
"os"
"os/signal"
"sync"
"syscall"
"time"
)
var (
args []string
)
func init() {
flag.Parse()
args = flag.Args()
if len(args) == 0 {
fmt.Fprintln(os.Stderr, "Missing filename: specify one argument")
os.Exit(1)
}
}
func tail(c context.Context, name string) error {
fi, err := os.Open(name)
if err != nil {
return err
}
defer fi.Close()
const size = 10
ch := make(chan string, size)
// prepare
scanner := bufio.NewScanner(fi)
for scanner.Scan() {
if len(ch) >= size {
<-ch
}
ch <- scanner.Text()
}
// output
go func() {
for {
t, ok := <-ch
if !ok {
return
}
fmt.Fprintln(os.Stdout, t)
}
}()
var mu sync.Mutex
go func() {
for {
if ch == nil {
return
}
scanner := bufio.NewScanner(fi)
for scanner.Scan() {
mu.Lock()
ch <- scanner.Text()
mu.Unlock()
}
time.Sleep(time.Millisecond * 10)
}
}()
<-c.Done()
mu.Lock()
close(ch)
ch = nil
mu.Unlock()
return nil
}
func main() {
var wg sync.WaitGroup
c, cancel := context.WithCancel(context.Background())
for _, n := range args {
wg.Add(1)
go func() {
defer wg.Done()
tail(c, n)
}()
}
sig := make(chan os.Signal, 1)
signal.Notify(sig, syscall.SIGINT)
go func() {
for {
s := <-sig
switch s {
case syscall.SIGINT:
cancel()
}
}
}()
wg.Wait()
time.Sleep(time.Millisecond * 300)
}
@kaneshin
Copy link
Author

Execute

$ go run main.go -- [FILE]...

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment