improve checking

go-gorm · Aug 28, 2023 · e7b465e · e7b465e
1 parent c793c4c
commit e7b465e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sharding.go b/sharding.go
@@ -270,10 +270,8 @@ func (s *Sharding) switchConn(db *gorm.DB) {
 	if _, ok := db.Get(ShardingIgnoreStoreKey); !ok {
 		s.mutex.Lock()
 		if db.Statement.ConnPool != nil {
-			if _, ok = db.Statement.ConnPool.(ConnPool); !ok {
-				s.ConnPool = &ConnPool{ConnPool: db.Statement.ConnPool, sharding: s}
-				db.Statement.ConnPool = s.ConnPool
-			}
+			s.ConnPool = &ConnPool{ConnPool: db.Statement.ConnPool, sharding: s}
+			db.Statement.ConnPool = s.ConnPool
 		}
 		s.mutex.Unlock()
 	}