Use existing interface

conn/export.go

@@ -14,3 +14,5 @@ var (
 	// is not thread safe and should only be modified during init.
 	ControlFns = &controlFns
 )
+
+type BatchReader = batchReader

conn/receiver_creator.go

@@ -3,15 +3,8 @@ package conn
 import (
 	"net"
 	"sync"
-
-	"golang.org/x/net/ipv6"
 )
 
-// PacketReader abstracts ipv4.PacketConn and ipv6.PacketConn which both implement ReadBatch
-type PacketReader interface {
-	ReadBatch([]ipv6.Message, int) (int, error)
-}
-
 type ReceiverCreator interface {
-	CreateReceiverFn(pc PacketReader, conn *net.UDPConn, rxOffload bool, msgPool *sync.Pool) ReceiveFunc
+	CreateReceiverFn(pc BatchReader, conn *net.UDPConn, rxOffload bool, msgPool *sync.Pool) ReceiveFunc
 }