initial commit

1 files changed, 34 insertions, 0 deletions

diff --git a/timer.go b/timer.go
new file mode 100644
index 0000000..8227c7d
--- /dev/null
+++ b/timer.go
@@ -0,0 +1,34 @@
+package fasthttp
+
+import (
+	"sync"
+	"time"
+)
+
+var timerPool sync.Pool
+
+func acquireTimer(timeout time.Duration) *time.Timer {
+	tv := timerPool.Get()
+	if tv == nil {
+		return time.NewTimer(timeout)
+	}
+
+	t := tv.(*time.Timer)
+	if t.Reset(timeout) {
+		panic("BUG: Active timer trapped into AcquireTimer()")
+	}
+	return t
+}
+
+func releaseTimer(t *time.Timer) {
+	if !t.Stop() {
+		// Collect possibly added time from the channel
+		// if timer has been stopped and nobody collected its' value.
+		select {
+		case <-t.C:
+		default:
+		}
+	}
+
+	timerPool.Put(t)
+}