Adding access to underlying gorm DB

interface.go

@@ -17,6 +17,8 @@ type Interface interface {
 	Value() interface{}
 	Error() error
 	RowsAffected() int64
+	// Allow users access to the underlying gorm DB
+	GetGormDB() *gorm.DB
 	// Wrap the functions
 	New() Interface
 	Close() error

jorm.go

@@ -371,7 +371,7 @@ func (db *DB) HasTable(value interface{}) bool {
 
 // AutoMigrate run auto migration for given models, will only add missing fields, won't delete/change current data
 func (db *DB) AutoMigrate(values ...interface{}) Interface {
-	return &DB{db: db.db.AutoMigrate(values)}
+	return &DB{db: db.db.AutoMigrate(values...)}
 }
 
 // ModifyColumn modify column to type
@@ -451,3 +451,8 @@ func (db *DB) AddError(err error) error {
 func (db *DB) GetErrors() []error {
 	return db.db.GetErrors()
 }
+
+// GetGormDB returns the underlying gorm DB
+func (db *DB) GetGormDB() *gorm.DB {
+	return db.db
+}