refactor

cheng-zhongliang · Oct 11, 2023 · b8e0606 · b8e0606
1 parent 16dc9d2
commit b8e0606
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/epoll.go b/epoll.go
@@ -8,6 +8,7 @@
 package event
 
 import (
+	"sync"
 	"syscall"
 	"time"
 	"unsafe"
@@ -19,6 +20,12 @@ const (
 	maxUint32     = 0xFFFFFFFF
 )
 
+var fdEvPool = sync.Pool{
+	New: func() any {
+		return new(fdEvent)
+	},
+}
+
 type fdEvent struct {
 	r   *Event
 	w   *Event
@@ -51,7 +58,7 @@ func (ep *poller) add(ev *Event) error {
 	if ok {
 		op = syscall.EPOLL_CTL_MOD
 	} else {
-		es = new(fdEvent)
+		es = fdEvPool.Get().(*fdEvent)
 		ep.fdEvents[ev.fd] = es
 	}
 
@@ -95,6 +102,7 @@ func (ep *poller) del(ev *Event) error {
 	op := syscall.EPOLL_CTL_DEL
 	if es.evs&(syscall.EPOLLIN|syscall.EPOLLOUT) == 0 {
 		delete(ep.fdEvents, ev.fd)
+		fdEvPool.Put(es)
 	} else {
 		op = syscall.EPOLL_CTL_MOD
 	}