don't overwrite cached object fields on fetch all since we can't guarantee that it...
[dbsrgits/DBIx-Data-Store-old.git] / lib / DBIx / Data / Collection / Set.pm
index 1d772ce..1acf096 100644 (file)
@@ -6,31 +6,135 @@ use Data::Perl::Stream::Array;
 
 has _store => (is => 'ro', required => 1, init_arg => 'store');
 
-has _column_order => (is => 'ro', required => 1, init_arg => 'column_order');
-
 has _class => (is => 'ro', predicate => '_has_class');
 
-has _member_cache => (is => 'rw', lazy_build => 1);
+has _set_over => (is => 'ro', required => 1, init_arg => 'set_over');
+
+## member cache (all members)
+
+has _member_cache => (
+  is => 'rw', lazy_build => 1,
+  predicate => '_member_cache_built',
+);
 
 method _build__member_cache {
   my $stream = $self->_new_raw_stream;
   my @cache;
   while (my ($raw) = $stream->next) {
-    push @cache, $self->_inflate($raw);
+    my $obj = do {
+      if (my ($obj) = $self->_key_cache_get_raw($raw)) {
+        $obj # can't $self->_merge($obj, $raw) since $obj might have changed
+      } else {
+        $self->_add_to_key_cache($self->_inflate($raw))
+      }
+    };
+    push @cache, $obj;
   }
-  \@cache;
+  \@cache
 }
 
-method _new_raw_stream {
-  $self->_store->new_select_command([])->execute;
+method _add_to_member_cache ($to_add) {
+  return $to_add unless $self->_member_cache_built;
+  push @{$self->_member_cache}, $to_add;
+  $to_add
 }
 
+method _remove_from_member_cache ($to_remove) {
+  return $to_remove unless $self->_member_cache_built;
+  @{$self->_member_cache} = grep $_ ne $to_remove, @{$self->_member_cache};
+  $to_remove
+}
+
+## key cache - by primary/unique key
+
+has _key_cache => (is => 'ro', default => sub { {} });
+
+method _add_to_key_cache ($to_add) {
+  $self->_key_cache->{$self->_object_to_id($to_add)} = $to_add;
+  $to_add
+}
+
+method _remove_from_key_cache ($to_remove) {
+  # should return $to_remove
+  delete $self->_key_cache->{$self->_object_to_id($to_remove)}
+}
+
+method _key_cache_has_raw ($raw) {
+  exists $self->_key_cache->{$self->_raw_to_id($raw)}
+}
+
+method _key_cache_has_object ($obj) {
+  exists $self->_key_cache->{$self->_object_to_id($obj)}
+}
+
+method _key_cache_get_raw ($raw) {
+  $self->_key_cache_get_id($self->_raw_to_id($raw))
+}
+
+method _key_cache_get_object ($obj) {
+  $self->_key_cache_get_id($self->_object_to_id($obj))
+}
+
+method _key_cache_get_object_spec ($spec) {
+  # see _object_spec_to_id for doc of what the difference is
+  $self->_key_cache_get_id($self->_object_spec_to_id($spec))
+}
+
+method _key_cache_get_id ($id) {
+  exists $self->_key_cache->{$id}
+    ? ($self->_key_cache->{$id})
+    : ()
+}
+
+## thunking between the store representation and the set representation
+#
+# _inflate is raw data -> final repr
+# _deflate is final repr -> raw data
+# _merge takes final repr + raw data and updates the repr
+#    (this is used for pk-generated values and later lazy loading)
+#
+# _deflate_spec is attributes of final repr -> raw data
+
 method _inflate ($raw) {
-  my @order = @{$self->_column_order};
-  my %final;
-  @final{@order} = @$raw;
-  bless(\%final, $self->_class) if $self->_has_class;
-  \%final;
+  bless($raw, $self->_class) if $self->_has_class;
+  $raw
+}
+
+method _deflate ($obj) {
+  +{ %$obj }
+}
+
+method _merge ($obj, $raw) {
+  @{$obj}{keys %$raw} = values %$raw;
+  $obj
+}
+
+method _deflate_spec ($spec) {
+  $spec
+}
+
+## methods to get ids
+
+method _raw_to_id ($raw) {
+  # XXX must escape this. or do something else.
+  join ';', map $raw->{$_}, @{$self->_set_over}
+}
+
+method _object_to_id ($obj) {
+  $self->_raw_to_id($self->_deflate($obj))
+}
+
+method _object_spec_to_id ($spec) {
+  # intentionally C&P from _raw_to - this is not the same thing. If a column
+  # were mapped to an attribute of a different name, the raw would have the
+  # column name as a key but an object spec would have the attribute name
+  join ';', map $spec->{$_}, @{$self->_set_over}
+}
+
+## array-ish operations - i.e. get all members
+
+method _new_raw_stream {
+  $self->_store->new_select_command([])->execute
 }
 
 method flatten {
@@ -41,4 +145,67 @@ method as_stream {
   Data::Perl::Stream::Array->new(array => $self->_member_cache);
 }
 
+## load single row
+
+method get ($spec) {
+  if (my ($got) = $self->_key_cache_get_object_spec($spec)) {
+    return $got
+  }
+  if (my ($raw) = $self->_get_from_store($self->_deflate_spec($spec))) {
+    return $self->_add_to_key_cache($self->_inflate($raw))
+  }
+  return undef # we aren't handling cache misses here yet
+}
+
+method _get_from_store ($raw) {
+  $self->_store->new_select_single_command($raw)->execute
+}
+
+## add to set
+
+method add ($new) {
+  $self->_add_to_store($new);
+  $self->_add_to_caches($new);
+  $new
+}
+
+method _add_to_store ($new) {
+  my $new_raw = $self->_deflate($new);
+  $self->_merge($new, $self->_store->new_insert_command($new_raw)->execute);
+  $new
+}
+
+method _add_to_caches ($new) {
+  $self->_add_to_member_cache($new);
+  $self->_add_to_key_cache($new);
+  $new
+}
+
+## remove from set
+
+method remove ($old) {
+  $self->_remove_from_store($old);
+  $self->_remove_from_caches($old);
+  $old
+}
+
+method _remove_from_store ($old) {
+  $self->_store->new_delete_command($self->_deflate($old))->execute
+}
+
+method _remove_from_caches ($old) {
+  $self->_remove_from_member_cache($old);
+  $self->_remove_from_key_cache($old);
+  $old
+}
+
+## update
+
+method _update_in_store ($obj) {
+  # this is currently a call command but we should think about it
+  # being a row command so that we can have RETURNING or other
+  # mechanisms handle things like set-on-update datetime values
+  $self->_store->new_update_command($self->_deflate($obj))->execute
+}
+
 1;