tidy

lesismal · Feb 13, 2024 · 50b8c1f · 50b8c1f
1 parent 5e99e20
commit 50b8c1f
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/conn.go b/conn.go
@@ -55,7 +55,12 @@ func (c *Conn) IsUnix() bool {
 
 // OnData registers callback for data.
 func (c *Conn) OnData(h func(conn *Conn, data []byte)) {
-	c.DataHandler = h
+	c.dataHandler = h
+}
+
+// DataHandler returns data handler.
+func (c *Conn) DataHandler() func(conn *Conn, data []byte) {
+	return c.dataHandler
 }
 
 // Dial wraps net.Dial.

diff --git a/conn_std.go b/conn_std.go
@@ -44,7 +44,7 @@ type Conn struct {
 
 	cache *bytes.Buffer
 
-	DataHandler func(c *Conn, data []byte)
+	dataHandler func(c *Conn, data []byte)
 }
 
 // Hash returns a hashcode.

diff --git a/conn_unix.go b/conn_unix.go
@@ -96,7 +96,7 @@ type Conn struct {
 
 	readEvents int32
 
-	DataHandler func(c *Conn, data []byte)
+	dataHandler func(c *Conn, data []byte)
 }
 
 // Hash returns a hash code.

diff --git a/nbhttp/engine.go b/nbhttp/engine.go
@@ -1037,7 +1037,7 @@ func NewEngine(conf Config) *Engine {
 	})
 
 	g.OnData(func(c *nbio.Conn, data []byte) {
-		c.DataHandler(c, data)
+		c.DataHandler()(c, data)
 	})
 
 	// engine.isOneshot = (conf.EpollMod == nbio.EPOLLET && conf.EPOLLONESHOT == nbio.EPOLLONESHOT && runtime.GOOS == "linux")