added a method to close the database connection

assetdb.go

@@ -23,6 +23,11 @@ func New(dbType repository.DBType, dsn string) *AssetDB {
 	}
 }
 
+// Close will close the assetdb and return any errors.
+func (as *AssetDB) Close() error {
+	return as.repository.Close()
+}
+
 // GetDBType returns the type of the underlying database.
 func (as *AssetDB) GetDBType() string {
 	return as.repository.GetDBType()

assetdb_test.go

@@ -634,6 +634,10 @@ type mockAssetDB struct {
 	mock.Mock
 }
 
+func (m *mockAssetDB) Close() error {
+	return nil
+}
+
 func (m *mockAssetDB) GetDBType() string {
 	args := m.Called()
 	return args.String(0)

repository/repository.go

@@ -26,4 +26,5 @@ type Repository interface {
 	RawQuery(sqlstr string, results interface{}) error
 	AssetQuery(constraints string) ([]*types.Asset, error)
 	RelationQuery(constraints string) ([]*types.Relation, error)
+	Close() error
 }

repository/sql.go

@@ -70,6 +70,14 @@ func sqliteDatabase(dsn string) (*gorm.DB, error) {
 	return gorm.Open(sqlite.Open(dsn), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
 }
 
+// Close implements the Repository interface.
+func (sql *sqlRepository) Close() error {
+	if db, err := sql.db.DB(); err == nil {
+		return db.Close()
+	}
+	return errors.New("failed to obtain access to the database handle")
+}
+
 // GetDBType returns the type of the database.
 func (sql *sqlRepository) GetDBType() string {
 	return string(sql.dbType)