Class: DataMapper::Adapters::AppEngineAdapter

Inherits:
AbstractAdapter
  • Object
show all
Defined in:
lib/appengine_adapter.rb

Defined Under Namespace

Classes: QueryBuilder

Constant Summary collapse

Datastore = 
AppEngine::Datastore

Instance Method Summary collapse

Constructor Details

#initialize(name, uri_or_options) ⇒ AppEngineAdapter

Returns a new instance of AppEngineAdapter.



36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/appengine_adapter.rb', line 36

def initialize(name, uri_or_options)
  super
  if uri_or_options.kind_of? Hash
    options = uri_or_options
    if options['host'] == 'memory'
      require 'appengine-apis/testing'
      AppEngine::Testing::install_test_datastore
    end
  end
  @resource_naming_convention = lambda do |value|
    Extlib::Inflection.pluralize(Extlib::Inflection.camelize(value))
  end
end

Instance Method Details

#convert_value(property, value) ⇒ Object 



122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/appengine_adapter.rb', line 122

def convert_value(property, value)
  value = property.value(value)
  if property.type == DataMapper::Types::Text && value
    AppEngine::Datastore::Text.new(value)
  else
    case value
    when Date, DateTime
      Time.parse(value.to_s)
    when BigDecimal
      value.to_s
    when Class
      value.name
    else
      value
    end
  end
end

#create(resources) ⇒ Object 



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
 
# File 'lib/appengine_adapter.rb', line 54

def create(resources)
  created = 0
  entities = []
  resources.each do |resource|
    repository = resource.repository
    model = resource.model
    attributes = resource.attributes
    properties = model.properties(repository.name)
    
    kind = self.kind(model)
    keys = properties.key
    raise "Multiple keys in #{resource.inspect}" if keys.size > 1
    if keys.size == 1
      name = keys.first.name
      property = properties[name]
      key = convert_value(property, attributes.delete(name))
    end
    if key.nil? || keys.first.serial? || key==0
      entity = Datastore::Entity.new(kind)
    elsif key.kind_of?(AppEngine::Datastore::Key)
      entity = Datastore::Entity.new(key)
    elsif key.kind_of?(Hash) && property.type == Types::Key
      # AppEngine::Datastore::Key should already have filtered this.
      # Since it didn't, we know it's a serial object with a parent.
      entity = Datastore::Entity.new(kind, key[:parent])
    else
      entity = Datastore::Entity.new(kind, key)
    end
    
    attributes.each do |name, value|
      property = properties[name]
      value = convert_value(property, value)
      entity.set_property(property.field, value)
    end
    entities << entity
    created += 1
  end
  Datastore.put(entities)
  resources.zip(entities) do |resource, entity|
    key = entity.key
    if id = resource.model.serial(name)
      id.set!(resource, key.get_id)
    elsif id = resource.model.key(name).find{|k|k.type == Types::Key}
      id.set!(resource, key)
    end
    resource.instance_variable_set :@__entity__, entity
  end
  return created
end

#delete(collection) ⇒ Object 



140
141
142
143
144
145
146
 
# File 'lib/appengine_adapter.rb', line 140

def delete(collection)
  keys = collection.collect do |resource|
    entity = resource.instance_variable_get :@__entity__
    entity.key
  end
  Datastore.delete(keys)
end

#kind(model) ⇒ Object 



50
51
52
 
# File 'lib/appengine_adapter.rb', line 50

def kind(model)
  model.storage_name(name)
end

#read(query) ⇒ Object 



104
105
106
107
 
# File 'lib/appengine_adapter.rb', line 104

def read(query)
  query = QueryBuilder.new(query, kind(query.model), self)
  query.run
end

#update(attributes, collection) ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
 
# File 'lib/appengine_adapter.rb', line 109

def update(attributes, collection)
  attributes = attributes.map do |property, value|
    [property.field, convert_value(property, value)]
  end
  entities = collection.collect do |resource|
    entity = resource.instance_variable_get :@__entity__
    entity.update(attributes)
  end

  Datastore.put(entities)
  entities.size
end