--- a/Entity/DocumentRepository.php	Thu Nov 17 12:07:53 2011 +0100
+++ b/Entity/DocumentRepository.php	Thu Nov 17 16:48:19 2011 +0100
@@ -7,6 +7,7 @@
 use Doctrine\ORM\Query\ResultSetMapping;
 use \ReflectionClass;
 use Doctrine\ORM\AbstractQuery;
+use Doctrine\ORM\Mapping\ClassMetadataInfo;
 
 /**
  * DocumentRepository
@@ -135,7 +136,32 @@
         throw new \Exception("Unknown accessor $accessor");
     }
     
-    function writeDocument($document,  $document_id_column, $fields)
+    
+    private function getColumnName($field_name)
+    {
+        if(isset($this->getClassMetadata()->columnNames[$field_name]))
+        {
+            return $this->getClassMetadata()->columnNames[$field_name];
+        }
+        
+        $res = $field_name;
+        if(isset($this->getClassMetadata()->associationMappings[$field_name]))
+        {
+            $association_mapping = $this->getClassMetadata()->associationMappings[$field_name];
+            if(
+                isset($association_mapping['type'])
+                && ( $association_mapping['type'] === ClassMetadataInfo::ONE_TO_ONE || $association_mapping['type'] === ClassMetadataInfo::MANY_TO_ONE)
+                && count($association_mapping['joinColumns']) > 0
+            )
+            {
+                $res = $association_mapping['joinColumns'][0]['name'];
+            }
+        }
+        
+        return $res;
+    }
+    
+    public function writeDocument($document,  $document_id_column, $fields)
     {
         // get document from id
          
@@ -231,7 +257,14 @@
         $i = 0;
         foreach ($values as $fielddef) {
             $i++;
-            $columns = $fielddef["columns"];
+            $field_list = explode(",", $fielddef["columns"]);
+            $column_list = array();
+            foreach($field_list as $field_name)
+            {
+                $column_list[] = $this->getColumnName(trim($field_name));
+            }
+            $columns = join(",", $column_list);
+            
             $value = $fielddef["value"];
             $weight = isset($fielddef["weight"])?$fielddef["weight"]:1.0;
             
@@ -251,6 +284,7 @@
             foreach ($conditions as $field => $conddef)
             {
                 $i++;
+                $col = $this->getColumnName($field);
                 if(is_array($conddef) && isset($conddef['operator']))
                 {
                     $operator = $conddef["operator"];
@@ -274,11 +308,11 @@
                         $parameters["cond_val_$i_$j"] = $values[$j];
                         $in_parameters[] = ":cond_val_$i_$j";
                     }
-                    $cond = "($field IN (".implode(",",$in_parameters)."))";
+                    $cond = "($col IN (".implode(",",$in_parameters)."))";
                 }
                 else
                 {
-                    $cond = "($field $operator :cond_val_$i)";
+                    $cond = "($col $operator :cond_val_$i)";
                     $parameters["cond_val_$i"] = $values;
                 }
                 $conditions_array[] = $cond;