add timeout pkg

timeout/run.go

@@ -0,0 +1,33 @@
+package timeout
+
+import (
+	"sync"
+	"time"
+)
+
+type ExecuteChan struct {
+	c    chan bool
+	once sync.Once
+}
+
+// 返回函数f是否被运行
+func Run(limit time.Duration, f func()) bool {
+	ec := &ExecuteChan{c: make(chan bool), once: sync.Once{}}
+	whether := true
+
+	go func() {
+		f()
+		ec.once.Do(func() {
+			ec.c <- true
+			close(ec.c)
+		})
+	}()
+
+	select {
+	case <-time.After(limit):
+		whether = false
+	case <-ec.c:
+	}
+
+	return whether
+}

timeout/run_test.go

@@ -0,0 +1,70 @@
+package timeout_test
+
+import (
+	"sync"
+	"testing"
+	"time"
+
+	"git.links123.net/links123.com/pkg/timeout"
+)
+
+type ExcuteChan struct {
+	c    chan bool
+	once sync.Once
+}
+
+func Run(limit time.Duration, f func()) bool {
+	done := make(chan bool)
+
+	go func() {
+		f()
+		done <- true
+		close(done)
+	}()
+
+	select {
+	case <-time.After(limit):
+		return false
+	case <-done:
+		return true
+	}
+}
+
+func Run1(limit time.Duration, f func()) bool {
+	ec := &ExcuteChan{c: make(chan bool), once: sync.Once{}}
+
+	go func() {
+		f()
+		ec.once.Do(func() {
+			ec.c <- true
+			close(ec.c)
+		})
+	}()
+
+	select {
+	case <-time.After(limit):
+		return false
+	case <-ec.c:
+		return true
+	}
+}
+
+func TestRun(t *testing.T) {
+	// timeout: OK
+	if !timeout.Run(1*time.Second, func() {
+		time.Sleep(2 * time.Second)
+	}) {
+		t.Log("time out: OK")
+	} else {
+		t.Fatal("time out: FAILED")
+	}
+
+	// timeout: FAILED
+	if !timeout.Run(3*time.Second, func() {
+		time.Sleep(2 * time.Second)
+	}) {
+		t.Log("time out: OK")
+	} else {
+		t.Fatal("time out: FAILED")
+	}
+}