From 548df9a5f9454d2c8e647d58fd453fe19156b191 Mon Sep 17 00:00:00 2001
From: daheige <zhuwei313@hotmail.com>
Date: Sun, 7 Mar 2021 00:29:28 +0800
Subject: [PATCH] invalid db error for DB method

---
 errors.go | 2 ++
 gorm.go   | 3 +--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/errors.go b/errors.go
index 08755083..a559aaa6 100644
--- a/errors.go
+++ b/errors.go
@@ -31,4 +31,6 @@ var (
 	ErrEmptySlice = errors.New("empty slice found")
 	// ErrDryRunModeUnsupported dry run mode unsupported
 	ErrDryRunModeUnsupported = errors.New("dry run mode unsupported")
+	// ErrInvaildDB invalid db
+	ErrInvaildDB = errors.New("invalid db")
 )
diff --git a/gorm.go b/gorm.go
index f11fb9e1..0eb377e9 100644
--- a/gorm.go
+++ b/gorm.go
@@ -3,7 +3,6 @@ package gorm
 import (
 	"context"
 	"database/sql"
-	"errors"
 	"fmt"
 	"sync"
 	"time"
@@ -331,7 +330,7 @@ func (db *DB) DB() (*sql.DB, error) {
 		return sqldb, nil
 	}
 
-	return nil, errors.New("invalid db")
+	return nil, ErrInvaildDB
 }
 
 func (db *DB) getInstance() *DB {