Sha256: d36a13f72d310794f0950318f10ff711ca29b2594e41980068d0b2614293b452
Contents?: true
Size: 1.47 KB
Versions: 11
Compression:
Stored size: 1.47 KB
Contents
require 'active_support/inflector'
require 'neo4j/core/node'
wrapping_proc = proc do |node|
found_class = Neo4j::NodeWrapping.class_to_wrap(node.labels)
next node if not found_class
found_class.new.tap do |wrapped_node|
wrapped_node.init_on_load(node, node.props)
end
end
Neo4j::Driver::Types::Node.wrapper_callback(wrapping_proc)
module Neo4j
module NodeWrapping
# Only load classes once for performance
CONSTANTS_FOR_LABELS_CACHE = {}
class << self
def class_to_wrap(labels)
load_classes_from_labels(labels)
Neo4j::ActiveNode::Labels.model_for_labels(labels).tap do |model_class|
populate_constants_for_labels_cache(model_class, labels)
end
end
private
def load_classes_from_labels(labels)
labels.each { |label| constant_for_label(label) }
end
def constant_for_label(label)
CONSTANTS_FOR_LABELS_CACHE[label] || CONSTANTS_FOR_LABELS_CACHE[label] = constantized_label(label)
end
def constantized_label(label)
"#{association_model_namespace}::#{label}".constantize
rescue NameError, LoadError
nil
end
def populate_constants_for_labels_cache(model_class, labels)
labels.each do |label|
CONSTANTS_FOR_LABELS_CACHE[label] = model_class if CONSTANTS_FOR_LABELS_CACHE[label].nil?
end
end
def association_model_namespace
Neo4j::Config.association_model_namespace_string
end
end
end
end
Version data entries
11 entries across 11 versions & 2 rubygems