From daa718588373352d5a56f682e5d6f1130667e225 Mon Sep 17 00:00:00 2001
From: Siyuan Miao <i@xswan.net>
Date: Tue, 15 Apr 2025 16:47:23 +0200
Subject: [PATCH] refactor(nbd): move platform-specific code to different files

---
 block_device.go          | 28 ----------------------------
 block_device_linux.go    | 34 ++++++++++++++++++++++++++++++++++
 block_device_notlinux.go | 17 +++++++++++++++++
 3 files changed, 51 insertions(+), 28 deletions(-)
 create mode 100644 block_device_linux.go
 create mode 100644 block_device_notlinux.go

diff --git a/block_device.go b/block_device.go
index d93f474..aa1fce8 100644
--- a/block_device.go
+++ b/block_device.go
@@ -7,7 +7,6 @@ import (
 	"os"
 	"time"
 
-	"github.com/pojntfx/go-nbd/pkg/client"
 	"github.com/pojntfx/go-nbd/pkg/server"
 	"github.com/rs/zerolog"
 )
@@ -148,30 +147,3 @@ func (d *NBDDevice) runServerConn() {
 
 	d.l.Info().Err(err).Msg("nbd server exited")
 }
-
-func (d *NBDDevice) runClientConn() {
-	err := client.Connect(d.clientConn, d.dev, &client.Options{
-		ExportName: "jetkvm",
-		BlockSize:  uint32(4 * 1024),
-	})
-	d.l.Info().Err(err).Msg("nbd client exited")
-}
-
-func (d *NBDDevice) Close() {
-	if d.dev != nil {
-		err := client.Disconnect(d.dev)
-		if err != nil {
-			d.l.Warn().Err(err).Msg("error disconnecting nbd client")
-		}
-		_ = d.dev.Close()
-	}
-	if d.listener != nil {
-		_ = d.listener.Close()
-	}
-	if d.clientConn != nil {
-		_ = d.clientConn.Close()
-	}
-	if d.serverConn != nil {
-		_ = d.serverConn.Close()
-	}
-}
diff --git a/block_device_linux.go b/block_device_linux.go
new file mode 100644
index 0000000..8ca9372
--- /dev/null
+++ b/block_device_linux.go
@@ -0,0 +1,34 @@
+//go:build linux
+
+package kvm
+
+import (
+	"github.com/pojntfx/go-nbd/pkg/client"
+)
+
+func (d *NBDDevice) runClientConn() {
+	err := client.Connect(d.clientConn, d.dev, &client.Options{
+		ExportName: "jetkvm",
+		BlockSize:  uint32(4 * 1024),
+	})
+	d.l.Info().Err(err).Msg("nbd client exited")
+}
+
+func (d *NBDDevice) Close() {
+	if d.dev != nil {
+		err := client.Disconnect(d.dev)
+		if err != nil {
+			d.l.Warn().Err(err).Msg("error disconnecting nbd client")
+		}
+		_ = d.dev.Close()
+	}
+	if d.listener != nil {
+		_ = d.listener.Close()
+	}
+	if d.clientConn != nil {
+		_ = d.clientConn.Close()
+	}
+	if d.serverConn != nil {
+		_ = d.serverConn.Close()
+	}
+}
diff --git a/block_device_notlinux.go b/block_device_notlinux.go
new file mode 100644
index 0000000..b6a9aba
--- /dev/null
+++ b/block_device_notlinux.go
@@ -0,0 +1,17 @@
+//go:build !linux
+
+package kvm
+
+import (
+	"os"
+)
+
+func (d *NBDDevice) runClientConn() {
+	d.l.Error().Msg("platform not supported")
+	os.Exit(1)
+}
+
+func (d *NBDDevice) Close() {
+	d.l.Error().Msg("platform not supported")
+	os.Exit(1)
+}