From f521b180f5b0e144e731c4a08b876ce2d47c9f30 Mon Sep 17 00:00:00 2001
From: Shelikhoo <xiaokangwang@outlook.com>
Date: Tue, 14 May 2024 11:13:24 +0100
Subject: [PATCH] add connwrapper

---
 client/lib/connwrapper.go | 45 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 client/lib/connwrapper.go

diff --git a/client/lib/connwrapper.go b/client/lib/connwrapper.go
new file mode 100644
index 0000000..ca96a24
--- /dev/null
+++ b/client/lib/connwrapper.go
@@ -0,0 +1,45 @@
+package snowflake_client
+
+import (
+	"io"
+	"net"
+	"time"
+)
+
+type packetConnWrapper struct {
+	io.ReadWriter
+	remoteAddr net.Addr
+	localAddr  net.Addr
+}
+
+func (pcw *packetConnWrapper) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
+	n, err = pcw.Read(p)
+	if err != nil {
+		return 0, nil, err
+	}
+	return n, pcw.remoteAddr, nil
+}
+
+func (pcw *packetConnWrapper) WriteTo(p []byte, addr net.Addr) (n int, err error) {
+	return pcw.Write(p)
+}
+
+func (pcw *packetConnWrapper) Close() error {
+	return nil
+}
+
+func (pcw *packetConnWrapper) LocalAddr() net.Addr {
+	return pcw.localAddr
+}
+
+func (pcw *packetConnWrapper) SetDeadline(t time.Time) error {
+	return nil
+}
+
+func (pcw *packetConnWrapper) SetReadDeadline(t time.Time) error {
+	return nil
+}
+
+func (pcw *packetConnWrapper) SetWriteDeadline(t time.Time) error {
+	return nil
+}