class Hayabusa::Objects
attr_reader :args, :events, :data, :ids_cache, :ids_cache_should
def initialize(args)
require "monitor"
@callbacks = {}
@args = args
@args[:col_id] = :id if !@args[:col_id]
@args[:class_pre] = "class_" if !@args[:class_pre]
@args[:module] = Kernel if !@args[:module]
@args[:cache] = :weak if !@args.key?(:cache)
@objects = {}
@locks = {}
@data = {}
@lock_require = Monitor.new
Knj.gem_require(:Wref, "wref") if @args[:cache] == :weak and !Kernel.const_defined?(:Wref)
require "#{@args[:array_enumerator_path]}array_enumerator" if @args[:array_enum] and !Kernel.const_defined?(:Array_enumerator)
#Set up various events.
@events = Knj::Event_handler.new
@events.add_event(:name => :no_html, :connections_max => 1)
@events.add_event(:name => :no_name, :connections_max => 1)
@events.add_event(:name => :no_date, :connections_max => 1)
@events.add_event(:name => :missing_class, :connections_max => 1)
@events.add_event(:name => :require_class, :connections_max => 1)
raise "No DB given." if !@args[:db] and !@args[:custom]
raise "No class path given." if !@args[:class_path] and (@args[:require] or !@args.key?(:require))
if args[:require_all]
Knj.gem_require(:Php4r, "php4r")
loads = []
Dir.foreach(@args[:class_path]) do |file|
next if file == "." or file == ".." or !file.match(/\.rb$/)
file_parsed = file
file_parsed.gsub!(@args[:class_pre], "") if @args.key?(:class_pre)
file_parsed.gsub!(/\.rb$/, "")
file_parsed = Php4r.ucwords(file_parsed)
loads << file_parsed
self.requireclass(file_parsed, {:load => false})
end
loads.each do |load_class|
self.load_class(load_class)
end
end
#Set up ID-caching.
@ids_cache_should = {}
if @args[:models]
@ids_cache = {}
@args[:models].each do |classname, classargs|
@ids_cache_should[classname] = true if classargs[:cache_ids]
self.cache_ids(classname)
end
end
end
#Caches all IDs for a specific classname.
def cache_ids(classname)
classname = classname.to_sym
return nil if !@ids_cache_should or !@ids_cache_should[classname]
newcache = {}
@args[:db].q("SELECT `#{@args[:col_id]}` FROM `#{classname}` ORDER BY `#{@args[:col_id]}`") do |data|
newcache[data[@args[:col_id]].to_i] = true
end
@ids_cache[classname] = newcache
end
def init_class(classname)
classname = classname.to_sym
return false if @objects.key?(classname)
if @args[:cache] == :weak
@objects[classname] = Wref::Map.new
else
@objects[classname] = {}
end
@locks[classname] = Monitor.new
end
def uninit_class(classname)
@objects.delete(classname)
@locks.delete(classname)
end
#Returns a cloned version of the @objects variable. Cloned because iteration on it may crash some of the other methods in Ruby 1.9+
def objects
objs_cloned = {}
@objects.keys.each do |key|
objs_cloned[key] = @objects[key].clone
end
return objs_cloned
end
#Returns the database-connection used by this instance of Objects.
def db
return @args[:db]
end
#Returns the total count of objects currently held by this instance.
def count_objects
count = 0
@objects.keys.each do |key|
count += @objects[key].length
end
return count
end
#This connects a block to an event. When the event is called the block will be executed.
def connect(args, &block)
raise "No object given." if !args["object"]
raise "No signals given." if !args.key?("signal") and !args.key?("signals")
args["block"] = block if block_given?
object = args["object"].to_sym
@callbacks[object] = {} if !@callbacks[object]
conn_id = @callbacks[object].length.to_s
@callbacks[object][conn_id] = args
return conn_id
end
#Returns true if the given signal is connected to the given object.
def connected?(args)
raise "No object given." if !args["object"]
raise "No signal given." if !args.key?("signal")
object = args["object"].to_sym
if @callbacks.key?(object)
@callbacks[object].clone.each do |ckey, callback|
return true if callback.key?("signal") and callback["signal"].to_s == args["signal"].to_s
return true if callback.key?("signals") and (callback["signals"].include?(args["signal"].to_s) or callback["signals"].include?(args["signal"].to_sym))
end
end
return false
end
#Unconnects a connect by 'object' and 'conn_id'.
def unconnect(args)
raise ArgumentError, "No object given." if !args["object"]
object = args["object"].to_sym
raise ArgumentError, "Object doesnt exist: '#{object}'." if !@callbacks.key?(object)
if args["conn_id"]
conn_ids = [args["conn_id"]]
elsif args["conn_ids"]
conn_ids = args["conn_ids"]
else
raise ArgumentError, "Could not figure out connection IDs."
end
conn_ids.each do |conn_id|
raise Errno::ENOENT, "Conn ID doest exist: '#{conn_id}' (#{args})." if !@callbacks[object].key?(conn_id)
@callbacks[object].delete(conn_id)
end
end
#This method is used to call the connected callbacks for an event.
def call(args, &block)
classstr = args["object"].class.classname.to_sym
if @callbacks.key?(classstr)
@callbacks[classstr].clone.each do |callback_key, callback|
docall = false
if callback.key?("signal") and args.key?("signal") and callback["signal"].to_s == args["signal"].to_s
docall = true
elsif callback["signals"] and args["signal"] and (callback["signals"].include?(args["signal"].to_s) or callback["signals"].include?(args["signal"].to_sym))
docall = true
end
next if !docall
if callback["block"]
callargs = []
arity = callback["block"].arity
if arity <= 0
#do nothing
elsif arity == 1
callargs << args["object"]
else
raise "Unknown number of arguments: #{arity}"
end
callback["block"].call(*callargs)
elsif callback["callback"]
require "php4r" if !Kernel.const_defined?(:Php4r)
Php4r.call_user_func(callback["callback"], args)
else
raise "No valid callback given."
end
end
end
end
def requireclass(classname, args = {})
classname = classname.to_sym
return false if @objects.key?(classname)
@lock_require.synchronize do
#Maybe the classname got required meanwhile the synchronized wait - check again.
return false if @objects.key?(classname)
if @events.connected?(:require_class)
@events.call(:require_class, {
:class => classname
})
else
doreq = false
if args[:require]
doreq = true
elsif args.key?(:require) and !args[:require]
doreq = false
elsif @args[:require] or !@args.key?(:require)
doreq = true
end
if doreq
filename = "#{@args[:class_path]}/#{@args[:class_pre]}#{classname.to_s.downcase}.rb"
filename_req = "#{@args[:class_path]}/#{@args[:class_pre]}#{classname.to_s.downcase}"
raise "Class file could not be found: #{filename}." if !File.exists?(filename)
require filename_req
end
end
if args[:class]
classob = args[:class]
else
begin
classob = @args[:module].const_get(classname)
rescue NameError => e
if @events.connected?(:missing_class)
@events.call(:missing_class, {
:class => classname
})
classob = @args[:module].const_get(classname)
else
raise e
end
end
end
if (classob.respond_to?(:load_columns) or classob.respond_to?(:datarow_init)) and (!args.key?(:load) or args[:load])
self.load_class(classname, args)
end
self.init_class(classname)
end
end
#Loads a Datarow-class by calling various static methods.
def load_class(classname, args = {})
if args[:class]
classob = args[:class]
else
classob = @args[:module].const_get(classname)
end
pass_arg = Knj::Hash_methods.new(:ob => self, :db => @args[:db])
classob.load_columns(pass_arg) if classob.respond_to?(:load_columns)
classob.datarow_init(pass_arg) if classob.respond_to?(:datarow_init)
end
#Returns the instance of classname, but only if it already exists.
def get_if_cached(classname, id)
classname = classname.to_sym
id = id.to_i
if @objects[classname] and obj = Wref::Map.get!(id)
return obj
end
return nil
end
#Returns true if a row of the given classname and the ID exists. Will use ID-cache if set in arguments and spawned otherwise it will do an actual lookup.
#===Examples
# print "User 5 exists." if ob.exists?(:User, 5)
def exists?(classname, id)
#Make sure the given data are in the correct types.
classname = classname.to_sym
id = id.to_i
#Check if ID-cache is enabled for that classname. Avoid SQL-lookup by using that.
if @ids_cache_should.key?(classname)
if @ids_cache[classname].key?(id)
return true
else
return false
end
end
#If the object currently exists in cache, we dont have to do a lookup either.
return true if @objects.key?(classname) and obj = @objects[classname].get!(id) and !obj.deleted?
#Okay - no other options than to actually do a real lookup.
begin
table = @args[:module].const_get(classname).table
row = @args[:db].single(table, {@args[:col_id] => id})
if row
return true
else
return false
end
rescue Errno::ENOENT
return false
end
end
#Gets an object from the ID or the full data-hash in the database.
#===Examples
# inst = ob.get(:User, 5)
def get(classname, data, args = nil)
classname = classname.to_sym
if data.is_a?(Integer) or data.is_a?(String) or data.is_a?(Fixnum)
id = data.to_i
elsif data.is_a?(Hash) and data.key?(@args[:col_id].to_sym)
id = data[@args[:col_id].to_sym].to_i
elsif data.is_a?(Hash) and data.key?(@args[:col_id].to_s)
id = data[@args[:col_id].to_s].to_i
elsif
raise ArgumentError, "Unknown data for class '#{classname}': '#{data.class.to_s}' (#{data})."
end
if @objects.key?(classname)
case @args[:cache]
when :weak
if obj = @objects[classname].get(id) and obj.id.to_i == id
return obj
end
else
return @objects[classname][id] if @objects[classname].key?(id)
end
end
self.requireclass(classname) if !@objects.key?(classname)
@locks[classname].synchronize do
#Maybe the object got spawned while we waited for the lock? If so we shouldnt spawn another instance.
if @args[:cache] == :weak and obj = @objects[classname].get(id) and obj.id.to_i == id
return obj
end
#Spawn object.
if @args[:datarow] or @args[:custom]
obj = @args[:module].const_get(classname).new(data, args)
else
pass_args = [data]
pass_args = pass_args | @args[:extra_args] if @args[:extra_args]
obj = @args[:module].const_get(classname).new(*pass_args)
end
#Save object in cache.
case @args[:cache]
when :none
return obj
else
@objects[classname][id] = obj
return obj
end
end
raise "Unexpected run?"
end
#Same as normal get but returns false if not found instead of raising error.
def get!(*args, &block)
begin
return self.get(*args, &block)
rescue Errno::ENOENT
return false
end
end
def object_finalizer(id)
classname = @objects_idclass[id]
if classname
@objects[classname].delete(id)
@objects_idclass.delete(id)
end
end
#Returns the first object found from the given arguments. Also automatically limits the results to 1.
def get_by(classname, args = {})
classname = classname.to_sym
self.requireclass(classname)
classob = @args[:module].const_get(classname)
raise "list-function has not been implemented for '#{classname}'." if !classob.respond_to?(:list)
args["limit"] = 1
self.list(classname, args) do |obj|
return obj
end
return false
end
def get_try(obj, col_name, obj_name = nil)
if !obj_name
if match = col_name.to_s.match(/^(.+)_id$/)
obj_name = Php4r.ucwords(match[1]).to_sym
else
raise "Could not figure out objectname for: #{col_name}."
end
end
id_data = obj[col_name].to_i
return false if id_data.to_i <= 0
begin
return self.get(obj_name, id_data)
rescue Errno::ENOENT
return false
end
end
#Returns an array-list of objects. If given a block the block will be called for each element and memory will be spared if running weak-link-mode.
#===Examples
# ob.list(:User) do |user|
# print "Username: #{user.name}\n"
# end
def list(classname, args = {}, &block)
args = {} if args == nil
classname = classname.to_sym
self.requireclass(classname)
classob = @args[:module].const_get(classname)
raise "list-function has not been implemented for '#{classname}'." if !classob.respond_to?("list")
if @args[:datarow] or @args[:custom]
ret = classob.list(Knj::Hash_methods.new(:args => args, :ob => self, :db => @args[:db]), &block)
else
realargs = [args]
realargs = realargs | @args[:extra_args] if @args[:extra_args]
ret = classob.list(*realargs, &block)
end
#If 'ret' is an array and a block is given then the list-method didnt return blocks. We emulate it instead with the following code.
if block and ret.is_a?(Array)
ret.each do |obj|
block.call(obj)
end
return nil
elsif block and ret != nil
raise "Return should return nil because of block but didnt. It wasnt an array either..."
elsif block
return nil
else
return ret
end
end
#Yields every object that is missing certain required objects (based on 'has_many' required-argument).
def list_invalid_required(args, &block)
enum = Enumerator.new do |yielder|
classname = args[:class]
classob = @args[:module].const_get(classname)
required_data = classob.required_data
if required_data and !required_data.empty?
required_data.each do |req_data|
self.list(args[:class], :cloned_ubuf => true) do |obj|
puts "Checking #{obj.classname}(#{obj.id}) for required #{req_data[:class]}." if args[:debug]
id = obj[req_data[:col]]
begin
raise Errno::ENOENT if !id
obj_req = self.get(req_data[:class], id)
rescue Errno::ENOENT
yielder << {:obj => obj, :type => :required, :id => id, :data => req_data}
end
end
end
end
end
return Knj.handle_return(:enum => enum, :block => block)
end
#Returns select-options-HTML for inserting into a HTML-select-element.
def list_opts(classname, args = {})
Knj::ArrayExt.hash_sym(args)
classname = classname.to_sym
if args[:list_args].is_a?(Hash)
list_args = args[:list_args]
else
list_args = {}
end
html = ""
if args[:addnew] or args[:add]
html << ""
elsif args[:none]
html << ""
end
self.list(classname, args[:list_args]) do |object|
html << ""
rescue => e
html << ">[#{object.class.name}: #{e.message}]"
end
end
return html
end
#Returns a hash which can be used to generate HTML-select-elements.
def list_optshash(classname, args = {})
Knj::ArrayExt.hash_sym(args)
classname = classname.to_sym
if args[:list_args].is_a?(Hash)
list_args = args[:list_args]
else
list_args = {}
end
if RUBY_VERSION[0..2] == 1.8 and Php4r.class_exists("Dictionary")
list = Dictionary.new
else
list = {}
end
if args[:addnew] or args[:add]
list["0"] = _("Add new")
elsif args[:choose]
list["0"] = _("Choose") + ":"
elsif args[:all]
list["0"] = _("All")
elsif args[:none]
list["0"] = _("None")
end
self.list(classname, args[:list_args]) do |object|
if object.respond_to?(:name)
list[object.id] = object.name
elsif object.respond_to?(:title)
list[object.id] = object.title
else
raise "Object of class '#{object.class.name}' doesnt support 'name' or 'title."
end
end
return list
end
#Returns a list of a specific object by running specific SQL against the database.
def list_bysql(classname, sql, args = nil, &block)
classname = classname.to_sym
ret = [] if !block
qargs = nil
if args
args.each do |key, val|
case key
when :cloned_ubuf
qargs = {:cloned_ubuf => true}
else
raise "Invalid key: '#{key}'."
end
end
end
if @args[:array_enum]
enum = Enumerator.new do |yielder|
@args[:db].q(sql, qargs) do |d_obs|
yielder << self.get(classname, d_obs)
end
end
if block
enum.each(&block)
return nil
else
return Array_enumerator.new(enum)
end
else
@args[:db].q(sql, qargs) do |d_obs|
if block
block.call(self.get(classname, d_obs))
else
ret << self.get(classname, d_obs)
end
end
if !block
return ret
else
return nil
end
end
end
#Add a new object to the database and to the cache.
#===Examples
# obj = ob.add(:User, {:username => "User 1"})
def add(classname, data = {}, args = nil)
raise "data-variable was not a hash: '#{data.class.name}'." if !data.is_a?(Hash)
classname = classname.to_sym
self.requireclass(classname)
if @args[:datarow]
classobj = @args[:module].const_get(classname)
#Run the class 'add'-method to check various data.
classobj.add(Knj::Hash_methods.new(:ob => self, :db => @args[:db], :data => data)) if classobj.respond_to?(:add)
#Check if various required data is given. If not then raise an error telling about it.
required_data = classobj.required_data
required_data.each do |req_data|
raise "No '#{req_data[:class]}' given by the data '#{req_data[:col]}'." if !data.key?(req_data[:col])
raise "The '#{req_data[:class]}' by ID '#{data[req_data[:col]]}' could not be found with the data '#{req_data[:col]}'." if !self.exists?(req_data[:class], data[req_data[:col]])
end
#If 'skip_ret' is given, then the ID wont be looked up and the object wont be spawned. Be aware the connected events wont be executed either. In return it will go a lot faster.
if args and args[:skip_ret] and !@ids_cache_should.key?(classname)
ins_args = nil
else
ins_args = {:return_id => true}
end
#Insert and (maybe?) get ID.
ins_id = @args[:db].insert(classobj.table, data, ins_args).to_i
#Add ID to ID-cache if ID-cache is active for that classname.
@ids_cache[classname][ins_id] = true if ins_id != 0 and @ids_cache_should.key?(classname)
#Skip the rest if we are told not to return result.
return nil if args and args[:skip_ret]
#Spawn the object.
retob = self.get(classname, ins_id, {:skip_reload => true})
elsif @args[:custom]
classobj = @args[:module].const_get(classname)
retob = classobj.add(Knj::Hash_methods.new(
:ob => self,
:data => data
))
else
args = [data]
args = args | @args[:extra_args] if @args[:extra_args]
retob = @args[:module].const_get(classname).add(*args)
end
self.call("object" => retob, "signal" => "add")
retob.send(:add_after, {}) if retob.respond_to?(:add_after)
return retob
end
#Adds several objects to the database at once. This is faster than adding every single object by itself, since this will do multi-inserts if supported by the database.
#===Examples
# ob.adds(:User, [{:username => "User 1"}, {:username => "User 2"})
def adds(classname, datas)
if !@args[:datarow]
datas.each do |data|
@args[:module].const_get(classname).add(*args)
self.call("object" => retob, "signal" => "add")
end
else
if @args[:module].const_get(classname).respond_to?(:add)
datas.each do |data|
@args[:module].const_get(classname).add(Knj::Hash_methods.new(
:ob => self,
:db => self.db,
:data => data
))
end
end
db.insert_multi(classname, datas)
end
self.cache_ids(classname)
end
#Calls a static method on a class. Passes the d-variable which contains the Objects-object, database-reference and more...
def static(class_name, method_name, *args, &block)
raise "Only available with datarow enabled." if !@args[:datarow] and !@args[:custom]
class_name = class_name
method_name = method_name
self.requireclass(class_name)
class_obj = @args[:module].const_get(class_name)
#Sometimes this raises the exception but actually responds to the class? Therefore commented out. - knj
#raise "The class '#{class_obj.name}' has no such method: '#{method_name}' (#{class_obj.methods.sort.join(", ")})." if !class_obj.respond_to?(method_name)
pass_args = []
if @args[:datarow]
pass_args << Knj::Hash_methods.new(:ob => self, :db => self.db)
else
pass_args << Knj::Hash_methods.new(:ob => self)
end
args.each do |arg|
pass_args << arg
end
class_obj.send(method_name, *pass_args, &block)
end
#Unset object. Do this if you are sure, that there are no more references left. This will be done automatically when deleting it.
def unset(object)
if object.is_a?(Array)
object.each do |obj|
unset(obj)
end
return nil
end
classname = object.class.name
if @args[:module]
classname = classname.gsub(@args[:module].name + "::", "")
end
classname = classname.to_sym
@objects[classname].delete(object.id.to_i)
end
def unset_class(classname)
if classname.is_a?(Array)
classname.each do |classn|
self.unset_class(classn)
end
return false
end
classname = classname.to_sym
return false if !@objects.key?(classname)
@objects.delete(classname)
end
#Delete an object. Both from the database and from the cache.
#===Examples
# user = ob.get(:User, 1)
# ob.delete(user)
def delete(object, args = nil)
#Return false if the object has already been deleted.
return false if object.deleted?
classname = object.class.classname.to_sym
self.call("object" => object, "signal" => "delete_before")
self.unset(object)
obj_id = object.id
object.delete if object.respond_to?(:delete)
if @args[:datarow]
#If autodelete is set by 'has_many'-method, go through it and delete the various objects first.
if autodelete_data = object.class.autodelete_data
autodelete_data.each do |adel_data|
self.list(adel_data[:classname], {adel_data[:colname].to_s => object.id}) do |obj_del|
self.delete(obj_del, args)
end
end
end
#If depend is set by 'has_many'-method, check if any objects exists and raise error if so.
if dep_datas = object.class.depending_data
dep_datas.each do |dep_data|
if obj = self.get_by(dep_data[:classname], {dep_data[:colname].to_s => object.id})
raise "Cannot delete <#{object.class.name}:#{object.id}> because <#{obj.class.name}:#{obj.id}> depends on it."
end
end
end
#If autozero is set by 'has_many'-method, check if any objects exists and set the ID to zero.
if autozero_datas = object.class.autozero_data
autozero_datas.each do |zero_data|
self.list(zero_data[:classname], {zero_data[:colname].to_s => object.id}) do |obj_zero|
obj_zero[zero_data[:colname].to_sym] = 0
end
end
end
#Delete any translations that has been set on the object by 'has_translation'-method.
if object.class.translations
begin
_hb.trans_del(object)
rescue NameError
_kas.trans_del(object)
end
end
#If a buffer is given in arguments, then use that to delete the object.
if args and buffer = args[:db_buffer]
buffer.delete(object.table, {:id => obj_id})
else
@args[:db].delete(object.table, {:id => obj_id})
end
end
@ids_cache[classname].delete(obj_id.to_i) if @ids_cache_should.key?(classname)
self.call("object" => object, "signal" => "delete")
object.destroy
return nil
end
#Deletes several objects as one. If running datarow-mode it checks all objects before it starts to actually delete them. Its faster than deleting every single object by itself...
def deletes(objs)
if !@args[:datarow]
objs.each do |obj|
self.delete(obj)
end
else
tables = {}
begin
objs.each do |obj|
next if obj.deleted?
tablen = obj.table
if !tables.key?(tablen)
tables[tablen] = []
end
tables[tablen] << obj.id
obj.delete if obj.respond_to?(:delete)
#Remove from ID-cache.
classname = obj.class.classname.to_sym
@ids_cache[classname].delete(obj.id.to_i) if @ids_cache_should.key?(classname)
#Unset any data on the object, so it seems deleted.
obj.destroy
end
ensure
#An exception may occur, and we should make sure, that objects that has gotten 'delete' called also are deleted from their tables.
tables.each do |table, ids|
ids.each_slice(1000) do |ids_slice|
@args[:db].delete(table, {:id => ids_slice})
end
end
end
end
end
#Deletes all objects with the given IDs 500 at a time to prevent memory exhaustion or timeout.
#===Examples
# ob.delete_ids(:class => :Person, :ids => [1, 3, 5, 6, 7, 8, 9])
def delete_ids(args)
while !args[:ids].empty? and ids = args[:ids].shift(500)
objs = self.list(args[:class], "id" => ids)
self.deletes(objs)
end
return nil
end
#Try to clean up objects by unsetting everything, start the garbagecollector, get all the remaining objects via ObjectSpace and set them again. Some (if not all) should be cleaned up and our cache should still be safe... dirty but works.
def clean(classn)
if classn.is_a?(Array)
classn.each do |realclassn|
self.clean(realclassn)
end
return nil
end
if @args[:cache] == :weak
@objects[classn].clean
elsif @args[:cache] == :none
return false
else
return false if !@objects.key?(classn)
@objects[classn] = {}
GC.start
@objects.keys.each do |classn|
data = @objects[classn]
classobj = @args[:module].const_get(classn)
ObjectSpace.each_object(classobj) do |obj|
begin
data[obj.id.to_i] = obj
rescue => e
if e.message == "No data on object."
#Object has been unset - skip it.
next
end
raise e
end
end
end
end
end
#Erases the whole cache and regenerates it from ObjectSpace if not running weak-link-caching. If running weaklink-caching then it will only removes the dead links.
def clean_all
self.clean(@objects.keys)
end
def classes_loaded
return @objects.keys
end
#This method helps build SQL from Objects-instances list-method. It should not be called directly but only through Objects.list.
def sqlhelper(list_args, args_def)
args = args_def
if args[:db]
db = args[:db]
else
db = @args[:db]
end
if args[:table]
table_def = "`#{db.escape_table(args[:table])}`."
else
table_def = ""
end
sql_joins = ""
sql_where = ""
sql_order = ""
sql_limit = ""
sql_groupby = ""
do_joins = {}
limit_from = nil
limit_to = nil
if list_args.key?("orderby")
orders = []
orderstr = list_args["orderby"]
list_args["orderby"] = [list_args["orderby"]] if list_args["orderby"].is_a?(Hash)
if list_args["orderby"].is_a?(String)
found = false
found = true if args[:cols].key?(orderstr)
if found
sql_order << " ORDER BY "
ordermode = " ASC"
if list_args.key?("ordermode")
if list_args["ordermode"] == "desc"
ordermode = " DESC"
elsif list_args["ordermode"] == "asc"
ordermode = " ASC"
raise "Unknown ordermode: #{list_args["ordermode"]}"
end
list_args.delete("ordermode")
end
sql_order << "#{table_def}`#{db.escape_column(list_args["orderby"])}`#{ordermode}"
list_args.delete("orderby")
end
elsif list_args["orderby"].is_a?(Array)
sql_order << " ORDER BY "
list_args["orderby"].each do |val|
ordermode = nil
orderstr = nil
found = false
if val.is_a?(Array)
if val[1] == "asc"
ordermode = " ASC"
elsif val[1] == "desc"
ordermode = " DESC"
end
if val[0].is_a?(Array)
if args[:joined_tables]
args[:joined_tables].each do |table_name, table_data|
next if table_name.to_s != val[0][0].to_s
do_joins[table_name] = true
orders << "`#{db.escape_table(table_name)}`.`#{db.escape_column(val[0][1])}`#{ordermode}"
found = true
break
end
end
raise "Could not find joined table for ordering: '#{val[0][0]}'." if !found
else
orderstr = val[0]
end
elsif val.is_a?(String)
orderstr = val
ordermode = " ASC"
elsif val.is_a?(Hash) and val[:type] == :sql
orders << val[:sql]
found = true
elsif val.is_a?(Hash) and val[:type] == :case
caseorder = " CASE"
val[:case].each do |key, caseval|
col = key.first
isval = key.last
col_str = nil
if col.is_a?(Array)
raise "No joined tables for '#{args[:table]}'." if !args[:joined_tables]
found = false
args[:joined_tables].each do |table_name, table_data|
if table_name == col.first
do_joins[table_name] = true
col_str = "`#{db.escape_table(table_name)}`.`#{db.escape_column(col.last)}`"
found = true
break
end
end
raise "No such joined table on '#{args[:table]}': '#{col.first}' (#{col.first.class.name}) with the following joined table:\n#{Php4r.print_r(args[:joined_tables], true)}" if !found
elsif col.is_a?(String) or col.is_a?(Symbol)
col_str = "#{table_def}`#{col}`"
found = true
else
raise "Unknown type for case-ordering: '#{col.class.name}'."
end
raise "'colstr' was not set." if !col_str
caseorder << " WHEN #{col_str} = '#{db.esc(isval)}' THEN '#{db.esc(caseval)}'"
end
if val[:else]
caseorder << " ELSE '#{db.esc(val[:else])}'"
end
caseorder << " END"
orders << caseorder
elsif val.is_a?(Hash)
raise "No joined tables." if !args.key?(:joined_tables)
if val[:mode] == "asc"
ordermode = " ASC"
elsif val[:mode] == "desc"
ordermode = " DESC"
end
if args[:joined_tables]
args[:joined_tables].each do |table_name, table_data|
if table_data[:parent_table]
table_name_real = table_name
elsif table_data[:datarow]
table_name_real = self.datarow_from_datarow_argument(table_data[:datarow]).classname
else
table_name_real = @args[:module].const_get(table_name).classname
end
if table_name.to_s == val[:table].to_s
do_joins[table_name] = true
if val[:sql]
orders << val[:sql]
elsif val[:col]
orders << "`#{db.escape_table(table_name_real)}`.`#{db.escape_column(val[:col])}`#{ordermode}"
else
raise "Couldnt figure out how to order based on keys: '#{val.keys.sort}'."
end
found = true
break
end
end
end
else
raise "Unknown object: #{val.class.name}"
end
found = true if args[:cols].key?(orderstr)
if !found
raise "Column not found for ordering: #{orderstr}."
end
orders << "#{table_def}`#{db.escape_column(orderstr)}`#{ordermode}" if orderstr
end
sql_order << orders.join(", ")
list_args.delete("orderby")
else
raise "Unknown orderby object: #{list_args["orderby"].class.name}."
end
end
list_args.each do |realkey, val|
found = false
if realkey.is_a?(Array)
if !args[:joins_skip]
datarow_obj = self.datarow_obj_from_args(args_def, list_args, realkey[0])
args = datarow_obj.columns_sqlhelper_args
raise "Couldnt get arguments from SQLHelper." if !args
else
datarow_obj = @args[:module].const_get(realkey[0])
args = args_def
end
table_sym = realkey[0].to_sym
do_joins[table_sym] = true
list_table_name_real = table_sym
table = "`#{db.escape_table(list_table_name_real)}`."
key = realkey[1]
else
table = table_def
args = args_def
key = realkey
end
if args.key?(:cols_bools) and args[:cols_bools].index(key) != nil
val_s = val.to_s
if val_s == "1" or val_s == "true"
realval = "1"
elsif val_s == "0" or val_s == "false"
realval = "0"
else
raise "Could not make real value out of class: #{val.class.name} => #{val}."
end
sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(realval)}'"
found = true
elsif args[:cols].key?(key)
if val.is_a?(Array)
if val.empty? and db.opts[:type].to_s == "mysql"
sql_where << " AND false"
else
escape_sql = Knj::ArrayExt.join(
:arr => val,
:callback => proc{|value|
db.escape(value)
},
:sep => ",",
:surr => "'"
)
sql_where << " AND #{table}`#{db.escape_column(key)}` IN (#{escape_sql})"
end
elsif val.is_a?(Hash) and val[:type].to_sym == :col
raise "No table was given for join: '#{val}', key: '#{key}' on table #{table}." if !val.key?(:table)
do_joins[val[:table].to_sym] = true
sql_where << " AND #{table}`#{db.escape_column(key)}` = `#{db.escape_table(val[:table])}`.`#{db.escape_column(val[:name])}`"
elsif val.is_a?(Hash) and val[:type] == :sqlval and val[:val] == :null
sql_where << " AND #{table}`#{db.escape_column(key)}` IS NULL"
elsif val.is_a?(Proc)
call_args = Knj::Hash_methods.new(:ob => self, :db => db)
sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(val.call(call_args))}'"
else
sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(val)}'"
end
found = true
elsif key.to_s == "limit_from"
limit_from = val.to_i
found = true
elsif key.to_s == "limit_to"
limit_to = val.to_i
found = true
elsif key.to_s == "limit"
limit_from = 0
limit_to = val.to_i
found = true
elsif args.key?(:cols_dbrows) and args[:cols_dbrows].index("#{key.to_s}_id") != nil
if val == false
sql_where << " AND #{table}`#{db.escape_column(key.to_s + "_id")}` = '0'"
elsif val.is_a?(Array)
if val.empty?
sql_where << " AND false"
else
sql_where << " AND #{table}`#{db.escape_column("#{key}_id")}` IN (#{Knj::ArrayExt.join(:arr => val, :sep => ",", :surr => "'", :callback => proc{|obj| obj.id.sql})})"
end
else
sql_where << " AND #{table}`#{db.escape_column(key.to_s + "_id")}` = '#{db.esc(val.id)}'"
end
found = true
elsif match = key.match(/^([A-z_\d]+)_(search|has)$/) and args[:cols].key?(match[1]) != nil
if match[2] == "search"
Knj::Strings.searchstring(val).each do |str|
sql_where << " AND #{table}`#{db.escape_column(match[1])}` LIKE '%#{db.esc(str)}%'"
end
elsif match[2] == "has"
if val
sql_where << " AND #{table}`#{db.escape_column(match[1])}` != ''"
else
sql_where << " AND #{table}`#{db.escape_column(match[1])}` = ''"
end
end
found = true
elsif match = key.match(/^([A-z_\d]+)_(not|lower)$/) and args[:cols].key?(match[1])
if match[2] == "not"
if val.is_a?(Array)
if val.empty?
#ignore.
else
escape_sql = Knj::ArrayExt.join(
:arr => val,
:callback => proc{|value|
db.escape(value)
},
:sep => ",",
:surr => "'"
)
sql_where << " AND #{table}`#{db.escape_column(match[1])}` NOT IN (#{escape_sql})"
end
else
sql_where << " AND #{table}`#{db.escape_column(match[1])}` != '#{db.esc(val)}'"
end
elsif match[2] == "lower"
sql_where << " AND LOWER(#{table}`#{db.escape_column(match[1])}`) = LOWER('#{db.esc(val)}')"
else
raise "Unknown mode: '#{match[2]}'."
end
found = true
elsif args.key?(:cols_date) and match = key.match(/^(.+)_(day|week|month|year|from|to|below|above)(|_(not))$/) and args[:cols_date].index(match[1]) != nil
not_v = match[4]
val = Datet.in(val) if val.is_a?(Time)
if match[2] == "day"
if val.is_a?(Array)
sql_where << " AND ("
first = true
val.each do |realval|
if first
first = false
else
sql_where << " OR "
end
sql_where << "#{db.sqlspecs.strftime("%d %m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%d %m %Y", "'#{db.esc(realval.dbstr)}'")}"
end
sql_where << ")"
else
sql_where << " AND #{db.sqlspecs.strftime("%d %m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%d %m %Y", "'#{db.esc(val.dbstr)}'")}"
end
elsif match[2] == "week"
sql_where << " AND #{db.sqlspecs.strftime("%W %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%W %Y", "'#{db.esc(val.dbstr)}'")}"
elsif match[2] == "month"
sql_where << " AND #{db.sqlspecs.strftime("%m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%m %Y", "'#{db.esc(val.dbstr)}'")}"
elsif match[2] == "year"
sql_where << " AND #{db.sqlspecs.strftime("%Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%Y", "'#{db.esc(val.dbstr)}'")}"
elsif match[2] == "from" or match[2] == "above"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` >= '#{db.esc(val.dbstr)}'"
elsif match[2] == "to" or match[2] == "below"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` <= '#{db.esc(val.dbstr)}'"
else
raise "Unknown date-key: #{match[2]}."
end
found = true
elsif args.key?(:cols_num) and match = key.match(/^(.+)_(from|to|above|below|numeric)$/) and args[:cols_num].index(match[1]) != nil
if match[2] == "from"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` >= '#{db.esc(val)}'"
elsif match[2] == "to"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` <= '#{db.esc(val)}'"
elsif match[2] == "above"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` > '#{db.esc(val)}'"
elsif match[2] == "below"
sql_where << " AND #{table}`#{db.escape_column(match[1])}` < '#{db.esc(val)}'"
else
raise "Unknown method of treating cols-num-argument: #{match[2]}."
end
found = true
elsif match = key.match(/^(.+)_lookup$/) and args[:cols].key?("#{match[1]}_id") and args[:cols].key?("#{match[1]}_class")
sql_where << " AND #{table}`#{db.escape_column("#{match[1]}_class")}` = '#{db.esc(val.table)}'"
sql_where << " AND #{table}`#{db.escape_column("#{match[1]}_id")}` = '#{db.esc(val.id)}'"
found = true
elsif realkey == "groupby"
found = true
if val.is_a?(Array)
val.each do |col_name|
raise "Column '#{val}' not found on table '#{table}'." if !args[:cols].key?(col_name)
sql_groupby << ", " if sql_groupby.length > 0
sql_groupby << "#{table}`#{db.escape_column(col_name)}`"
end
elsif val.is_a?(String)
sql_groupby << ", " if sql_groupby.length > 0
sql_groupby << "#{table}`#{db.escape_column(val)}`"
else
raise "Unknown class given for 'groupby': '#{val.class.name}'."
end
end
list_args.delete(realkey) if found
end
args = args_def
if !args[:joins_skip]
raise "No joins defined on '#{args[:table]}' for: '#{args[:table]}'." if !do_joins.empty? and !args[:joined_tables]
do_joins.each do |table_name, temp_val|
raise "No join defined on table '#{args[:table]}' for table '#{table_name}'." if !args[:joined_tables].key?(table_name)
table_data = args[:joined_tables][table_name]
if table_data.key?(:parent_table)
join_table_name_real = table_name
sql_joins << " LEFT JOIN `#{table_data[:parent_table]}` AS `#{table_name}` ON 1=1"
else
const = @args[:module].const_get(table_name)
join_table_name_real = const.classname
sql_joins << " LEFT JOIN `#{const.table}` AS `#{const.classname}` ON 1=1"
end
if table_data[:ob]
ob = table_data[:ob]
else
ob = self
end
class_name = args[:table].to_sym
if table_data[:datarow]
datarow = self.datarow_from_datarow_argument(table_data[:datarow])
else
self.requireclass(class_name) if @objects.key?(class_name)
datarow = @args[:module].const_get(class_name)
end
if !datarow.columns_sqlhelper_args
ob.requireclass(datarow.table.to_sym)
raise "No SQL-helper-args on class '#{datarow.table}' ???" if !datarow.columns_sqlhelper_args
end
newargs = datarow.columns_sqlhelper_args.clone
newargs[:table] = join_table_name_real
newargs[:joins_skip] = true
#Clone the where-arguments and run them against another sqlhelper to sub-join.
join_args = table_data[:where].clone
ret = self.sqlhelper(join_args, newargs)
sql_joins << ret[:sql_where]
#If any of the join-arguments are left, then we should throw an error.
join_args.each do |key, val|
raise "Invalid key '#{key}' when trying to join table '#{table_name}' on table '#{args_def[:table]}'."
end
end
end
#If limit arguments has been given then add them.
if limit_from and limit_to
sql_limit = " LIMIT #{limit_from}, #{limit_to}"
end
sql_groupby = nil if sql_groupby.length <= 0
return {
:sql_joins => sql_joins,
:sql_where => sql_where,
:sql_limit => sql_limit,
:sql_order => sql_order,
:sql_groupby => sql_groupby
}
end
#Used by sqlhelper-method to look up datarow-classes and automatically load them if they arent loaded already.
def datarow_obj_from_args(args, list_args, class_name)
class_name = class_name.to_sym
if !args.key?(:joined_tables)
raise "No joined tables on '#{args[:table]}' to find datarow for: '#{class_name}'."
end
args[:joined_tables].each do |table_name, table_data|
next if table_name.to_sym != class_name
return self.datarow_from_datarow_argument(table_data[:datarow]) if table_data[:datarow]
self.requireclass(class_name) if @objects.key?(class_name)
return @args[:module].const_get(class_name)
end
raise "Could not figure out datarow for: '#{class_name}'."
end
def datarow_from_datarow_argument(datarow_argument)
if datarow_argument.is_a?(String)
const = Knj::Strings.const_get_full(datarow_argument)
else
const = datarow_argument
end
self.load_class(datarow_argument.to_s.split("::").last) if !const.initialized? #Make sure the class is initialized.
return const
end
def not(not_v, val)
if not_v == "not" or not_v == "not_"
return val
end
return ""
end
end
require "#{$knjpath}objects/objects_sqlhelper"