Class: FC::DbBase

Inherits:
Object
  • Object
show all
Defined in:
lib/fc/base.rb

Direct Known Subclasses

CopyRule, Error, Item, ItemStorage, Policy, Storage

Class Attribute Summary collapse

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(params = {}) ⇒ DbBase

Returns a new instance of DbBase.



11
12
13
14
15
16
 
# File 'lib/fc/base.rb', line 11

def initialize(params = {})
  self.class.table_fields.each {|key| self.send("#{key}=", params[key] || params[key.to_sym]) }
  @id = (params["id"] || params[:id]).to_i if params["id"] || params[:id]
  @database_fields = params[:database_fields] || {}
  @additional_fields = params[:additional_fields] || {}
end

Class Attribute Details

.table_fields ⇒ Object

Returns the value of attribute table_fields.



8
9
10
 
# File 'lib/fc/base.rb', line 8

def table_fields
  @table_fields
end

.table_name ⇒ Object

Returns the value of attribute table_name.



8
9
10
 
# File 'lib/fc/base.rb', line 8

def table_name
  @table_name
end

Instance Attribute Details

#additional_fields ⇒ Object

Returns the value of attribute additional_fields.



5
6
7
 
# File 'lib/fc/base.rb', line 5

def additional_fields
  @additional_fields
end

#database_fields ⇒ Object

Returns the value of attribute database_fields.



5
6
7
 
# File 'lib/fc/base.rb', line 5

def database_fields
  @database_fields
end

#id ⇒ Object

Returns the value of attribute id.



5
6
7
 
# File 'lib/fc/base.rb', line 5

def id
  @id
end

Class Method Details

.apply!(data: {}, key_field: :name, update_only: false) ⇒ Object 



103
104
105
106
107
108
109
110
111
 
# File 'lib/fc/base.rb', line 103

def self.apply!(data: {}, key_field: :name, update_only: false)
  return nil unless data[key_field]
  obj_instance = where("#{key_field} = ?", data[key_field]).first
  return "#{name} \"#{data[key_field]}\" not found" if obj_instance.nil? && update_only
  obj_instance ||= new
  obj_instance.load(data: data)
  obj_instance.save
  obj_instance
end

.create_from_fiels(data) ⇒ Object

make instance on fields hash



31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/fc/base.rb', line 31

def self.create_from_fiels(data)
  # use only defined in set_table fields
  #database_fields = data.select{|key, val| self.table_fields.include?(key.to_s)}
  additional_fields = {}
  database_fields = {}
  data.each do |key, val| 
    if self.table_fields.include?(key.to_s)
      database_fields[key] = val
    elsif key != 'id'
      additional_fields[key.to_sym] = val
    end
  end
  self.new(database_fields.merge({:id => data["id"].to_s, :database_fields => database_fields, :additional_fields => additional_fields}))
end

.find(id) ⇒ Object

get element by id



47
48
49
50
51
 
# File 'lib/fc/base.rb', line 47

def self.find(id)
  r = FC::DB.query("SELECT * FROM #{self.table_name} WHERE id=#{id.to_i}")
  raise "Record not found (#{self.table_name}.id=#{id})" if r.count == 0
  self.create_from_fiels(r.first)
end

.set_table(name, fields) ⇒ Object

define table name and fields



24
25
26
27
28
 
# File 'lib/fc/base.rb', line 24

def self.set_table(name, fields)
  @table_name = name
  self.table_fields = fields.split(',').map{|e| e.gsub(' ','')}
  self.table_fields.each{|e| attr_accessor e.to_sym}
end

.where(cond = "1", *params) ⇒ Object

get elements array by sql condition (possibles ‘?’ placeholders)



54
55
56
57
58
59
 
# File 'lib/fc/base.rb', line 54

def self.where(cond = "1", *params)
  i = -1
  sql = "SELECT * FROM #{self.table_name} WHERE #{cond.gsub('?'){i+=1; "'#{Mysql2::Client.escape(params[i].to_s)}'"}}"
  r = FC::DB.query(sql)
  r.map{|data| self.create_from_fiels(data)}
end

Instance Method Details

#delete ⇒ Object

delete object from DB



99
100
101
 
# File 'lib/fc/base.rb', line 99

def delete
  FC::DB.query("DELETE FROM #{self.class.table_name} WHERE id=#{@id.to_i}") if @id
end

#dump(exclude = []) ⇒ Object 



119
120
121
122
123
124
125
126
127
 
# File 'lib/fc/base.rb', line 119

def dump(exclude = [])
  obj_dump = {}
  self.class.table_fields.each do |field|
    sym_field = field.to_sym
    next if exclude.include?(sym_field)
    obj_dump[sym_field] = database_fields[field]
  end
  obj_dump
end

#load(data: {}) ⇒ Object 



113
114
115
116
117
 
# File 'lib/fc/base.rb', line 113

def load(data: {})
  self.class.table_fields.each do |field|
    send("#{field}=", data[field.to_sym]) if data[field.to_sym]
  end
end

#reload ⇒ Object

reload object from DB



91
92
93
94
95
96
 
# File 'lib/fc/base.rb', line 91

def reload
  raise "Can't reload object without id" if !@id || @id.to_i == 0
  new_obj = self.class.find(@id)
  self.database_fields = new_obj.database_fields
  self.class.table_fields.each {|key| self.send("#{key}=", new_obj.send(key)) }
end

#save ⇒ Object

save changed fields



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
 
# File 'lib/fc/base.rb', line 62

def save
  sql = @id.to_i != 0 ? "UPDATE #{self.class.table_name} SET " : "INSERT IGNORE INTO #{self.class.table_name} SET "
  fields = []
  self.class.table_fields.each do |key|
    db_val = @database_fields[key]
    val = self.send(key)
    val = 1 if val == true
    val = 0 if val == false
    if val.to_s != db_val.to_s || val.nil? && !db_val.nil? || !val.nil? && db_val.nil?
      fields << "#{key}=#{val ? (val.class == String ? "'#{FC::DB.connect.escape(val)}'" : val.to_i) : 'NULL'}"
    end
  end
  @additional_fields.each do |key, val|
    val = 1 if val == true
    val = 0 if val == false
    fields << "#{key}=#{val ? (val.class == String ? "'#{FC::DB.connect.escape(val)}'" : val.to_i) : 'NULL'}"
  end
  if fields.length > 0
    sql << fields.join(',')
    sql << " WHERE id=#{@id.to_i}" if @id
    FC::DB.query(sql)
    @id = FC::DB.connect.last_id unless @id
    self.class.table_fields.each do |key|
      @database_fields[key] = self.send(key)
    end
  end
end