diff -r 540607cf3447 -r 9ba15af20acc Entity/DocumentRepository.php --- 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;