Merge pull request #1668 from ydb-platform/xruntime

Added package internal/xruntime with helper function AddCleanup

Author: asmyasnikov

internal/xruntime/cleanup.go

@@ -0,0 +1,11 @@
+//go:build !go1.24
+
+package xruntime
+
+import "runtime"
+
+func AddCleanup[T, S any](ptr *T, cleanup func(S), arg S) {
+	runtime.SetFinalizer(ptr, func(ptr *T) {
+		cleanup(arg)
+	})
+}

internal/xruntime/cleanup_go1.24.go

@@ -0,0 +1,9 @@
+//go:build go1.24
+
+package xruntime
+
+import "runtime"
+
+func AddCleanup[T, S any](ptr *T, cleanup func(S), arg S) {
+	runtime.AddCleanup(ptr, cleanup, arg)
+}

internal/xruntime/cleanup_test.go

@@ -0,0 +1,27 @@
+package xruntime
+
+import (
+	"runtime"
+	"testing"
+	"time"
+)
+
+func TestAddCleanup(t *testing.T) {
+	cleanupCalled := make(chan struct{})
+	func() {
+		v := &struct {
+			a int64
+			b string
+			c bool
+		}{}
+		AddCleanup(v, func(cleanupCalled chan struct{}) {
+			close(cleanupCalled)
+		}, cleanupCalled)
+	}()
+	runtime.GC()
+	select {
+	case <-cleanupCalled:
+	case <-time.After(time.Second):
+		t.Fail()
+	}
+}