Fix for file descriptor type difference on Windows (#56)

Author: blinktag

pkg/stream/client.go

@@ -9,10 +9,8 @@ import (
 	"math/rand"
 	"net"
 	"net/url"
-	"runtime"
 	"sync"
 	"sync/atomic"
-	"syscall"
 	"time"
 )
 
@@ -106,24 +104,7 @@ func (c *Client) connect() error {
 		c.tuneState.requestedMaxFrameSize = 1048576
 
 		var dialer = &net.Dialer{
-			Control: func(network, address string, c syscall.RawConn) error {
-				return c.Control(func(fd uintptr) {
-					err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_RCVBUF, defaultSocketBuffer)
-					runtime.KeepAlive(fd)
-					if err != nil {
-						logs.LogError("Set socket option error: %s", err)
-						return
-					}
-
-					err = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_SNDBUF, defaultSocketBuffer)
-					runtime.KeepAlive(fd)
-					if err != nil {
-						logs.LogError("Set socket option error: %s", err)
-						return
-					}
-
-				})
-			},
+			Control: controlFunc,
 		}
 
 		var connection net.Conn

pkg/stream/dialer_posix.go

@@ -0,0 +1,33 @@
+// +build !windows
+
+package stream
+
+import (
+	"runtime"
+	"syscall"
+
+	"github.com/rabbitmq/rabbitmq-stream-go-client/pkg/logs"
+)
+
+var controlFunc func(network, address string, c syscall.RawConn) error
+
+func init() {
+	controlFunc = func(network, address string, c syscall.RawConn) error {
+		return c.Control(func(fd uintptr) {
+			err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_RCVBUF, defaultSocketBuffer)
+			runtime.KeepAlive(fd)
+			if err != nil {
+				logs.LogError("Set socket option error: %s", err)
+				return
+			}
+
+			err = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_SNDBUF, defaultSocketBuffer)
+			runtime.KeepAlive(fd)
+			if err != nil {
+				logs.LogError("Set socket option error: %s", err)
+				return
+			}
+
+		})
+	}
+}

pkg/stream/dialer_windows.go

@@ -0,0 +1,33 @@
+// +build windows
+
+package stream
+
+import (
+	"runtime"
+	"syscall"
+
+	"github.com/rabbitmq/rabbitmq-stream-go-client/pkg/logs"
+)
+
+var controlFunc func(network, address string, c syscall.RawConn) error
+
+func init() {
+	controlFunc = func(network, address string, c syscall.RawConn) error {
+		return c.Control(func(fd uintptr) {
+			err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_RCVBUF, defaultSocketBuffer)
+			runtime.KeepAlive(fd)
+			if err != nil {
+				logs.LogError("Set socket option error: %s", err)
+				return
+			}
+
+			err = syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_SNDBUF, defaultSocketBuffer)
+			runtime.KeepAlive(fd)
+			if err != nil {
+				logs.LogError("Set socket option error: %s", err)
+				return
+			}
+
+		})
+	}
+}