Explorar o código

[mapping] applied ornicar fix for file driver

gediminasm %!s(int64=14) %!d(string=hai) anos
pai
achega
8c00fc5378
Modificáronse 1 ficheiros con 4 adicións e 3 borrados
  1. 4 3
      lib/Gedmo/Mapping/Driver/File.php

+ 4 - 3
lib/Gedmo/Mapping/Driver/File.php

@@ -2,8 +2,9 @@
 
 namespace Gedmo\Mapping\Driver;
 
-use Gedmo\Mapping\Driver,
-    Doctrine\ORM\Mapping\Driver\AbstractFileDriver;
+use Gedmo\Mapping\Driver;
+use Doctrine\ORM\Mapping\Driver\AbstractFileDriver as ORMAbstractFileDriver;
+use Doctrine\ODM\MongoDB\Mapping\Driver\AbstractFileDriver as MongoDBAbstractFileDriver;
 
 /**
  * The mapping FileDriver abstract class, defines the
@@ -99,7 +100,7 @@ abstract class File implements Driver
         //try loading mapping from original driver first
         $mapping = null;
         if (!is_null($this->_originalDriver)) {
-            if ($this->_originalDriver instanceof AbstractFileDriver) {
+            if ($this->_originalDriver instanceof ORMAbstractFileDriver || $this->_originalDriver instanceof MongoDBAbstractFileDriver) {
                 $mapping = $this->_originalDriver->getElement($className);
             }
         }