瀏覽代碼

[mapping] applied ornicar fix for file driver

gediminasm 14 年之前
父節點
當前提交
8c00fc5378
共有 1 個文件被更改,包括 4 次插入3 次删除
  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);
             }
         }