wrap

datafile/entry.go

@@ -3,7 +3,6 @@ package datafile
 import (
 	"hash/crc32"
 	"io"
-	goruntime "runtime"
 	"time"
 
 	"github.com/octu0/bitcaskdb/runtime"
@@ -34,7 +33,7 @@ type Entry struct {
 }
 
 func (e *Entry) setFinalizer() {
-	goruntime.SetFinalizer(e, finalizeEntry)
+	runtime.SetFinalizer(e, finalizeEntry)
 }
 
 func (e *Entry) Read(p []byte) (int, error) {
@@ -44,7 +43,7 @@ func (e *Entry) Read(p []byte) (int, error) {
 func (e *Entry) Close() error {
 	if e.closed != true {
 		e.closed = true
-		goruntime.SetFinalizer(e, nil) // clear finalizer
+		runtime.SetFinalizer(e, nil) // clear finalizer
 		if e.release != nil {
 			e.release()
 		}
@@ -76,6 +75,6 @@ func (e *Entry) Validate(ctx runtime.Context) error {
 }
 
 func finalizeEntry(e *Entry) {
-	goruntime.SetFinalizer(e, nil) // clear finalizer
+	runtime.SetFinalizer(e, nil) // clear finalizer
 	e.Close()
 }

merge.go

@@ -7,7 +7,7 @@ import (
 	"math"
 	"os"
 	"path/filepath"
-	goruntime "runtime"
+	"runtime"
 	"sort"
 	"sync"
 	"time"
@@ -402,7 +402,7 @@ func (t *mergeTempDB) Destroy(lim *priorate.Limiter) {
 		return
 	}
 
-	goruntime.SetFinalizer(t, nil) // clear
+	runtime.SetFinalizer(t, nil) // clear
 	t.SyncAndClose()
 
 	files, err := filepath.Glob(filepath.Join(t.tempDir, "*"))
@@ -436,7 +436,7 @@ func openMergeTempDB(basedir string, opt *option) (*mergeTempDB, error) {
 		closed:    false,
 		destroyed: false,
 	}
-	goruntime.SetFinalizer(st, finalizeMergeTempDB)
+	runtime.SetFinalizer(st, finalizeMergeTempDB)
 	return st, nil
 }
 
@@ -491,7 +491,7 @@ func (st *snapshotTrie) Destroy(lim *priorate.Limiter) {
 		return
 	}
 
-	goruntime.SetFinalizer(st, nil)
+	runtime.SetFinalizer(st, nil)
 	st.Close()
 	removeFileSlowly([]string{st.file.Name()}, lim)
 	st.destroyed = true
@@ -512,7 +512,7 @@ func openSnapshotTrie(tempDir string) (*snapshotTrie, error) {
 		closed:    false,
 		destroyed: false,
 	}
-	goruntime.SetFinalizer(st, finalizeSnapshotTrie)
+	runtime.SetFinalizer(st, finalizeSnapshotTrie)
 	return st, nil
 }
 

runtime/runtime.go

@@ -0,0 +1,9 @@
+package runtime
+
+import (
+	goruntime "runtime"
+)
+
+func SetFinalizer(obj interface{}, finalizer interface{}) {
+	goruntime.SetFinalizer(obj, finalizer)
+}