diff options
| -rw-r--r-- | lib/arel/engines/sql/compilers/ibm_db_compiler.rb | 62 | ||||
| -rw-r--r-- | lib/arel/engines/sql/relations/compiler.rb | 7 | ||||
| -rw-r--r-- | lib/arel/engines/sql/relations/writes.rb | 6 |
3 files changed, 70 insertions, 5 deletions
diff --git a/lib/arel/engines/sql/compilers/ibm_db_compiler.rb b/lib/arel/engines/sql/compilers/ibm_db_compiler.rb new file mode 100644 index 0000000000..e6be73d6a1 --- /dev/null +++ b/lib/arel/engines/sql/compilers/ibm_db_compiler.rb @@ -0,0 +1,62 @@ +# +-----------------------------------------------------------------------+ +# | | +# | Copyright (c) 2010 IBM Corporation | +# | | +# | Permission is hereby granted, free of charge, to any person obtaining | +# | a copy of this software and associated documentation files (the | +# | "Software"), to deal in the Software without restriction, including | +# | without limitation the rights to use, copy, modify, merge, publish, | +# | distribute, sublicense, and/or sell copies of the Software, and to | +# | permit persons to whom the Software is furnished to do so, subject to | +# | the following conditions: | +# | | +# | The above copyright notice and this permission notice shall be | +# | included in all copies or substantial portions of the Software. | +# | | +# | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | +# | EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | +# | MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.| +# | IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR | +# | ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION | +# | OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION | +# | WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | +# | | +# +-----------------------------------------------------------------------+ + +# +# Author: Praveen Devarao <praveendrl@in.ibm.com> +# + +module Arel + module SqlCompiler + class IBM_DBCompiler < GenericCompiler + + def select_sql + query = build_query \ + "SELECT #{select_clauses.join(', ')}", + "FROM #{from_clauses}", + (joins(self) unless joins(self).blank? ), + ("WHERE #{where_clauses.join(" AND ")}" unless wheres.blank? ), + ("GROUP BY #{group_clauses.join(', ')}" unless groupings.blank? ), + ("HAVING #{having_clauses.join(', ')}" unless havings.blank? ), + ("ORDER BY #{order_clauses.join(', ')}" unless orders.blank? ) + engine.add_limit_offset!(query,{:limit=>taken,:offset=>skipped}) unless taken.blank? + query << "#{locked}" unless locked.blank? + query + end + + def limited_update_conditions(conditions, taken) + quoted_primary_key = engine.quote_table_name(primary_key) + update_conditions = "WHERE #{quoted_primary_key} IN (SELECT #{quoted_primary_key} FROM #{engine.connection.quote_table_name table.name} #{conditions} " #Note: - ')' not added, limit segment is to be appended + engine.add_limit_offset!(update_conditions,{:limit=>taken,:offset=>nil}) + update_conditions << ")" # Close the sql segment + update_conditions + end + + def add_limit_on_delete(taken) + "" # Limiting the number of rows to be deleted is not supported by IBM_DB + end + + end + end +end diff --git a/lib/arel/engines/sql/relations/compiler.rb b/lib/arel/engines/sql/relations/compiler.rb index c974e43b88..597ed88683 100644 --- a/lib/arel/engines/sql/relations/compiler.rb +++ b/lib/arel/engines/sql/relations/compiler.rb @@ -21,11 +21,16 @@ module Arel ("#{locked}" unless locked.blank?) end - def limited_update_conditions(conditions) + def limited_update_conditions(conditions, taken) + conditions << " LIMIT #{taken}" quoted_primary_key = engine.quote_table_name(primary_key) "WHERE #{quoted_primary_key} IN (SELECT #{quoted_primary_key} FROM #{engine.connection.quote_table_name table.name} #{conditions})" end + def add_limit_on_delete(taken) + "LIMIT #{taken}" + end + def supports_insert_with_returning? false end diff --git a/lib/arel/engines/sql/relations/writes.rb b/lib/arel/engines/sql/relations/writes.rb index 0a6250a3b9..54eaee9ead 100644 --- a/lib/arel/engines/sql/relations/writes.rb +++ b/lib/arel/engines/sql/relations/writes.rb @@ -5,7 +5,7 @@ module Arel "DELETE", "FROM #{table_sql}", ("WHERE #{wheres.collect(&:to_sql).join(' AND ')}" unless wheres.blank? ), - ("LIMIT #{taken}" unless taken.blank? ) + (compiler.add_limit_on_delete(taken) unless taken.blank? ) end end @@ -68,9 +68,7 @@ module Arel conditions << " ORDER BY #{order_clauses.join(', ')}" unless orders.blank? unless taken.blank? - conditions << " LIMIT #{taken}" - - conditions = compiler.limited_update_conditions(conditions) + conditions = compiler.limited_update_conditions(conditions,taken) end conditions |