| 1 |
require 'base64' |
|---|
| 2 |
require 'yaml' |
|---|
| 3 |
require 'set' |
|---|
| 4 |
require 'active_record/deprecated_finders' |
|---|
| 5 |
|
|---|
| 6 |
module ActiveRecord |
|---|
| 7 |
class ActiveRecordError < StandardError |
|---|
| 8 |
end |
|---|
| 9 |
class SubclassNotFound < ActiveRecordError |
|---|
| 10 |
end |
|---|
| 11 |
class AssociationTypeMismatch < ActiveRecordError |
|---|
| 12 |
end |
|---|
| 13 |
class SerializationTypeMismatch < ActiveRecordError |
|---|
| 14 |
end |
|---|
| 15 |
class AdapterNotSpecified < ActiveRecordError |
|---|
| 16 |
end |
|---|
| 17 |
class AdapterNotFound < ActiveRecordError |
|---|
| 18 |
end |
|---|
| 19 |
class ConnectionNotEstablished < ActiveRecordError |
|---|
| 20 |
end |
|---|
| 21 |
class ConnectionFailed < ActiveRecordError |
|---|
| 22 |
end |
|---|
| 23 |
class RecordNotFound < ActiveRecordError |
|---|
| 24 |
end |
|---|
| 25 |
class RecordNotSaved < ActiveRecordError |
|---|
| 26 |
end |
|---|
| 27 |
class StatementInvalid < ActiveRecordError |
|---|
| 28 |
end |
|---|
| 29 |
class PreparedStatementInvalid < ActiveRecordError |
|---|
| 30 |
end |
|---|
| 31 |
class StaleObjectError < ActiveRecordError |
|---|
| 32 |
end |
|---|
| 33 |
class ConfigurationError < StandardError |
|---|
| 34 |
end |
|---|
| 35 |
class ReadOnlyRecord < StandardError |
|---|
| 36 |
end |
|---|
| 37 |
|
|---|
| 38 |
class AttributeAssignmentError < ActiveRecordError |
|---|
| 39 |
attr_reader :exception, :attribute |
|---|
| 40 |
def initialize(message, exception, attribute) |
|---|
| 41 |
@exception = exception |
|---|
| 42 |
@attribute = attribute |
|---|
| 43 |
@message = message |
|---|
| 44 |
end |
|---|
| 45 |
end |
|---|
| 46 |
|
|---|
| 47 |
class MultiparameterAssignmentErrors < ActiveRecordError |
|---|
| 48 |
attr_reader :errors |
|---|
| 49 |
def initialize(errors) |
|---|
| 50 |
@errors = errors |
|---|
| 51 |
end |
|---|
| 52 |
end |
|---|
| 53 |
|
|---|
| 54 |
|
|---|
| 55 |
|
|---|
| 56 |
|
|---|
| 57 |
|
|---|
| 58 |
|
|---|
| 59 |
|
|---|
| 60 |
|
|---|
| 61 |
|
|---|
| 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 |
|
|---|
| 89 |
|
|---|
| 90 |
|
|---|
| 91 |
|
|---|
| 92 |
|
|---|
| 93 |
|
|---|
| 94 |
|
|---|
| 95 |
|
|---|
| 96 |
|
|---|
| 97 |
|
|---|
| 98 |
|
|---|
| 99 |
|
|---|
| 100 |
|
|---|
| 101 |
|
|---|
| 102 |
|
|---|
| 103 |
|
|---|
| 104 |
|
|---|
| 105 |
|
|---|
| 106 |
|
|---|
| 107 |
|
|---|
| 108 |
|
|---|
| 109 |
|
|---|
| 110 |
|
|---|
| 111 |
|
|---|
| 112 |
|
|---|
| 113 |
|
|---|
| 114 |
|
|---|
| 115 |
|
|---|
| 116 |
|
|---|
| 117 |
|
|---|
| 118 |
|
|---|
| 119 |
|
|---|
| 120 |
|
|---|
| 121 |
|
|---|
| 122 |
|
|---|
| 123 |
|
|---|
| 124 |
|
|---|
| 125 |
|
|---|
| 126 |
|
|---|
| 127 |
|
|---|
| 128 |
|
|---|
| 129 |
|
|---|
| 130 |
|
|---|
| 131 |
|
|---|
| 132 |
|
|---|
| 133 |
|
|---|
| 134 |
|
|---|
| 135 |
|
|---|
| 136 |
|
|---|
| 137 |
|
|---|
| 138 |
|
|---|
| 139 |
|
|---|
| 140 |
|
|---|
| 141 |
|
|---|
| 142 |
|
|---|
| 143 |
|
|---|
| 144 |
|
|---|
| 145 |
|
|---|
| 146 |
|
|---|
| 147 |
|
|---|
| 148 |
|
|---|
| 149 |
|
|---|
| 150 |
|
|---|
| 151 |
|
|---|
| 152 |
|
|---|
| 153 |
|
|---|
| 154 |
|
|---|
| 155 |
|
|---|
| 156 |
|
|---|
| 157 |
|
|---|
| 158 |
|
|---|
| 159 |
|
|---|
| 160 |
|
|---|
| 161 |
|
|---|
| 162 |
|
|---|
| 163 |
|
|---|
| 164 |
|
|---|
| 165 |
|
|---|
| 166 |
|
|---|
| 167 |
|
|---|
| 168 |
|
|---|
| 169 |
|
|---|
| 170 |
|
|---|
| 171 |
|
|---|
| 172 |
|
|---|
| 173 |
|
|---|
| 174 |
|
|---|
| 175 |
|
|---|
| 176 |
|
|---|
| 177 |
|
|---|
| 178 |
|
|---|
| 179 |
|
|---|
| 180 |
|
|---|
| 181 |
|
|---|
| 182 |
|
|---|
| 183 |
|
|---|
| 184 |
|
|---|
| 185 |
|
|---|
| 186 |
|
|---|
| 187 |
|
|---|
| 188 |
|
|---|
| 189 |
|
|---|
| 190 |
|
|---|
| 191 |
|
|---|
| 192 |
|
|---|
| 193 |
|
|---|
| 194 |
|
|---|
| 195 |
|
|---|
| 196 |
|
|---|
| 197 |
|
|---|
| 198 |
|
|---|
| 199 |
|
|---|
| 200 |
|
|---|
| 201 |
|
|---|
| 202 |
|
|---|
| 203 |
|
|---|
| 204 |
|
|---|
| 205 |
|
|---|
| 206 |
|
|---|
| 207 |
|
|---|
| 208 |
|
|---|
| 209 |
|
|---|
| 210 |
|
|---|
| 211 |
|
|---|
| 212 |
|
|---|
| 213 |
|
|---|
| 214 |
|
|---|
| 215 |
|
|---|
| 216 |
|
|---|
| 217 |
|
|---|
| 218 |
|
|---|
| 219 |
|
|---|
| 220 |
|
|---|
| 221 |
|
|---|
| 222 |
|
|---|
| 223 |
|
|---|
| 224 |
|
|---|
| 225 |
|
|---|
| 226 |
|
|---|
| 227 |
|
|---|
| 228 |
|
|---|
| 229 |
|
|---|
| 230 |
|
|---|
| 231 |
|
|---|
| 232 |
|
|---|
| 233 |
|
|---|
| 234 |
|
|---|
| 235 |
|
|---|
| 236 |
|
|---|
| 237 |
|
|---|
| 238 |
|
|---|
| 239 |
|
|---|
| 240 |
|
|---|
| 241 |
|
|---|
| 242 |
|
|---|
| 243 |
|
|---|
| 244 |
|
|---|
| 245 |
|
|---|
| 246 |
|
|---|
| 247 |
|
|---|
| 248 |
|
|---|
| 249 |
|
|---|
| 250 |
|
|---|
| 251 |
|
|---|
| 252 |
|
|---|
| 253 |
|
|---|
| 254 |
|
|---|
| 255 |
|
|---|
| 256 |
|
|---|
| 257 |
|
|---|
| 258 |
|
|---|
| 259 |
|
|---|
| 260 |
|
|---|
| 261 |
|
|---|
| 262 |
class Base |
|---|
| 263 |
|
|---|
| 264 |
|
|---|
| 265 |
cattr_accessor :logger |
|---|
| 266 |
|
|---|
| 267 |
include Reloadable::Deprecated |
|---|
| 268 |
|
|---|
| 269 |
def self.inherited(child) |
|---|
| 270 |
@@subclasses[self] ||= [] |
|---|
| 271 |
@@subclasses[self] << child |
|---|
| 272 |
super |
|---|
| 273 |
end |
|---|
| 274 |
|
|---|
| 275 |
def self.reset_subclasses |
|---|
| 276 |
nonreloadables = [] |
|---|
| 277 |
subclasses.each do |klass| |
|---|
| 278 |
unless Dependencies.autoloaded? klass |
|---|
| 279 |
nonreloadables << klass |
|---|
| 280 |
next |
|---|
| 281 |
end |
|---|
| 282 |
klass.instance_variables.each { |var| klass.send(:remove_instance_variable, var) } |
|---|
| 283 |
klass.instance_methods(false).each { |m| klass.send :undef_method, m } |
|---|
| 284 |
end |
|---|
| 285 |
@@subclasses = {} |
|---|
| 286 |
nonreloadables.each { |klass| (@@subclasses[klass.superclass] ||= []) << klass } |
|---|
| 287 |
end |
|---|
| 288 |
|
|---|
| 289 |
@@subclasses = {} |
|---|
| 290 |
|
|---|
| 291 |
cattr_accessor :configurations |
|---|
| 292 |
@@configurations = {} |
|---|
| 293 |
|
|---|
| 294 |
|
|---|
| 295 |
|
|---|
| 296 |
|
|---|
| 297 |
|
|---|
| 298 |
cattr_accessor :primary_key_prefix_type |
|---|
| 299 |
@@primary_key_prefix_type = nil |
|---|
| 300 |
|
|---|
| 301 |
|
|---|
| 302 |
|
|---|
| 303 |
|
|---|
| 304 |
cattr_accessor :table_name_prefix |
|---|
| 305 |
@@table_name_prefix = "" |
|---|
| 306 |
|
|---|
| 307 |
|
|---|
| 308 |
|
|---|
| 309 |
cattr_accessor :table_name_suffix |
|---|
| 310 |
@@table_name_suffix = "" |
|---|
| 311 |
|
|---|
| 312 |
|
|---|
| 313 |
|
|---|
| 314 |
|
|---|
| 315 |
cattr_accessor :pluralize_table_names |
|---|
| 316 |
@@pluralize_table_names = true |
|---|
| 317 |
|
|---|
| 318 |
|
|---|
| 319 |
|
|---|
| 320 |
|
|---|
| 321 |
cattr_accessor :colorize_logging |
|---|
| 322 |
@@colorize_logging = true |
|---|
| 323 |
|
|---|
| 324 |
|
|---|
| 325 |
|
|---|
| 326 |
cattr_accessor :default_timezone |
|---|
| 327 |
@@default_timezone = :local |
|---|
| 328 |
|
|---|
| 329 |
|
|---|
| 330 |
|
|---|
| 331 |
cattr_accessor :allow_concurrency |
|---|
| 332 |
@@allow_concurrency = false |
|---|
| 333 |
|
|---|
| 334 |
|
|---|
| 335 |
|
|---|
| 336 |
|
|---|
| 337 |
|
|---|
| 338 |
cattr_accessor :generate_read_methods |
|---|
| 339 |
@@generate_read_methods = true |
|---|
| 340 |
|
|---|
| 341 |
|
|---|
| 342 |
|
|---|
| 343 |
|
|---|
| 344 |
|
|---|
| 345 |
|
|---|
| 346 |
|
|---|
| 347 |
cattr_accessor :schema_format |
|---|
| 348 |
@@schema_format = :ruby |
|---|
| 349 |
|
|---|
| 350 |
class << self |
|---|
| 351 |
|
|---|
| 352 |
|
|---|
| 353 |
|
|---|
| 354 |
|
|---|
| 355 |
|
|---|
| 356 |
|
|---|
| 357 |
|
|---|
| 358 |
|
|---|
| 359 |
|
|---|
| 360 |
|
|---|
| 361 |
|
|---|
| 362 |
|
|---|
| 363 |
|
|---|
| 364 |
|
|---|
| 365 |
|
|---|
| 366 |
|
|---|
| 367 |
|
|---|
| 368 |
|
|---|
| 369 |
|
|---|
| 370 |
|
|---|
| 371 |
|
|---|
| 372 |
|
|---|
| 373 |
|
|---|
| 374 |
|
|---|
| 375 |
|
|---|
| 376 |
|
|---|
| 377 |
|
|---|
| 378 |
|
|---|
| 379 |
|
|---|
| 380 |
|
|---|
| 381 |
|
|---|
| 382 |
|
|---|
| 383 |
|
|---|
| 384 |
|
|---|
| 385 |
|
|---|
| 386 |
|
|---|
| 387 |
|
|---|
| 388 |
|
|---|
| 389 |
|
|---|
| 390 |
|
|---|
| 391 |
|
|---|
| 392 |
|
|---|
| 393 |
|
|---|
| 394 |
|
|---|
| 395 |
|
|---|
| 396 |
|
|---|
| 397 |
|
|---|
| 398 |
|
|---|
| 399 |
|
|---|
| 400 |
|
|---|
| 401 |
|
|---|
| 402 |
|
|---|
| 403 |
|
|---|
| 404 |
|
|---|
| 405 |
|
|---|
| 406 |
|
|---|
| 407 |
|
|---|
| 408 |
def find(*args) |
|---|
| 409 |
options = extract_options_from_args!(args) |
|---|
| 410 |
validate_find_options(options) |
|---|
| 411 |
set_readonly_option!(options) |
|---|
| 412 |
|
|---|
| 413 |
case args.first |
|---|
| 414 |
when :first then find_initial(options) |
|---|
| 415 |
when :all then find_every(options) |
|---|
| 416 |
else find_from_ids(args, options) |
|---|
| 417 |
end |
|---|
| 418 |
end |
|---|
| 419 |
|
|---|
| 420 |
|
|---|
| 421 |
|
|---|
| 422 |
|
|---|
| 423 |
def find_by_sql(sql) |
|---|
| 424 |
connection.select_all(sanitize_sql(sql), "#{name} Load").collect! { |record| instantiate(record) } |
|---|
| 425 |
end |
|---|
| 426 |
|
|---|
| 427 |
|
|---|
| 428 |
|
|---|
| 429 |
|
|---|
| 430 |
|
|---|
| 431 |
|
|---|
| 432 |
|
|---|
| 433 |
|
|---|
| 434 |
def exists?(id_or_conditions) |
|---|
| 435 |
!find(:first, :conditions => expand_id_conditions(id_or_conditions)).nil? |
|---|
| 436 |
rescue ActiveRecord::ActiveRecordError |
|---|
| 437 |
false |
|---|
| 438 |
end |
|---|
| 439 |
|
|---|
| 440 |
|
|---|
| 441 |
|
|---|
| 442 |
def create(attributes = nil) |
|---|
| 443 |
if attributes.is_a?(Array) |
|---|
| 444 |
attributes.collect { |attr| create(attr) } |
|---|
| 445 |
else |
|---|
| 446 |
object = new(attributes) |
|---|
| 447 |
scope(:create).each { |att,value| object.send("#{att}=", value) } if scoped?(:create) |
|---|
| 448 |
object.save |
|---|
| 449 |
object |
|---|
| 450 |
end |
|---|
| 451 |
end |
|---|
| 452 |
|
|---|
| 453 |
|
|---|
| 454 |
|
|---|
| 455 |
|
|---|
| 456 |
|
|---|
| 457 |
|
|---|
| 458 |
|
|---|
| 459 |
|
|---|
| 460 |
|
|---|
| 461 |
|
|---|
| 462 |
|
|---|
| 463 |
|
|---|
| 464 |
|
|---|
| 465 |
def update(id, attributes) |
|---|
| 466 |
if id.is_a?(Array) |
|---|
| 467 |
idx = -1 |
|---|
| 468 |
id.collect { |id| idx += 1; update(id, attributes[idx]) } |
|---|
| 469 |
else |
|---|
| 470 |
object = find(id) |
|---|
| 471 |
object.update_attributes(attributes) |
|---|
| 472 |
object |
|---|
| 473 |
end |
|---|
| 474 |
end |
|---|
| 475 |
|
|---|
| 476 |
|
|---|
| 477 |
|
|---|
| 478 |
def delete(id) |
|---|
| 479 |
delete_all([ "#{primary_key} IN (?)", id ]) |
|---|
| 480 |
end |
|---|
| 481 |
|
|---|
| 482 |
|
|---|
| 483 |
|
|---|
| 484 |
def destroy(id) |
|---|
| 485 |
id.is_a?(Array) ? id.each { |id| destroy(id) } : find(id).destroy |
|---|
| 486 |
end |
|---|
| 487 |
|
|---|
| 488 |
|
|---|
| 489 |
|
|---|
| 490 |
|
|---|
| 491 |
def update_all(updates, conditions = nil) |
|---|
| 492 |
sql = "UPDATE #{table_name} SET #{sanitize_sql(updates)} " |
|---|
| 493 |
add_conditions!(sql, conditions, scope(:find)) |
|---|
| 494 |
connection.update(sql, "#{name} Update") |
|---|
| 495 |
end |
|---|
| 496 |
|
|---|
| 497 |
|
|---|
| 498 |
|
|---|
| 499 |
|
|---|
| 500 |
def destroy_all(conditions = nil) |
|---|
| 501 |
find(:all, :conditions => conditions).each { |object| object.destroy } |
|---|
| 502 |
end |
|---|
| 503 |
|
|---|
| 504 |
|
|---|
| 505 |
|
|---|
| 506 |
|
|---|
| 507 |
def delete_all(conditions = nil) |
|---|
| 508 |
sql = "DELETE FROM #{table_name} " |
|---|
| 509 |
add_conditions!(sql, conditions, scope(:find)) |
|---|
| 510 |
connection.delete(sql, "#{name} Delete all") |
|---|
| 511 |
end |
|---|
| 512 |
|
|---|
| 513 |
|
|---|
| 514 |
|
|---|
| 515 |
def count_by_sql(sql) |
|---|
| 516 |
sql = sanitize_conditions(sql) |
|---|
| 517 |
connection.select_value(sql, "#{name} Count").to_i |
|---|
| 518 |
end |
|---|
| 519 |
|
|---|
| 520 |
|
|---|
| 521 |
|
|---|
| 522 |
|
|---|
| 523 |
|
|---|
| 524 |
|
|---|
| 525 |
def increment_counter(counter_name, id) |
|---|
| 526 |
update_all "#{counter_name} = #{counter_name} + 1", "#{primary_key} = #{quote_value(id)}" |
|---|
| 527 |
end |
|---|
| 528 |
|
|---|
| 529 |
|
|---|
| 530 |
def decrement_counter(counter_name, id) |
|---|
| 531 |
update_all "#{counter_name} = #{counter_name} - 1", "#{primary_key} = #{quote_value(id)}" |
|---|
| 532 |
end |
|---|
| 533 |
|
|---|
| 534 |
|
|---|
| 535 |
|
|---|
| 536 |
|
|---|
| 537 |
|
|---|
| 538 |
|
|---|
| 539 |
|
|---|
| 540 |
|
|---|
| 541 |
|
|---|
| 542 |
|
|---|
| 543 |
|
|---|
| 544 |
|
|---|
| 545 |
|
|---|
| 546 |
|
|---|
| 547 |
|
|---|
| 548 |
|
|---|
| 549 |
|
|---|
| 550 |
def attr_protected(*attributes) |
|---|
| 551 |
write_inheritable_array("attr_protected", attributes - (protected_attributes || [])) |
|---|
| 552 |
end |
|---|
| 553 |
|
|---|
| 554 |
|
|---|
| 555 |
def protected_attributes |
|---|
| 556 |
read_inheritable_attribute("attr_protected") |
|---|
| 557 |
end |
|---|
| 558 |
|
|---|
| 559 |
|
|---|
| 560 |
|
|---|
| 561 |
|
|---|
| 562 |
|
|---|
| 563 |
def attr_accessible(*attributes) |
|---|
| 564 |
write_inheritable_array("attr_accessible", attributes - (accessible_attributes || [])) |
|---|
| 565 |
end |
|---|
| 566 |
|
|---|
| 567 |
|
|---|
| 568 |
def accessible_attributes |
|---|
| 569 |
read_inheritable_attribute("attr_accessible") |
|---|
| 570 |
end |
|---|
| 571 |
|
|---|
| 572 |
|
|---|
| 573 |
|
|---|
| 574 |
|
|---|
| 575 |
|
|---|
| 576 |
def serialize(attr_name, class_name = Object) |
|---|
| 577 |
serialized_attributes[attr_name.to_s] = class_name |
|---|
| 578 |
end |
|---|
| 579 |
|
|---|
| 580 |
|
|---|
| 581 |
def serialized_attributes |
|---|
| 582 |
read_inheritable_attribute("attr_serialized") or write_inheritable_attribute("attr_serialized", {}) |
|---|
| 583 |
end |
|---|
| 584 |
|
|---|
| 585 |
|
|---|
| 586 |
|
|---|
| 587 |
|
|---|
| 588 |
|
|---|
| 589 |
|
|---|
| 590 |
|
|---|
| 591 |
|
|---|
| 592 |
|
|---|
| 593 |
|
|---|
| 594 |
|
|---|
| 595 |
|
|---|
| 596 |
|
|---|
| 597 |
|
|---|
| 598 |
|
|---|
| 599 |
|
|---|
| 600 |
|
|---|
| 601 |
|
|---|
| 602 |
|
|---|
| 603 |
|
|---|
| 604 |
|
|---|
| 605 |
|
|---|
| 606 |
|
|---|
| 607 |
|
|---|
| 608 |
def table_name |
|---|
| 609 |
reset_table_name |
|---|
| 610 |
end |
|---|
| 611 |
|
|---|
| 612 |
def reset_table_name |
|---|
| 613 |
base = base_class |
|---|
| 614 |
|
|---|
| 615 |
name = |
|---|
| 616 |
|
|---|
| 617 |
unless self == base |
|---|
| 618 |
base.table_name |
|---|
| 619 |
else |
|---|
| 620 |
|
|---|
| 621 |
if parent < ActiveRecord::Base && !parent.abstract_class? |
|---|
| 622 |
contained = parent.table_name |
|---|
| 623 |
contained = contained.singularize if parent.pluralize_table_names |
|---|
| 624 |
contained << '_' |
|---|
| 625 |
end |
|---|
| 626 |
name = "#{table_name_prefix}#{contained}#{undecorated_table_name(base.name)}#{table_name_suffix}" |
|---|
| 627 |
end |
|---|
| 628 |
|
|---|
| 629 |
set_table_name(name) |
|---|
| 630 |
name |
|---|
| 631 |
end |
|---|
| 632 |
|
|---|
| 633 |
|
|---|
| 634 |
|
|---|
| 635 |
def primary_key |
|---|
| 636 |
reset_primary_key |
|---|
| 637 |
end |
|---|
| 638 |
|
|---|
| 639 |
def reset_primary_key |
|---|
| 640 |
key = 'id' |
|---|
| 641 |
case primary_key_prefix_type |
|---|
| 642 |
when :table_name |
|---|
| 643 |
key = Inflector.foreign_key(base_class.name, false) |
|---|
| 644 |
when :table_name_with_underscore |
|---|
| 645 |
key = Inflector.foreign_key(base_class.name) |
|---|
| 646 |
end |
|---|
| 647 |
set_primary_key(key) |
|---|
| 648 |
key |
|---|
| 649 |
end |
|---|
| 650 |
|
|---|
| 651 |
|
|---|
| 652 |
|
|---|
| 653 |
def inheritance_column |
|---|
| 654 |
@inheritance_column ||= "type".freeze |
|---|
| 655 |
end |
|---|
| 656 |
|
|---|
| 657 |
|
|---|
| 658 |
|
|---|
| 659 |
def sequence_name |
|---|
| 660 |
reset_sequence_name |
|---|
| 661 |
end |
|---|
| 662 |
|
|---|
| 663 |
def reset_sequence_name |
|---|
| 664 |
default = connection.default_sequence_name(table_name, primary_key) |
|---|
| 665 |
set_sequence_name(default) |
|---|
| 666 |
default |
|---|
| 667 |
end |
|---|
| 668 |
|
|---|
| 669 |
|
|---|
| 670 |
|
|---|
| 671 |
|
|---|
| 672 |
|
|---|
| 673 |
|
|---|
| 674 |
|
|---|
| 675 |
|
|---|
| 676 |
|
|---|
| 677 |
def set_table_name(value = nil, &block) |
|---|
| 678 |
define_attr_method :table_name, value, &block |
|---|
| 679 |
end |
|---|
| 680 |
alias :table_name= :set_table_name |
|---|
| 681 |
|
|---|
| 682 |
|
|---|
| 683 |
|
|---|
| 684 |
|
|---|
| 685 |
|
|---|
| 686 |
|
|---|
| 687 |
|
|---|
| 688 |
|
|---|
| 689 |
|
|---|
| 690 |
|
|---|
| 691 |
def set_primary_key(value = nil, &block) |
|---|
| 692 |
define_attr_method :primary_key, value, &block |
|---|
| 693 |
end |
|---|
| 694 |
alias :primary_key= :set_primary_key |
|---|
| 695 |
|
|---|
| 696 |
|
|---|
| 697 |
|
|---|
| 698 |
|
|---|
| 699 |
|
|---|
| 700 |
|
|---|
| 701 |
|
|---|
| 702 |
|
|---|
| 703 |
|
|---|
| 704 |
|
|---|
| 705 |
|
|---|
| 706 |
|
|---|
| 707 |
def set_inheritance_column(value = nil, &block) |
|---|
| 708 |
define_attr_method :inheritance_column, value, &block |
|---|
| 709 |
end |
|---|
| 710 |
alias :inheritance_column= :set_inheritance_column |
|---|
| 711 |
|
|---|
| 712 |
|
|---|
| 713 |
|
|---|
| 714 |
|
|---|
| 715 |
|
|---|
| 716 |
|
|---|
| 717 |
|
|---|
| 718 |
|
|---|
| 719 |
|
|---|
| 720 |
|
|---|
| 721 |
|
|---|
| 722 |
|
|---|
| 723 |
|
|---|
| 724 |
|
|---|
| 725 |
|
|---|
| 726 |
|
|---|
| 727 |
|
|---|
| 728 |
def set_sequence_name(value = nil, &block) |
|---|
| 729 |
define_attr_method :sequence_name, value, &block |
|---|
| 730 |
end |
|---|
| 731 |
alias :sequence_name= :set_sequence_name |
|---|
| 732 |
|
|---|
| 733 |
|
|---|
| 734 |
def class_name(table_name = table_name) |
|---|
| 735 |
|
|---|
| 736 |
class_name = table_name[table_name_prefix.length..-(table_name_suffix.length + 1)].camelize |
|---|
| 737 |
class_name = class_name.singularize if pluralize_table_names |
|---|
| 738 |
class_name |
|---|
| 739 |
end |
|---|
| 740 |
|
|---|
| 741 |
|
|---|
| 742 |
def table_exists? |
|---|
| 743 |
if connection.respond_to?(:tables) |
|---|
| 744 |
connection.tables.include? table_name |
|---|
| 745 |
else |
|---|
| 746 |
|
|---|
| 747 |
|
|---|
| 748 |
begin |
|---|
| 749 |
reset_column_information |
|---|
| 750 |
columns.size > 0 |
|---|
| 751 |
rescue ActiveRecord::StatementInvalid |
|---|
| 752 |
false |
|---|
| 753 |
end |
|---|
| 754 |
end |
|---|
| 755 |
end |
|---|
| 756 |
|
|---|
| 757 |
|
|---|
| 758 |
def columns |
|---|
| 759 |
unless @columns |
|---|
| 760 |
@columns = connection.columns(table_name, "#{name} Columns") |
|---|
| 761 |
@columns.each {|column| column.primary = column.name == primary_key} |
|---|
| 762 |
end |
|---|
| 763 |
@columns |
|---|
| 764 |
end |
|---|
| 765 |
|
|---|
| 766 |
|
|---|
| 767 |
def columns_hash |
|---|
| 768 |
@columns_hash ||= columns.inject({}) { |hash, column| hash[column.name] = column; hash } |
|---|
| 769 |
end |
|---|
| 770 |
|
|---|
| 771 |
|
|---|
| 772 |
def column_names |
|---|
| 773 |
@column_names ||= columns.map { |column| column.name } |
|---|
| 774 |
end |
|---|
| 775 |
|
|---|
| 776 |
|
|---|
| 777 |
|
|---|
| 778 |
def content_columns |
|---|
| 779 |
@content_columns ||= columns.reject { |c| c.primary || c.name =~ /(_id|_count)$/ || c.name == inheritance_column } |
|---|
| 780 |
end |
|---|
| 781 |
|
|---|
| 782 |
|
|---|
| 783 |
|
|---|
| 784 |
|
|---|
| 785 |
def column_methods_hash |
|---|
| 786 |
@dynamic_methods_hash ||= column_names.inject(Hash.new(false)) do |methods, attr| |
|---|
| 787 |
attr_name = attr.to_s |
|---|
| 788 |
methods[attr.to_sym] = attr_name |
|---|
| 789 |
methods["#{attr}=".to_sym] = attr_name |
|---|
| 790 |
methods["#{attr}?".to_sym] = attr_name |
|---|
| 791 |
methods["#{attr}_before_type_cast".to_sym] = attr_name |
|---|
| 792 |
methods |
|---|
| 793 |
end |
|---|
| 794 |
end |
|---|
| 795 |
|
|---|
| 796 |
|
|---|
| 797 |
def read_methods |
|---|
| 798 |
@read_methods ||= Set.new |
|---|
| 799 |
end |
|---|
| 800 |
|
|---|
| 801 |
|
|---|
| 802 |
def reset_column_information |
|---|
| 803 |
read_methods.each { |name| undef_method(name) } |
|---|
| 804 |
@column_names = @columns = @columns_hash = @content_columns = @dynamic_methods_hash = @read_methods = @inheritance_column = nil |
|---|
| 805 |
end |
|---|
| 806 |
|
|---|
| 807 |
def reset_column_information_and_inheritable_attributes_for_all_subclasses |
|---|
| 808 |
subclasses.each { |klass| klass.reset_inheritable_attributes; klass.reset_column_information } |
|---|
| 809 |
end |
|---|
| 810 |
|
|---|
| 811 |
|
|---|
| 812 |
|
|---|
| 813 |
|
|---|
| 814 |
def human_attribute_name(attribute_key_name) |
|---|
| 815 |
attribute_key_name.humanize |
|---|
| 816 |
end |
|---|
| 817 |
|
|---|
| 818 |
def descends_from_active_record? |
|---|
| 819 |
superclass == Base || !columns_hash.include?(inheritance_column) |
|---|
| 820 |
end |
|---|
| 821 |
|
|---|
| 822 |
|
|---|
| 823 |
def quote_value(value, column = nil) |
|---|
| 824 |
connection.quote(value,column) |
|---|
| 825 |
end |
|---|
| 826 |
|
|---|
| 827 |
def quote(value, column = nil) |
|---|
| 828 |
connection.quote(value, column) |
|---|
| 829 |
end |
|---|
| 830 |
deprecate :quote => :quote_value |
|---|
| 831 |
|
|---|
| 832 |
|
|---|
| 833 |
def sanitize(object) |
|---|
| 834 |
connection.quote(object) |
|---|
| 835 |
end |
|---|
| 836 |
|
|---|
| 837 |
|
|---|
| 838 |
|
|---|
| 839 |
|
|---|
| 840 |
|
|---|
| 841 |
|
|---|
| 842 |
|
|---|
| 843 |
|
|---|
| 844 |
|
|---|
| 845 |
|
|---|
| 846 |
|
|---|
| 847 |
|
|---|
| 848 |
|
|---|
| 849 |
|
|---|
| 850 |
def benchmark(title, log_level = Logger::DEBUG, use_silence = true) |
|---|
| 851 |
if logger && logger.level == log_level |
|---|
| 852 |
result = nil |
|---|
| 853 |
seconds = Benchmark.realtime { result = use_silence ? silence { yield } : yield } |
|---|
| 854 |
logger.add(log_level, "#{title} (#{'%.5f' % seconds})") |
|---|
| 855 |
result |
|---|
| 856 |
else |
|---|
| 857 |
yield |
|---|
| 858 |
end |
|---|
| 859 |
end |
|---|
| 860 |
|
|---|
| 861 |
|
|---|
| 862 |
def silence |
|---|
| 863 |
old_logger_level, logger.level = logger.level, Logger::ERROR if logger |
|---|
| 864 |
yield |
|---|
| 865 |
ensure |
|---|
| 866 |
logger.level = old_logger_level if logger |
|---|
| 867 |
end |
|---|
| 868 |
|
|---|
| 869 |
|
|---|
| 870 |
|
|---|
| 871 |
|
|---|
| 872 |
|
|---|
| 873 |
|
|---|
| 874 |
|
|---|
| 875 |
|
|---|
| 876 |
|
|---|
| 877 |
|
|---|
| 878 |
|
|---|
| 879 |
|
|---|
| 880 |
|
|---|
| 881 |
|
|---|
| 882 |
|
|---|
| 883 |
|
|---|
| 884 |
|
|---|
| 885 |
|
|---|
| 886 |
|
|---|
| 887 |
|
|---|
| 888 |
|
|---|
| 889 |
|
|---|
| 890 |
|
|---|
| 891 |
|
|---|
| 892 |
|
|---|
| 893 |
|
|---|
| 894 |
|
|---|
| 895 |
|
|---|
| 896 |
|
|---|
| 897 |
|
|---|
| 898 |
def with_scope(method_scoping = {}, action = :merge, &block) |
|---|
| 899 |
method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping) |
|---|
| 900 |
|
|---|
| 901 |
|
|---|
| 902 |
method_scoping = method_scoping.inject({}) do |hash, (method, params)| |
|---|
| 903 |
hash[method] = (params == true) ? params : params.dup |
|---|
| 904 |
hash |
|---|
| 905 |
end |
|---|
| 906 |
|
|---|
| 907 |
method_scoping.assert_valid_keys([ :find, :create ]) |
|---|
| 908 |
|
|---|
| 909 |
if f = method_scoping[:find] |
|---|
| 910 |
f.assert_valid_keys([ :conditions, :joins, :select, :include, :from, :offset, :limit, :order, :readonly, :lock ]) |
|---|
| 911 |
f[:readonly] = true if !f[:joins].blank? && !f.has_key?(:readonly) |
|---|
| 912 |
end |
|---|
| 913 |
|
|---|
| 914 |
|
|---|
| 915 |
if action == :merge && current_scoped_methods |
|---|
| 916 |
method_scoping = current_scoped_methods.inject(method_scoping) do |hash, (method, params)| |
|---|
| 917 |
case hash[method] |
|---|
| 918 |
when Hash |
|---|
| 919 |
if method == :find |
|---|
| 920 |
(hash[method].keys + params.keys).uniq.each do |key| |
|---|
| 921 |
merge = hash[method][key] && params[key] |
|---|
| 922 |
if key == :conditions && merge |
|---|
| 923 |
hash[method][key] = [params[key], hash[method][key]].collect{ |sql| "( %s )" % sanitize_sql(sql) }.join(" AND ") |
|---|
| 924 |
elsif key == :include && merge |
|---|
| 925 |
hash[method][key] = merge_includes(hash[method][key], params[key]).uniq |
|---|
| 926 |
else |
|---|
| 927 |
hash[method][key] = hash[method][key] || params[key] |
|---|
| 928 |
end |
|---|
| 929 |
end |
|---|
| 930 |
else |
|---|
| 931 |
hash[method] = params.merge(hash[method]) |
|---|
| 932 |
end |
|---|
| 933 |
else |
|---|
| 934 |
hash[method] = params |
|---|
| 935 |
end |
|---|
| 936 |
hash |
|---|
| 937 |
end |
|---|
| 938 |
end |
|---|
| 939 |
|
|---|
| 940 |
self.scoped_methods << method_scoping |
|---|
| 941 |
|
|---|
| 942 |
begin |
|---|
| 943 |
yield |
|---|
| 944 |
ensure |
|---|
| 945 |
self.scoped_methods.pop |
|---|
| 946 |
end |
|---|
| 947 |
end |
|---|
| 948 |
|
|---|
| 949 |
|
|---|
| 950 |
def with_exclusive_scope(method_scoping = {}, &block) |
|---|
| 951 |
with_scope(method_scoping, :overwrite, &block) |
|---|
| 952 |
end |
|---|
| 953 |
|
|---|
| 954 |
|
|---|
| 955 |
def ===(object) |
|---|
| 956 |
object.is_a?(self) |
|---|
| 957 |
end |
|---|
| 958 |
|
|---|
| 959 |
|
|---|
| 960 |
def threaded_connections |
|---|
| 961 |
allow_concurrency |
|---|
| 962 |
end |
|---|
| 963 |
|
|---|
| 964 |
|
|---|
| 965 |
def threaded_connections=(value) |
|---|
| 966 |
self.allow_concurrency = value |
|---|
| 967 |
end |
|---|
| 968 |
|
|---|
| 969 |
|
|---|
| 970 |
|
|---|
| 971 |
|
|---|
| 972 |
def base_class |
|---|
| 973 |
class_of_active_record_descendant(self) |
|---|
| 974 |
end |
|---|
| 975 |
|
|---|
| 976 |
|
|---|
| 977 |
attr_accessor :abstract_class |
|---|
| 978 |
|
|---|
| 979 |
|
|---|
| 980 |
|
|---|
| 981 |
def abstract_class? |
|---|
| 982 |
abstract_class == true |
|---|
| 983 |
end |
|---|
| 984 |
|
|---|
| 985 |
private |
|---|
| 986 |
def find_initial(options) |
|---|
| 987 |
options.update(:limit => 1) unless options[:include] |
|---|
| 988 |
find_every(options).first |
|---|
| 989 |
end |
|---|
| 990 |
|
|---|
| 991 |
def find_every(options) |
|---|
| 992 |
records = scoped?(:find, :include) || options[:include] ? |
|---|
| 993 |
find_with_associations(options) : |
|---|
| 994 |
find_by_sql(construct_finder_sql(options)) |
|---|
| 995 |
|
|---|
| 996 |
records.each { |record| record.readonly! } if options[:readonly] |
|---|
| 997 |
|
|---|
| 998 |
records |
|---|
| 999 |
end |
|---|
| 1000 |
|
|---|
| 1001 |
def find_from_ids(ids, options) |
|---|
| 1002 |
expects_array = ids.first.kind_of?(Array) |
|---|
| 1003 |
return ids.first if expects_array && ids.first.empty? |
|---|
| 1004 |
|
|---|
| 1005 |
ids = ids.flatten.compact.uniq |
|---|
| 1006 |
|
|---|
| 1007 |
case ids.size |
|---|
| 1008 |
when 0 |
|---|
| 1009 |
raise RecordNotFound, "Couldn't find #{name} without an ID" |
|---|
| 1010 |
when 1 |
|---|
| 1011 |
result = find_one(ids.first, options) |
|---|
| 1012 |
expects_array ? [ result ] : result |
|---|
| 1013 |
else |
|---|
| 1014 |
find_some(ids, options) |
|---|
| 1015 |
end |
|---|
| 1016 |
end |
|---|
| 1017 |
|
|---|
| 1018 |
def find_one(id, options) |
|---|
| 1019 |
conditions = " AND (#{sanitize_sql(options[:conditions])})" if options[:conditions] |
|---|
| 1020 |
options.update :conditions => "#{table_name}.#{primary_key} = #{quote_value(id,columns_hash[primary_key])}#{conditions}" |
|---|
| 1021 |
|
|---|
| 1022 |
|
|---|
| 1023 |
|
|---|
| 1024 |
|
|---|
| 1025 |
if result = find_every(options).first |
|---|
| 1026 |
result |
|---|
| 1027 |
else |
|---|
| 1028 |
raise RecordNotFound, "Couldn't find #{name} with ID=#{id}#{conditions}" |
|---|
| 1029 |
end |
|---|
| 1030 |
end |
|---|
| 1031 |
|
|---|
| 1032 |
def find_some(ids, options) |
|---|
| 1033 |
conditions = " AND (#{sanitize_sql(options[:conditions])})" if options[:conditions] |
|---|
| 1034 |
ids_list = ids.map { |id| quote_value(id,columns_hash[primary_key]) }.join(',') |
|---|
| 1035 |
options.update :conditions => "#{table_name}.#{primary_key} IN (#{ids_list})#{conditions}" |
|---|
| 1036 |
|
|---|
| 1037 |
result = find_every(options) |
|---|
| 1038 |
|
|---|
| 1039 |
if result.size == ids.size |
|---|
| 1040 |
result |
|---|
| 1041 |
else |
|---|
| 1042 |
raise RecordNotFound, "Couldn't find all #{name.pluralize} with IDs (#{ids_list})#{conditions}" |
|---|
| 1043 |
end |
|---|
| 1044 |
end |
|---|
| 1045 |
|
|---|
| 1046 |
|
|---|
| 1047 |
|
|---|
| 1048 |
|
|---|
| 1049 |
def instantiate(record) |
|---|
| 1050 |
object = |
|---|
| 1051 |
if subclass_name = record[inheritance_column] |
|---|
| 1052 |
|
|---|
| 1053 |
if subclass_name.empty? |
|---|
| 1054 |
allocate |
|---|
| 1055 |
|
|---|
| 1056 |
else |
|---|
| 1057 |
require_association_class(subclass_name) |
|---|
| 1058 |
|
|---|
| 1059 |
|
|---|
| 1060 |
|
|---|
| 1061 |
unless columns_hash.include?(inheritance_column) |
|---|
| 1062 |
allocate |
|---|
| 1063 |
|
|---|
| 1064 |
else |
|---|
| 1065 |
begin |
|---|
| 1066 |
compute_type(subclass_name).allocate |
|---|
| 1067 |
rescue NameError |
|---|
| 1068 |
raise SubclassNotFound, |
|---|
| 1069 |
"The single-table inheritance mechanism failed to locate the subclass: '#{record[inheritance_column]}'. " + |
|---|
| 1070 |
"This error is raised because the column '#{inheritance_column}' is reserved for storing the class in case of inheritance. " + |
|---|
| 1071 |
"Please rename this column if you didn't intend it to be used for storing the inheritance class " + |
|---|
| 1072 |
"or overwrite #{self.to_s}.inheritance_column to use another column for that information." |
|---|
| 1073 |
end |
|---|
| 1074 |
end |
|---|
| 1075 |
end |
|---|
| 1076 |
else |
|---|
| 1077 |
allocate |
|---|
| 1078 |
end |
|---|
| 1079 |
|
|---|
| 1080 |
object.instance_variable_set("@attributes", record) |
|---|
| 1081 |
object |
|---|
| 1082 |
end |
|---|
| 1083 |
|
|---|
| 1084 |
|
|---|
| 1085 |
|
|---|
| 1086 |
def type_name_with_module(type_name) |
|---|
| 1087 |
(/^::/ =~ type_name) ? type_name : "#{parent.name}::#{type_name}" |
|---|
| 1088 |
end |
|---|
| 1089 |
|
|---|
| 1090 |
def construct_finder_sql(options) |
|---|
| 1091 |
scope = scope(:find) |
|---|
| 1092 |
sql = "SELECT #{(scope && scope[:select]) || options[:select] || '*'} " |
|---|
| 1093 |
sql << "FROM #{(scope && scope[:from]) || options[:from] || table_name} " |
|---|
| 1094 |
|
|---|
| 1095 |
add_joins!(sql, options, scope) |
|---|
| 1096 |
add_conditions!(sql, options[:conditions], scope) |
|---|
| 1097 |
|
|---|
| 1098 |
sql << " GROUP BY #{options[:group]} " if options[:group] |
|---|
| 1099 |
|
|---|
| 1100 |
add_order!(sql, options[:order], scope) |
|---|
| 1101 |
add_limit!(sql, options, scope) |
|---|
| 1102 |
add_lock!(sql, options, scope) |
|---|
| 1103 |
|
|---|
| 1104 |
sql |
|---|
| 1105 |
end |
|---|
| 1106 |
|
|---|
| 1107 |
|
|---|
| 1108 |
def merge_includes(first, second) |
|---|
| 1109 |
(safe_to_array(first) + safe_to_array(second)).uniq |
|---|
| 1110 |
end |
|---|
| 1111 |
|
|---|
| 1112 |
|
|---|
| 1113 |
def safe_to_array(o) |
|---|
| 1114 |
case o |
|---|
| 1115 |
when NilClass |
|---|
| 1116 |
[] |
|---|
| 1117 |
when Array |
|---|
| 1118 |
o |
|---|
| 1119 |
else |
|---|
| 1120 |
[o] |
|---|
| 1121 |
end |
|---|
| 1122 |
end |
|---|
| 1123 |
|
|---|
| 1124 |
def add_order!(sql, order, scope = :auto) |
|---|
| 1125 |
scope = scope(:find) if :auto == scope |
|---|
| 1126 |
scoped_order = scope[:order] if scope |
|---|
| 1127 |
if order |
|---|
| 1128 |
sql << " ORDER BY #{order}" |
|---|
| 1129 |
sql << ", #{scoped_order}" if scoped_order |
|---|
| 1130 |
else |
|---|
| 1131 |
sql << " ORDER BY #{scoped_order}" if scoped_order |
|---|
| 1132 |
end |
|---|
| 1133 |
end |
|---|
| 1134 |
|
|---|
| 1135 |
|
|---|
| 1136 |
def add_limit!(sql, options, scope = :auto) |
|---|
| 1137 |
scope = scope(:find) if :auto == scope |
|---|
| 1138 |
options = options.reverse_merge(:limit => scope[:limit], :offset => scope[:offset]) if scope |
|---|
| 1139 |
connection.add_limit_offset!(sql, options) |
|---|
| 1140 |
end |
|---|
| 1141 |
|
|---|
| 1142 |
|
|---|
| 1143 |
|
|---|
| 1144 |
def add_lock!(sql, options, scope = :auto) |
|---|
| 1145 |
scope = scope(:find) if :auto == :scope |
|---|
| 1146 |
options = options.reverse_merge(:lock => scope[:lock]) if scope |
|---|
| 1147 |
connection.add_lock!(sql, options) |
|---|
| 1148 |
end |
|---|
| 1149 |
|
|---|
| 1150 |
|
|---|
| 1151 |
def add_joins!(sql, options, scope = :auto) |
|---|
| 1152 |
scope = scope(:find) if :auto == scope |
|---|
| 1153 |
join = (scope && scope[:joins]) || options[:joins] |
|---|
| 1154 |
sql << " #{join} " if join |
|---|
| 1155 |
end |
|---|
| 1156 |
|
|---|
| 1157 |
|
|---|
| 1158 |
|
|---|
| 1159 |
def add_conditions!(sql, conditions, scope = :auto) |
|---|
| 1160 |
scope = scope(:find) if :auto == scope |
|---|
| 1161 |
segments = [] |
|---|
| 1162 |
segments << sanitize_sql(scope[:conditions]) if scope && scope[:conditions] |
|---|
| 1163 |
segments << sanitize_sql(conditions) unless conditions.nil? |
|---|
| 1164 |
segments << type_condition unless descends_from_active_record? |
|---|
| 1165 |
segments.compact! |
|---|
| 1166 |
sql << "WHERE (#{segments.join(") AND (")}) " unless segments.empty? |
|---|
| 1167 |
end |
|---|
| 1168 |
|
|---|
| 1169 |
def type_condition |
|---|
| 1170 |
quoted_inheritance_column = connection.quote_column_name(inheritance_column) |
|---|
| 1171 |
type_condition = subclasses.inject("#{table_name}.#{quoted_inheritance_column} = '#{name.demodulize}' ") do |condition, subclass| |
|---|
| 1172 |
condition << "OR #{table_name}.#{quoted_inheritance_column} = '#{subclass.name.demodulize}' " |
|---|
| 1173 |
end |
|---|
| 1174 |
|
|---|
| 1175 |
" (#{type_condition}) " |
|---|
| 1176 |
end |
|---|
| 1177 |
|
|---|
| 1178 |
|
|---|
| 1179 |
def undecorated_table_name(class_name = base_class.name) |
|---|
| 1180 |
table_name = Inflector.underscore(Inflector.demodulize(class_name)) |
|---|
| 1181 |
table_name = Inflector.pluralize(table_name) if pluralize_table_names |
|---|
| 1182 |
table_name |
|---|
| 1183 |
end |
|---|
| 1184 |
|
|---|
| 1185 |
|
|---|
| 1186 |
|
|---|
| 1187 |
|
|---|
| 1188 |
|
|---|
| 1189 |
|
|---|
| 1190 |
|
|---|
| 1191 |
def method_missing(method_id, *arguments) |
|---|
| 1192 |
if match = /find_(all_by|by)_([_a-zA-Z]\w*)/.match(method_id.to_s) |
|---|
| 1193 |
finder, deprecated_finder = determine_finder(match), determine_deprecated_finder(match) |
|---|
| 1194 |
|
|---|
| 1195 |
attribute_names = extract_attribute_names_from_match(match) |
|---|
| 1196 |
super unless all_attributes_exists?(attribute_names) |
|---|
| 1197 |
|
|---|
| 1198 |
attributes = construct_attributes_from_arguments(attribute_names, arguments) |
|---|
| 1199 |
|
|---|
| 1200 |
case extra_options = arguments[attribute_names.size] |
|---|
| 1201 |
when nil |
|---|
| 1202 |
options = { :conditions => attributes } |
|---|
| 1203 |
set_readonly_option!(options) |
|---|
| 1204 |
ActiveSupport::Deprecation.silence { send(finder, options) } |
|---|
| 1205 |
|
|---|
| 1206 |
when Hash |
|---|
| 1207 |
finder_options = extra_options.merge(:conditions => attributes) |
|---|
| 1208 |
validate_find_options(finder_options) |
|---|
| 1209 |
set_readonly_option!(finder_options) |
|---|
| 1210 |
|
|---|
| 1211 |
if extra_options[:conditions] |
|---|
| 1212 |
with_scope(:find => { :conditions => extra_options[:conditions] }) do |
|---|
| 1213 |
ActiveSupport::Deprecation.silence { send(finder, finder_options) } |
|---|
| 1214 |
end |
|---|
| 1215 |
else |
|---|
| 1216 |
ActiveSupport::Deprecation.silence { send(finder, finder_options) } |
|---|
| 1217 |
end |
|---|
| 1218 |
|
|---|
| 1219 |
else |
|---|
| 1220 |
ActiveSupport::Deprecation.silence do |
|---|
| 1221 |
send(deprecated_finder, sanitize_sql(attributes), *arguments[attribute_names.length..-1]) |
|---|
| 1222 |
end |
|---|
| 1223 |
end |
|---|
| 1224 |
elsif match = /find_or_(initialize|create)_by_([_a-zA-Z]\w*)/.match(method_id.to_s) |
|---|
| 1225 |
instantiator = determine_instantiator(match) |
|---|
| 1226 |
attribute_names = extract_attribute_names_from_match(match) |
|---|
| 1227 |
super unless all_attributes_exists?(attribute_names) |
|---|
| 1228 |
|
|---|
| 1229 |
attributes = construct_attributes_from_arguments(attribute_names, arguments) |
|---|
| 1230 |
options = { :conditions => attributes } |
|---|
| 1231 |
set_readonly_option!(options) |
|---|
| 1232 |
|
|---|
| 1233 |
find_initial(options) || send(instantiator, attributes) |
|---|
| 1234 |
else |
|---|
| 1235 |
super |
|---|
| 1236 |
end |
|---|
| 1237 |
end |
|---|
| 1238 |
|
|---|
| 1239 |
def determine_finder(match) |
|---|
| 1240 |
match.captures.first == 'all_by' ? :find_every : :find_initial |
|---|
| 1241 |
end |
|---|
| 1242 |
|
|---|
| 1243 |
def determine_deprecated_finder(match) |
|---|
| 1244 |
match.captures.first == 'all_by' ? :find_all : :find_first |
|---|
| 1245 |
end |
|---|
| 1246 |
|
|---|
| 1247 |
def determine_instantiator(match) |
|---|
| 1248 |
match.captures.first == 'initialize' ? :new : :create |
|---|
| 1249 |
end |
|---|
| 1250 |
|
|---|
| 1251 |
def extract_attribute_names_from_match(match) |
|---|
| 1252 |
match.captures.last.split('_and_') |
|---|
| 1253 |
end |
|---|
| 1254 |
|
|---|
| 1255 |
def construct_attributes_from_arguments(attribute_names, arguments) |
|---|
| 1256 |
attributes = {} |
|---|
| 1257 |
attribute_names.each_with_index { |name, idx| attributes[name] = arguments[idx] } |
|---|
| 1258 |
attributes |
|---|
| 1259 |
end |
|---|
| 1260 |
|
|---|
| 1261 |
def all_attributes_exists?(attribute_names) |
|---|
| 1262 |
attribute_names.all? { |name| column_methods_hash.include?(name.to_sym) } |
|---|
| 1263 |
end |
|---|
| 1264 |
|
|---|
| 1265 |
def attribute_condition(argument) |
|---|
| 1266 |
case argument |
|---|
| 1267 |
when nil then "IS ?" |
|---|
| 1268 |
when Array then "IN (?)" |
|---|
| 1269 |
else "= ?" |
|---|
| 1270 |
end |
|---|
| 1271 |
end |
|---|
| 1272 |
|
|---|
| 1273 |
|
|---|
| 1274 |
def expand_id_conditions(id_or_conditions) |
|---|
| 1275 |
case id_or_conditions |
|---|
| 1276 |
when Array, Hash then id_or_conditions |
|---|
| 1277 |
else sanitize_sql(primary_key => id_or_conditions) |
|---|
| 1278 |
end |
|---|
| 1279 |
end |
|---|
| 1280 |
|
|---|
| 1281 |
|
|---|
| 1282 |
|
|---|
| 1283 |
|
|---|
| 1284 |
|
|---|
| 1285 |
|
|---|
| 1286 |
|
|---|
| 1287 |
|
|---|
| 1288 |
|
|---|
| 1289 |
|
|---|
| 1290 |
|
|---|
| 1291 |
|
|---|
| 1292 |
|
|---|
| 1293 |
|
|---|
| 1294 |
|
|---|
| 1295 |
|
|---|
| 1296 |
|
|---|
| 1297 |
|
|---|
| 1298 |
|
|---|
| 1299 |
|
|---|
| 1300 |
def define_attr_method(name, value=nil, &block) |
|---|
| 1301 |
sing = class << self; self; end |
|---|
| 1302 |
sing.send :alias_method, "original_#{name}", name |
|---|
| 1303 |
if block_given? |
|---|
| 1304 |
sing.send :define_method, name, &block |
|---|
| 1305 |
else |
|---|
| 1306 |
|
|---|
| 1307 |
|
|---|
| 1308 |
sing.class_eval "def #{name}; #{value.to_s.inspect}; end" |
|---|
| 1309 |
end |
|---|
| 1310 |
end |
|---|
| 1311 |
|
|---|
| 1312 |
protected |
|---|
| 1313 |
def subclasses |
|---|
| 1314 |
@@subclasses[self] ||= [] |
|---|
| 1315 |
@@subclasses[self] + extra = @@subclasses[self].inject([]) {|list, subclass| list + subclass.subclasses } |
|---|
| 1316 |
end |
|---|
| 1317 |
|
|---|
| 1318 |
|
|---|
| 1319 |
def scoped?(method, key = nil) |
|---|
| 1320 |
if current_scoped_methods && (scope = current_scoped_methods[method]) |
|---|
| 1321 |
!key || scope.has_key?(key) |
|---|
| 1322 |
end |
|---|
| 1323 |
end |
|---|
| 1324 |
|
|---|
| 1325 |
|
|---|
| 1326 |
def scope(method, key = nil) |
|---|
| 1327 |
if current_scoped_methods && (scope = current_scoped_methods[method]) |
|---|
| 1328 |
key ? scope[key] : scope |
|---|
| 1329 |
end |
|---|
| 1330 |
end |
|---|
| 1331 |
|
|---|
| 1332 |
def thread_safe_scoped_methods |
|---|
| 1333 |
scoped_methods = (Thread.current[:scoped_methods] ||= {}) |
|---|
| 1334 |
scoped_methods[self] ||= [] |
|---|
| 1335 |
end |
|---|
| 1336 |
|
|---|
| 1337 |
def single_threaded_scoped_methods |
|---|
| 1338 |
@scoped_methods ||= [] |
|---|
| 1339 |
end |
|---|
| 1340 |
|
|---|
| 1341 |
|
|---|
| 1342 |
if @@allow_concurrency |
|---|
| 1343 |
alias_method :scoped_methods, :thread_safe_scoped_methods |
|---|
| 1344 |
else |
|---|
| 1345 |
alias_method :scoped_methods, :single_threaded_scoped_methods |
|---|
| 1346 |
end |
|---|
| 1347 |
|
|---|
| 1348 |
def current_scoped_methods |
|---|
| 1349 |
scoped_methods.last |
|---|
| 1350 |
end |
|---|
| 1351 |
|
|---|
| 1352 |
|
|---|
| 1353 |
|
|---|
| 1354 |
def compute_type(type_name) |
|---|
| 1355 |
modularized_name = type_name_with_module(type_name) |
|---|
| 1356 |
begin |
|---|
| 1357 |
instance_eval(modularized_name) |
|---|
| 1358 |
rescue NameError => e |
|---|
| 1359 |
instance_eval(type_name) |
|---|
| 1360 |
end |
|---|
| 1361 |
end |
|---|
| 1362 |
|
|---|
| 1363 |
|
|---|
| 1364 |
def class_of_active_record_descendant(klass) |
|---|
| 1365 |
if klass.superclass == Base || klass.superclass.abstract_class? |
|---|
| 1366 |
klass |
|---|
| 1367 |
elsif klass.superclass.nil? |
|---|
| 1368 |
raise ActiveRecordError, "#{name} doesn't belong in a hierarchy descending from ActiveRecord" |
|---|
| 1369 |
else |
|---|
| 1370 |
class_of_active_record_descendant(klass.superclass) |
|---|
| 1371 |
end |
|---|
| 1372 |
end |
|---|
| 1373 |
|
|---|
| 1374 |
|
|---|
| 1375 |
def class_name_of_active_record_descendant(klass) |
|---|
| 1376 |
klass.base_class.name |
|---|
| 1377 |
end |
|---|
| 1378 |
|
|---|
| 1379 |
|
|---|
| 1380 |
|
|---|
| 1381 |
|
|---|
| 1382 |
|
|---|
| 1383 |
|
|---|
| 1384 |
def sanitize_sql(condition) |
|---|
| 1385 |
case condition |
|---|
| 1386 |
when Array; sanitize_sql_array(condition) |
|---|
| 1387 |
when Hash; sanitize_sql_hash(condition) |
|---|
| 1388 |
else condition |
|---|
| 1389 |
end |
|---|
| 1390 |
end |
|---|
| 1391 |
|
|---|
| 1392 |
|
|---|
| 1393 |
|
|---|
| 1394 |
|
|---|
| 1395 |
|
|---|
| 1396 |
|
|---|
| 1397 |
def sanitize_sql_hash(attrs) |
|---|
| 1398 |
conditions = attrs.map do |attr, value| |
|---|
| 1399 |
"#{table_name}.#{connection.quote_column_name(attr)} #{attribute_condition(value)}" |
|---|
| 1400 |
end.join(' AND ') |
|---|
| 1401 |
|
|---|
| 1402 |
replace_bind_variables(conditions, attrs.values) |
|---|
| 1403 |
end |
|---|
| 1404 |
|
|---|
| 1405 |
|
|---|
| 1406 |
|
|---|
| 1407 |
|
|---|
| 1408 |
def sanitize_sql_array(ary) |
|---|
| 1409 |
statement, *values = ary |
|---|
| 1410 |
if values.first.is_a?(Hash) and statement =~ /:\w+/ |
|---|
| 1411 |
replace_named_bind_variables(statement, values.first) |
|---|
| 1412 |
elsif statement.include?('?') |
|---|
| 1413 |
replace_bind_variables(statement, values) |
|---|
| 1414 |
else |
|---|
| 1415 |
statement % values.collect { |value| connection.quote_string(value.to_s) } |
|---|
| 1416 |
end |
|---|
| 1417 |
end |
|---|
| 1418 |
|
|---|
| 1419 |
alias_method :sanitize_conditions, :sanitize_sql |
|---|
| 1420 |
|
|---|
| 1421 |
def replace_bind_variables(statement, values) |
|---|
| 1422 |
raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size) |
|---|
| 1423 |
bound = values.dup |
|---|
| 1424 |
statement.gsub('?') { quote_bound_value(bound.shift) } |
|---|
| 1425 |
end |
|---|
| 1426 |
|
|---|
| 1427 |
def replace_named_bind_variables(statement, bind_vars) |
|---|
| 1428 |
statement.gsub(/:(\w+)/) do |
|---|
| 1429 |
match = $1.to_sym |
|---|
| 1430 |
if bind_vars.include?(match) |
|---|
| 1431 |
quote_bound_value(bind_vars[match]) |
|---|
| 1432 |
else |
|---|
| 1433 |
raise PreparedStatementInvalid, "missing value for :#{match} in #{statement}" |
|---|
| 1434 |
end |
|---|
| 1435 |
end |
|---|
| 1436 |
end |
|---|
| 1437 |
|
|---|
| 1438 |
def quote_bound_value(value) |
|---|
| 1439 |
if value.respond_to?(:map) && !value.is_a?(String) |
|---|
| 1440 |
if value.respond_to?(:empty?) && value.empty? |
|---|
| 1441 |
connection.quote(nil) |
|---|
| 1442 |
else |
|---|
| 1443 |
value.map { |v| connection.quote(v) }.join(',') |
|---|
| 1444 |
end |
|---|
| 1445 |
else |
|---|
| 1446 |
connection.quote(value) |
|---|
| 1447 |
end |
|---|
| 1448 |
end |
|---|
| 1449 |
|
|---|
| 1450 |
def raise_if_bind_arity_mismatch(statement, expected, provided) |
|---|
| 1451 |
unless expected == provided |
|---|
| 1452 |
raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}" |
|---|
| 1453 |
end |
|---|
| 1454 |
end |
|---|
| 1455 |
|
|---|
| 1456 |
def extract_options_from_args!(args) |
|---|
| 1457 |
args.last.is_a?(Hash) ? args.pop : {} |
|---|
| 1458 |
end |
|---|
| 1459 |
|
|---|
| 1460 |
VALID_FIND_OPTIONS = [ :conditions, :include, :joins, :limit, :offset, |
|---|
| 1461 |
:order, :select, :readonly, :group, :from, :lock ] |
|---|
| 1462 |
|
|---|
| 1463 |
def validate_find_options(options) |
|---|
| 1464 |
options.assert_valid_keys(VALID_FIND_OPTIONS) |
|---|
| 1465 |
end |
|---|
| 1466 |
|
|---|
| 1467 |
def set_readonly_option!(options) |
|---|
| 1468 |
|
|---|
| 1469 |
|
|---|
| 1470 |
unless options.has_key?(:readonly) |
|---|
| 1471 |
if scoped?(:find, :readonly) |
|---|
| 1472 |
options[:readonly] = scope(:find, :readonly) |
|---|
| 1473 |
elsif !options[:joins].blank? && !options[:select] |
|---|
| 1474 |
options[:readonly] = true |
|---|
| 1475 |
end |
|---|
| 1476 |
end |
|---|
| 1477 |
end |
|---|
| 1478 |
|
|---|
| 1479 |
def encode_quoted_value(value) |
|---|
| 1480 |
quoted_value = connection.quote(value) |
|---|
| 1481 |
quoted_value = "'#{quoted_value[1..-2].gsub(/\'/, "\\\\'")}'" if quoted_value.include?("\\\'") # (for ruby mode) " |
|---|
| 1482 |
quoted_value |
|---|
| 1483 |
end |
|---|
| 1484 |
end |
|---|
| 1485 |
|
|---|
| 1486 |
public |
|---|
| 1487 |
|
|---|
| 1488 |
|
|---|
| 1489 |
|
|---|
| 1490 |
|
|---|
| 1491 |
def initialize(attributes = nil) |
|---|
| 1492 |
@attributes = attributes_from_column_definition |
|---|
| 1493 |
@new_record = true |
|---|
| 1494 |
ensure_proper_type |
|---|
| 1495 |
self.attributes = attributes unless attributes.nil? |
|---|
| 1496 |
yield self if block_given? |
|---|
| 1497 |
end |
|---|
| 1498 |
|
|---|
| 1499 |
|
|---|
| 1500 |
|
|---|
| 1501 |
def id |
|---|
| 1502 |
attr_name = self.class.primary_key |
|---|
| 1503 |
column = column_for_attribute(attr_name) |
|---|
| 1504 |
define_read_method(:id, attr_name, column) if self.class.generate_read_methods |
|---|
| 1505 |
read_attribute(attr_name) |
|---|
| 1506 |
end |
|---|
| 1507 |
|
|---|
| 1508 |
|
|---|
| 1509 |
def to_param |
|---|
| 1510 |
|
|---|
| 1511 |
(id = self.id) ? id.to_s : nil |
|---|
| 1512 |
end |
|---|
| 1513 |
|
|---|
| 1514 |
def id_before_type_cast |
|---|
| 1515 |
read_attribute_before_type_cast(self.class.primary_key) |
|---|
| 1516 |
end |
|---|
| 1517 |
|
|---|
| 1518 |
def quoted_id |
|---|
| 1519 |
quote_value(id, column_for_attribute(self.class.primary_key)) |
|---|
| 1520 |
end |
|---|
| 1521 |
|
|---|
| 1522 |
|
|---|
| 1523 |
def id=(value) |
|---|
| 1524 |
write_attribute(self.class.primary_key, value) |
|---|
| 1525 |
end |
|---|
| 1526 |
|
|---|
| 1527 |
|
|---|
| 1528 |
def new_record? |
|---|
| 1529 |
@new_record |
|---|
| 1530 |
end |
|---|
| 1531 |
|
|---|
| 1532 |
|
|---|
| 1533 |
|
|---|
| 1534 |
def save |
|---|
| 1535 |
create_or_update |
|---|
| 1536 |
end |
|---|
| 1537 |
|
|---|
| 1538 |
|
|---|
| 1539 |
|
|---|
| 1540 |
def save! |
|---|
| 1541 |
create_or_update || raise(RecordNotSaved) |
|---|
| 1542 |
end |
|---|
| 1543 |
|
|---|
| 1544 |
|
|---|
| 1545 |
|
|---|
| 1546 |
def destroy |
|---|
| 1547 |
unless new_record? |
|---|
| 1548 |
connection.delete <<-end_sql, "#{self.class.name} Destroy" |
|---|
| 1549 |
DELETE FROM |
|---|
| 1550 |
WHERE |
|---|
| 1551 |
end_sql |
|---|
| 1552 |
end |
|---|
| 1553 |
|
|---|
| 1554 |
freeze |
|---|
| 1555 |
end |
|---|
| 1556 |
|
|---|
| 1557 |
|
|---|
| 1558 |
|
|---|
| 1559 |
|
|---|
| 1560 |
|
|---|
| 1561 |
|
|---|
| 1562 |
def clone |
|---|
| 1563 |
attrs = self.attributes_before_type_cast |
|---|
| 1564 |
attrs.delete(self.class.primary_key) |
|---|
| 1565 |
self.class.new do |record| |
|---|
| 1566 |
record.send :instance_variable_set, '@attributes', attrs |
|---|
| 1567 |
end |
|---|
| 1568 |
end |
|---|
| 1569 |
|
|---|
| 1570 |
|
|---|
| 1571 |
|
|---|
| 1572 |
|
|---|
| 1573 |
def update_attribute(name, value) |
|---|
| 1574 |
send(name.to_s + '=', value) |
|---|
| 1575 |
save |
|---|
| 1576 |
end |
|---|
| 1577 |
|
|---|
| 1578 |
|
|---|
| 1579 |
|
|---|
| 1580 |
def update_attributes(attributes) |
|---|
| 1581 |
self.attributes = attributes |
|---|
| 1582 |
save |
|---|
| 1583 |
end |
|---|
| 1584 |
|
|---|
| 1585 |
|
|---|
| 1586 |
def update_attributes!(attributes) |
|---|
| 1587 |
self.attributes = attributes |
|---|
| 1588 |
save! |
|---|
| 1589 |
end |
|---|
| 1590 |
|
|---|
| 1591 |
|
|---|
| 1592 |
def increment(attribute) |
|---|
| 1593 |
self[attribute] ||= 0 |
|---|
| 1594 |
self[attribute] += 1 |
|---|
| 1595 |
self |
|---|
| 1596 |
end |
|---|
| 1597 |
|
|---|
| 1598 |
|
|---|
| 1599 |
def increment!(attribute) |
|---|
| 1600 |
increment(attribute).update_attribute(attribute, self[attribute]) |
|---|
| 1601 |
end |
|---|
| 1602 |
|
|---|
| 1603 |
|
|---|
| 1604 |
def decrement(attribute) |
|---|
| 1605 |
self[attribute] ||= 0 |
|---|
| 1606 |
self[attribute] -= 1 |
|---|
| 1607 |
self |
|---|
| 1608 |
end |
|---|
| 1609 |
|
|---|
| 1610 |
|
|---|
| 1611 |
def decrement!(attribute) |
|---|
| 1612 |
decrement(attribute).update_attribute(attribute, self[attribute]) |
|---|
| 1613 |
end |
|---|
| 1614 |
|
|---|
| 1615 |
|
|---|
| 1616 |
def toggle(attribute) |
|---|
| 1617 |
self[attribute] = !send("#{attribute}?") |
|---|
| 1618 |
self |
|---|
| 1619 |
end |
|---|
| 1620 |
|
|---|
| 1621 |
|
|---|
| 1622 |
def toggle!(attribute) |
|---|
| 1623 |
toggle(attribute).update_attribute(attribute, self[attribute]) |
|---|
| 1624 |
end |
|---|
| 1625 |
|
|---|
| 1626 |
|
|---|
| 1627 |
|
|---|
| 1628 |
|
|---|
| 1629 |
|
|---|
| 1630 |
def reload(options = nil) |
|---|
| 1631 |
clear_aggregation_cache |
|---|
| 1632 |
clear_association_cache |
|---|
| 1633 |
@attributes.update(self.class.find(self.id, options).instance_variable_get('@attributes')) |
|---|
| 1634 |
self |
|---|
| 1635 |
end |
|---|
| 1636 |
|
|---|
| 1637 |
|
|---|
| 1638 |
|
|---|
| 1639 |
|
|---|
| 1640 |
def [](attr_name) |
|---|
| 1641 |
read_attribute(attr_name) |
|---|
| 1642 |
end |
|---|
| 1643 |
|
|---|
| 1644 |
|
|---|
| 1645 |
|
|---|
| 1646 |
def []=(attr_name, value) |
|---|
| 1647 |
write_attribute(attr_name, value) |
|---|
| 1648 |
end |
|---|
| 1649 |
|
|---|
| 1650 |
|
|---|
| 1651 |
|
|---|
| 1652 |
|
|---|
| 1653 |
|
|---|
| 1654 |
|
|---|
| 1655 |
def attributes=(new_attributes) |
|---|
| 1656 |
return if new_attributes.nil? |
|---|
| 1657 |
attributes = new_attributes.dup |
|---|
| 1658 |
attributes.stringify_keys! |
|---|
| 1659 |
|
|---|
| 1660 |
multi_parameter_attributes = [] |
|---|
| 1661 |
remove_attributes_protected_from_mass_assignment(attributes).each do |k, v| |
|---|
| 1662 |
k.include?("(") ? multi_parameter_attributes << [ k, v ] : send(k + "=", v) |
|---|
| 1663 |
end |
|---|
| 1664 |
|
|---|
| 1665 |
assign_multiparameter_attributes(multi_parameter_attributes) |
|---|
| 1666 |
end |
|---|
| 1667 |
|
|---|
| 1668 |
|
|---|
| 1669 |
|
|---|
| 1670 |
def attributes(options = nil) |
|---|
| 1671 |
attributes = clone_attributes :read_attribute |
|---|
| 1672 |
|
|---|
| 1673 |
if options.nil? |
|---|
| 1674 |
attributes |
|---|
| 1675 |
else |
|---|
| 1676 |
if except = options[:except] |
|---|
| 1677 |
except = Array(except).collect { |attribute| attribute.to_s } |
|---|
| 1678 |
except.each { |attribute_name| attributes.delete(attribute_name) } |
|---|
| 1679 |
attributes |
|---|
| 1680 |
elsif only = options[:only] |
|---|
| 1681 |
only = Array(only).collect { |attribute| attribute.to_s } |
|---|
| 1682 |
attributes.delete_if { |key, value| !only.include?(key) } |
|---|
| 1683 |
attributes |
|---|
| 1684 |
else |
|---|
| 1685 |
raise ArgumentError, "Options does not specify :except or :only (#{options.keys.inspect})" |
|---|
| 1686 |
end |
|---|
| 1687 |
end |
|---|
| 1688 |
end |
|---|
| 1689 |
|
|---|
| 1690 |
|
|---|
| 1691 |
def attributes_before_type_cast |
|---|
| 1692 |
clone_attributes :read_attribute_before_type_cast |
|---|
| 1693 |
end |
|---|
| 1694 |
|
|---|
| 1695 |
|
|---|
| 1696 |
|
|---|
| 1697 |
def attribute_present?(attribute) |
|---|
| 1698 |
value = read_attribute(attribute) |
|---|
| 1699 |
!value.blank? or value == 0 |
|---|
| 1700 |
end |
|---|
| 1701 |
|
|---|
| 1702 |
|
|---|
| 1703 |
def has_attribute?(attr_name) |
|---|
| 1704 |
@attributes.has_key?(attr_name.to_s) |
|---|
| 1705 |
end |
|---|
| 1706 |
|
|---|
| 1707 |
|
|---|
| 1708 |
def attribute_names |
|---|
| 1709 |
@attributes.keys.sort |
|---|
| 1710 |
end |
|---|
| 1711 |
|
|---|
| 1712 |
|
|---|
| 1713 |
def column_for_attribute(name) |
|---|
| 1714 |
self.class.columns_hash[name.to_s] |
|---|
| 1715 |
end |
|---|
| 1716 |
|
|---|
| 1717 |
|
|---|
| 1718 |
def ==(comparison_object) |
|---|
| 1719 |
comparison_object.equal?(self) || |
|---|
| 1720 |
(comparison_object.instance_of?(self.class) && |
|---|
| 1721 |
comparison_object.id == id && |
|---|
| 1722 |
!comparison_object.new_record?) |
|---|
| 1723 |
end |
|---|
| 1724 |
|
|---|
| 1725 |
|
|---|
| 1726 |
def eql?(comparison_object) |
|---|
| 1727 |
self == (comparison_object) |
|---|
| 1728 |
end |
|---|
| 1729 |
|
|---|
| 1730 |
|
|---|
| 1731 |
|
|---|
| 1732 |
def hash |
|---|
| 1733 |
id.hash |
|---|
| 1734 |
end |
|---|
| 1735 |
|
|---|
| 1736 |
|
|---|
| 1737 |
alias_method :respond_to_without_attributes?, :respond_to? |
|---|
| 1738 |
|
|---|
| 1739 |
|
|---|
| 1740 |
|
|---|
| 1741 |
def respond_to?(method, include_priv = false) |
|---|
| 1742 |
if @attributes.nil? |
|---|
| 1743 |
return super |
|---|
| 1744 |
elsif attr_name = self.class.column_methods_hash[method.to_sym] |
|---|
| 1745 |
return true if @attributes.include?(attr_name) || attr_name == self.class.primary_key |
|---|
| 1746 |
return false if self.class.read_methods.include?(attr_name) |
|---|
| 1747 |
elsif @attributes.include?(method_name = method.to_s) |
|---|
| 1748 |
return true |
|---|
| 1749 |
elsif md = self.class.match_attribute_method?(method.to_s) |
|---|
| 1750 |
return true if @attributes.include?(md.pre_match) |
|---|
| 1751 |
end |
|---|
| 1752 |
|
|---|
| 1753 |
|
|---|
| 1754 |
super |
|---|
| 1755 |
end |
|---|
| 1756 |
|
|---|
| 1757 |
|
|---|
| 1758 |
def freeze |
|---|
| 1759 |
@attributes.freeze; self |
|---|
| 1760 |
end |
|---|
| 1761 |
|
|---|
| 1762 |
def frozen? |
|---|
| 1763 |
@attributes.frozen? |
|---|
| 1764 |
end |
|---|
| 1765 |
|
|---|
| 1766 |
|
|---|
| 1767 |
def readonly? |
|---|
| 1768 |
@readonly == true |
|---|
| 1769 |
end |
|---|
| 1770 |
|
|---|
| 1771 |
def readonly! |
|---|
| 1772 |
@readonly = true |
|---|
| 1773 |
end |
|---|
| 1774 |
|
|---|
| 1775 |
|
|---|
| 1776 |
private |
|---|
| 1777 |
def create_or_update |
|---|
| 1778 |
raise ReadOnlyRecord if readonly? |
|---|
| 1779 |
result = new_record? ? create : update |
|---|
| 1780 |
result != false |
|---|
| 1781 |
end |
|---|
| 1782 |
|
|---|
| 1783 |
|
|---|
| 1784 |
|
|---|
| 1785 |
def update |
|---|
| 1786 |
connection.update( |
|---|
| 1787 |
"UPDATE #{self.class.table_name} " + |
|---|
| 1788 |
"SET #{quoted_comma_pair_list(connection, attributes_with_quotes(false))} " + |
|---|
| 1789 |
"WHERE #{self.class.primary_key} = #{quote_value(id)}", |
|---|
| 1790 |
"#{self.class.name} Update" |
|---|
| 1791 |
) |
|---|
| 1792 |
end |
|---|
| 1793 |
|
|---|
| 1794 |
|
|---|
| 1795 |
|
|---|
| 1796 |
def create |
|---|
| 1797 |
if self.id.nil? && connection.prefetch_primary_key?(self.class.table_name) |
|---|
| 1798 |
self.id = connection.next_sequence_value(self.class.sequence_name) |
|---|
| 1799 |
end |
|---|
| 1800 |
|
|---|
| 1801 |
self.id = connection.insert( |
|---|
| 1802 |
"INSERT INTO #{self.class.table_name} " + |
|---|
| 1803 |
"(#{quoted_column_names.join(', ')}) " + |
|---|
| 1804 |
"VALUES(#{attributes_with_quotes.values.join(', ')})", |
|---|
| 1805 |
"#{self.class.name} Create", |
|---|
| 1806 |
self.class.primary_key, self.id, self.class.sequence_name |
|---|
| 1807 |
) |
|---|
| 1808 |
|
|---|
| 1809 |
@new_record = false |
|---|
| 1810 |
id |
|---|
| 1811 |
end |
|---|
| 1812 |
|
|---|
| 1813 |
|
|---|
| 1814 |
|
|---|
| 1815 |
|
|---|
| 1816 |
|
|---|
| 1817 |
def ensure_proper_type |
|---|
| 1818 |
unless self.class.descends_from_active_record? |
|---|
| 1819 |
write_attribute(self.class.inheritance_column, Inflector.demodulize(self.class.name)) |
|---|
| 1820 |
end |
|---|
| 1821 |
end |
|---|
| 1822 |
|
|---|
| 1823 |
|
|---|
| 1824 |
|
|---|
| 1825 |
|
|---|
| 1826 |
|
|---|
| 1827 |
|
|---|
| 1828 |
|
|---|
| 1829 |
|
|---|
| 1830 |
|
|---|
| 1831 |
|
|---|
| 1832 |
def method_missing(method_id, *args, &block) |
|---|
| 1833 |
method_name = method_id.to_s |
|---|
| 1834 |
if @attributes.include?(method_name) or |
|---|
| 1835 |
(md = /\?$/.match(method_name) and |
|---|
| 1836 |
@attributes.include?(query_method_name = md.pre_match) and |
|---|
| 1837 |
method_name = query_method_name) |
|---|
| 1838 |
define_read_methods if self.class.read_methods.empty? && self.class.generate_read_methods |
|---|
| 1839 |
md ? query_attribute(method_name) : read_attribute(method_name) |
|---|
| 1840 |
elsif self.class.primary_key.to_s == method_name |
|---|
| 1841 |
id |
|---|
| 1842 |
elsif md = self.class.match_attribute_method?(method_name) |
|---|
| 1843 |
attribute_name, method_type = md.pre_match, md.to_s |
|---|
| 1844 |
if @attributes.include?(attribute_name) |
|---|
| 1845 |
__send__("attribute#{method_type}", attribute_name, *args, &block) |
|---|
| 1846 |
else |
|---|
| 1847 |
super |
|---|
| 1848 |
end |
|---|
| 1849 |
else |
|---|
| 1850 |
super |
|---|
| 1851 |
end |
|---|
| 1852 |
end |
|---|
| 1853 |
|
|---|
| 1854 |
|
|---|
| 1855 |
|
|---|
| 1856 |
def read_attribute(attr_name) |
|---|
| 1857 |
attr_name = attr_name.to_s |
|---|
| 1858 |
if !(value = @attributes[attr_name]).nil? |
|---|
| 1859 |
if column = column_for_attribute(attr_name) |
|---|
| 1860 |
if unserializable_attribute?(attr_name, column) |
|---|
| 1861 |
unserialize_attribute(attr_name) |
|---|
| 1862 |
else |
|---|
| 1863 |
column.type_cast(value) |
|---|
| 1864 |
end |
|---|
| 1865 |
else |
|---|
| 1866 |
value |
|---|
| 1867 |
end |
|---|
| 1868 |
else |
|---|
| 1869 |
nil |
|---|
| 1870 |
end |
|---|
| 1871 |
end |
|---|
| 1872 |
|
|---|
| 1873 |
def read_attribute_before_type_cast(attr_name) |
|---|
| 1874 |
@attributes[attr_name] |
|---|
| 1875 |
end |
|---|
| 1876 |
|
|---|
| 1877 |
|
|---|
| 1878 |
|
|---|
| 1879 |
|
|---|
| 1880 |
def define_read_methods |
|---|
| 1881 |
self.class.columns_hash.each do |name, column| |
|---|
| 1882 |
unless respond_to_without_attributes?(name) |
|---|
| 1883 |
if self.class.serialized_attributes[name] |
|---|
| 1884 |
define_read_method_for_serialized_attribute(name) |
|---|
| 1885 |
else |
|---|
| 1886 |
define_read_method(name.to_sym, name, column) |
|---|
| 1887 |
end |
|---|
| 1888 |
end |
|---|
| 1889 |
|
|---|
| 1890 |
unless respond_to_without_attributes?("#{name}?") |
|---|
| 1891 |
define_question_method(name) |
|---|
| 1892 |
end |
|---|
| 1893 |
end |
|---|
| 1894 |
end |
|---|
| 1895 |
|
|---|
| 1896 |
|
|---|
| 1897 |
def define_read_method(symbol, attr_name, column) |
|---|
| 1898 |
cast_code = column.type_cast_code('v') if column |
|---|
| 1899 |
access_code = cast_code ? "(v=@attributes['#{attr_name}']) && #{cast_code}" : "@attributes['#{attr_name}']" |
|---|
| 1900 |
|
|---|
| 1901 |
unless attr_name.to_s == self.class.primary_key.to_s |
|---|
| 1902 |
access_code = access_code.insert(0, "raise NoMethodError, 'missing attribute: #{attr_name}', caller unless @attributes.has_key?('#{attr_name}'); ") |
|---|
| 1903 |
self.class.read_methods << attr_name |
|---|
| 1904 |
end |
|---|
| 1905 |
|
|---|
| 1906 |
evaluate_read_method attr_name, "def #{symbol}; #{access_code}; end" |
|---|
| 1907 |
end |
|---|
| 1908 |
|
|---|
| 1909 |
|
|---|
| 1910 |
def define_read_method_for_serialized_attribute(attr_name) |
|---|
| 1911 |
unless attr_name.to_s == self.class.primary_key.to_s |
|---|
| 1912 |
self.class.read_methods << attr_name |
|---|
| 1913 |
end |
|---|
| 1914 |
|
|---|
| 1915 |
evaluate_read_method attr_name, "def #{attr_name}; unserialize_attribute('#{attr_name}'); end" |
|---|
| 1916 |
end |
|---|
| 1917 |
|
|---|
| 1918 |
|
|---|
| 1919 |
def define_question_method(attr_name) |
|---|
| 1920 |
unless attr_name.to_s == self.class.primary_key.to_s |
|---|
| 1921 |
self.class.read_methods << "#{attr_name}?" |
|---|
| 1922 |
end |
|---|
| 1923 |
|
|---|
| 1924 |
evaluate_read_method attr_name, "def #{attr_name}?; query_attribute('#{attr_name}'); end" |
|---|
| 1925 |
end |
|---|
| 1926 |
|
|---|
| 1927 |
|
|---|
| 1928 |
def evaluate_read_method(attr_name, method_definition) |
|---|
| 1929 |
begin |
|---|
| 1930 |
self.class.class_eval(method_definition) |
|---|
| 1931 |
rescue SyntaxError => err |
|---|
| 1932 |
self.class.read_methods.delete(attr_name) |
|---|
| 1933 |
if logger |
|---|
| 1934 |
logger.warn "Exception occurred during reader method compilation." |
|---|
| 1935 |
logger.warn "Maybe #{attr_name} is not a valid Ruby identifier?" |
|---|
| 1936 |
logger.warn "#{err.message}" |
|---|
| 1937 |
end |
|---|
| 1938 |
end |
|---|
| 1939 |
end |
|---|
| 1940 |
|
|---|
| 1941 |
|
|---|
| 1942 |
def unserializable_attribute?(attr_name, column) |
|---|
| 1943 |
column.text? && self.class.serialized_attributes[attr_name] |
|---|
| 1944 |
end |
|---|
| 1945 |
|
|---|
| 1946 |
|
|---|
| 1947 |
def unserialize_attribute(attr_name) |
|---|
| 1948 |
unserialized_object = object_from_yaml(@attributes[attr_name]) |
|---|
| 1949 |
|
|---|
| 1950 |
if unserialized_object.is_a?(self.class.serialized_attributes[attr_name]) |
|---|
| 1951 |
@attributes[attr_name] = unserialized_object |
|---|
| 1952 |
else |
|---|
| 1953 |
raise SerializationTypeMismatch, |
|---|
| 1954 |
"#{attr_name} was supposed to be a #{self.class.serialized_attributes[attr_name]}, but was a #{unserialized_object.class.to_s}" |
|---|
| 1955 |
end |
|---|
| 1956 |
end |
|---|
| 1957 |
|
|---|
| 1958 |
|
|---|
| 1959 |
|
|---|
| 1960 |
def write_attribute(attr_name, value) |
|---|
| 1961 |
attr_name = attr_name.to_s |
|---|
| 1962 |
if (column = column_for_attribute(attr_name)) && column.number? |
|---|
| 1963 |
@attributes[attr_name] = convert_number_column_value(value) |
|---|
| 1964 |
else |
|---|
| 1965 |
@attributes[attr_name] = value |
|---|
| 1966 |
end |
|---|
| 1967 |
end |
|---|
| 1968 |
|
|---|
| 1969 |
def convert_number_column_value(value) |
|---|
| 1970 |
case value |
|---|
| 1971 |
when FalseClass: 0 |
|---|
| 1972 |
when TrueClass: 1 |
|---|
| 1973 |
when '': nil |
|---|
| 1974 |
else value |
|---|
| 1975 |
end |
|---|
| 1976 |
end |
|---|
| 1977 |
|
|---|
| 1978 |
def query_attribute(attr_name) |
|---|
| 1979 |
unless value = read_attribute(attr_name) |
|---|
| 1980 |
false |
|---|
| 1981 |
else |
|---|
| 1982 |
column = self.class.columns_hash[attr_name] |
|---|
| 1983 |
if column.number? |
|---|
| 1984 |
!value.zero? |
|---|
| 1985 |
else |
|---|
| 1986 |
!value.blank? |
|---|
| 1987 |
end |
|---|
| 1988 |
end |
|---|
| 1989 |
end |
|---|
| 1990 |
|
|---|
| 1991 |
def remove_attributes_protected_from_mass_assignment(attributes) |
|---|
| 1992 |
if self.class.accessible_attributes.nil? && self.class.protected_attributes.nil? |
|---|
| 1993 |
attributes.reject { |key, value| attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) } |
|---|
| 1994 |
elsif self.class.protected_attributes.nil? |
|---|
| 1995 |
attributes.reject { |key, value| !self.class.accessible_attributes.include?(key.gsub(/\(.+/, "").intern) || attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) } |
|---|
| 1996 |
elsif self.class.accessible_attributes.nil? |
|---|
| 1997 |
attributes.reject { |key, value| self.class.protected_attributes.include?(key.gsub(/\(.+/,"").intern) || attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) } |
|---|
| 1998 |
end |
|---|
| 1999 |
end |
|---|
| 2000 |
|
|---|
| 2001 |
|
|---|
| 2002 |
def attributes_protected_by_default |
|---|
| 2003 |
default = [ self.class.primary_key, self.class.inheritance_column ] |
|---|
| 2004 |
default << 'id' unless self.class.primary_key.eql? 'id' |
|---|
| 2005 |
default |
|---|
| 2006 |
end |
|---|
| 2007 |
|
|---|
| 2008 |
|
|---|
| 2009 |
|
|---|
| 2010 |
def attributes_with_quotes(include_primary_key = true) |
|---|
| 2011 |
attributes.inject({}) do |quoted, (name, value)| |
|---|
| 2012 |
if column = column_for_attribute(name) |
|---|
| 2013 |
quoted[name] = quote_value(value, column) unless !include_primary_key && column.primary |
|---|
| 2014 |
end |
|---|
| 2015 |
quoted |
|---|
| 2016 |
end |
|---|
| 2017 |
end |
|---|
| 2018 |
|
|---|
| 2019 |
|
|---|
| 2020 |
def quote_value(value, column = nil) |
|---|
| 2021 |
self.class.connection.quote(value, column) |
|---|
| 2022 |
end |
|---|
| 2023 |
|
|---|
| 2024 |
|
|---|
| 2025 |
def quote(value, column = nil) |
|---|
| 2026 |
self.class.connection.quote(value, column) |
|---|
| 2027 |
end |
|---|
| 2028 |
deprecate :quote => :quote_value |
|---|
| 2029 |
|
|---|
| 2030 |
|
|---|
| 2031 |
|
|---|
| 2032 |
|
|---|
| 2033 |
def interpolate_sql(sql, record = nil) |
|---|
| 2034 |
instance_eval("%@#{sql.gsub('@', '\@')}@") |
|---|
| 2035 |
end |
|---|
| 2036 |
|
|---|
| 2037 |
|
|---|
| 2038 |
|
|---|
| 2039 |
|
|---|
| 2040 |
|
|---|
| 2041 |
def attributes_from_column_definition |
|---|
| 2042 |
self.class.columns.inject({}) do |attributes, column| |
|---|
| 2043 |
attributes[column.name] = column.default unless column.name == self.class.primary_key |
|---|
| 2044 |
attributes |
|---|
| 2045 |
end |
|---|
| 2046 |
end |
|---|
| 2047 |
|
|---|
| 2048 |
|
|---|
| 2049 |
|
|---|
| 2050 |
|
|---|
| 2051 |
|
|---|
| 2052 |
|
|---|
| 2053 |
|
|---|
| 2054 |
def assign_multiparameter_attributes(pairs) |
|---|
| 2055 |
execute_callstack_for_multiparameter_attributes( |
|---|
| 2056 |
extract_callstack_for_multiparameter_attributes(pairs) |
|---|
| 2057 |
) |
|---|
| 2058 |
end |
|---|
| 2059 |
|
|---|
| 2060 |
|
|---|
| 2061 |
def execute_callstack_for_multiparameter_attributes(callstack) |
|---|
| 2062 |
errors = [] |
|---|
| 2063 |
callstack.each do |name, values| |
|---|
| 2064 |
klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass |
|---|
| 2065 |
if values.empty? |
|---|
| 2066 |
send(name + "=", nil) |
|---|
| 2067 |
else |
|---|
| 2068 |
begin |
|---|
| 2069 |
send(name + "=", Time == klass ? (@@default_timezone == :utc ? klass.utc(*values) : klass.local(*values)) : klass.new(*values)) |
|---|
| 2070 |
rescue => ex |
|---|
| 2071 |
errors << AttributeAssignmentError.new("error on assignment #{values.inspect} to #{name}", ex, name) |
|---|
| 2072 |
end |
|---|
| 2073 |
end |
|---|
| 2074 |
end |
|---|
| 2075 |
unless errors.empty? |
|---|
| 2076 |
raise MultiparameterAssignmentErrors.new(errors), "#{errors.size} error(s) on assignment of multiparameter attributes" |
|---|
| 2077 |
end |
|---|
| 2078 |
end |
|---|
| 2079 |
|
|---|
| 2080 |
def extract_callstack_for_multiparameter_attributes(pairs) |
|---|
| 2081 |
attributes = { } |
|---|
| 2082 |
|
|---|
| 2083 |
for pair in pairs |
|---|
| 2084 |
multiparameter_name, value = pair |
|---|
| 2085 |
attribute_name = multiparameter_name.split("(").first |
|---|
| 2086 |
attributes[attribute_name] = [] unless attributes.include?(attribute_name) |
|---|
| 2087 |
|
|---|
| 2088 |
unless value.empty? |
|---|
| 2089 |
attributes[attribute_name] << |
|---|
| 2090 |
[ find_parameter_position(multiparameter_name), type_cast_attribute_value(multiparameter_name, value) ] |
|---|
| 2091 |
end |
|---|
| 2092 |
end |
|---|
| 2093 |
|
|---|
| 2094 |
attributes.each { |name, values| attributes[name] = values.sort_by{ |v| v.first }.collect { |v| v.last } } |
|---|
| 2095 |
end |
|---|
| 2096 |
|
|---|
| 2097 |
def type_cast_attribute_value(multiparameter_name, value) |
|---|
| 2098 |
multiparameter_name =~ /\([0-9]*([a-z])\)/ ? value.send("to_" + $1) : value |
|---|
| 2099 |
end |
|---|
| 2100 |
|
|---|
| 2101 |
def find_parameter_position(multiparameter_name) |
|---|
| 2102 |
multiparameter_name.scan(/\(([0-9]*).*\)/).first.first |
|---|
| 2103 |
end |
|---|
| 2104 |
|
|---|
| 2105 |
|
|---|
| 2106 |
def comma_pair_list(hash) |
|---|
| 2107 |
hash.inject([]) { |list, pair| list << "#{pair.first} = #{pair.last}" }.join(", ") |
|---|
| 2108 |
end |
|---|
| 2109 |
|
|---|
| 2110 |
def quoted_column_names(attributes = attributes_with_quotes) |
|---|
| 2111 |
attributes.keys.collect do |column_name| |
|---|
| 2112 |
self.class.connection.quote_column_name(column_name) |
|---|
| 2113 |
end |
|---|
| 2114 |
end |
|---|
| 2115 |
|
|---|
| 2116 |
def quote_columns(quoter, hash) |
|---|
| 2117 |
hash.inject({}) do |quoted, (name, value)| |
|---|
| 2118 |
quoted[quoter.quote_column_name(name)] = value |
|---|
| 2119 |
quoted |
|---|
| 2120 |
end |
|---|
| 2121 |
end |
|---|
| 2122 |
|
|---|
| 2123 |
def quoted_comma_pair_list(quoter, hash) |
|---|
| 2124 |
comma_pair_list(quote_columns(quoter, hash)) |
|---|
| 2125 |
end |
|---|
| 2126 |
|
|---|
| 2127 |
def object_from_yaml(string) |
|---|
| 2128 |
return string unless string.is_a?(String) |
|---|
| 2129 |
YAML::load(string) rescue string |
|---|
| 2130 |
end |
|---|
| 2131 |
|
|---|
| 2132 |
def clone_attributes(reader_method = :read_attribute, attributes = {}) |
|---|
| 2133 |
self.attribute_names.inject(attributes) do |attributes, name| |
|---|
| 2134 |
attributes[name] = clone_attribute_value(reader_method, name) |
|---|
| 2135 |
attributes |
|---|
| 2136 |
end |
|---|
| 2137 |
end |
|---|
| 2138 |
|
|---|
| 2139 |
def clone_attribute_value(reader_method, attribute_name) |
|---|
| 2140 |
value = send(reader_method, attribute_name) |
|---|
| 2141 |
value.clone |
|---|
| 2142 |
rescue TypeError, NoMethodError |
|---|
| 2143 |
value |
|---|
| 2144 |
end |
|---|
| 2145 |
end |
|---|
| 2146 |
end |
|---|