set finalizer for adapter (#5)

JunNishimura · Mar 31, 2024 · c1a9355 · c1a9355
1 parent 4a3edd8
commit c1a9355
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/adapter.go b/adapter.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"runtime"
 
 	"github.com/casbin/casbin/v2/model"
 	"github.com/casbin/casbin/v2/persist"
@@ -59,6 +60,12 @@ func NewAdapter(driverName, dataSourceName string, opts ...adapterOption) (*bunA
 		return nil, err
 	}
 
+	runtime.SetFinalizer(b, func(a *bunAdapter) {
+		if err := a.db.Close(); err != nil {
+			panic(err)
+		}
+	})
+
 	return b, nil
 }