Changeset 23553

Timestamp:

11/13/08 18:19:29 (5 years ago)

Author:

yappo

Message:

add replace

Location:

lang/perl/Data-Model/trunk/lib/Data

Files:

• Model.pm (modified) (3 diffs)
• Model/Driver/DBI.pm (modified) (1 diff)
• Model/Driver/DBI/DBD.pm (modified) (1 diff)
• Model/Driver/Memory.pm (modified) (1 diff)

lang/perl/Data-Model/trunk/lib/Data/Model.pm

@@ -167 +167 @@
 
 sub set {
-    my $self   = shift;
-    my $model  = shift;
+    shift->_insert_or_replace(0, @_);
+}
+
+sub replace {
+    shift->_insert_or_replace(1, @_);
+}
+
+sub _insert_or_replace {
+    my $self       = shift;
+    my $is_replace = shift;
+    my $model      = shift;
     return $self->update($model, @_) if ref($model) && $model->isa('Data::Model::Row');
     my $schema = $self->get_schema($model);
@@ -205 +214 @@
 
     local $schema->{schema_obj} = $self;
-    my $result = $schema->{driver}->set( $schema, $key_array => $columns, @_ );
+    my $method = $is_replace ? 'replace' : 'set';
+    my $result = $schema->{driver}->$method( $schema, $key_array => $columns, @_ );
     return unless $result;
 
@@ -218 +228 @@
 sub set_multi {
 }
+
 
 

lang/perl/Data-Model/trunk/lib/Data/Model/Driver/DBI.pm

@@ -136 +136 @@
 # insert or replace
 sub set {
-    my($self, $schema, $key, $columns, %args) = @_;
+    my $self = shift;
+    $self->_insert_or_replace(0, @_);
+}
+
+sub replace {
+    my($self, $schema, $key, $columns, %args) = @_;
+    if ($self->dbd->can_replace) {
+        return $self->_insert_or_replace(1, $schema, $key, $columns, %args);
+    } else {
+#        $self->thx(sub {
+        $self->delete($schema, $key, +{}, %args);
+        return $self->set($schema, $key, $columns, %args);
+#        });
+    }
+}
+
+sub _insert_or_replace {
+    my($self, $is_replace, $schema, $key, $columns, %args) = @_;
+    my $select_or_replace = $is_replace ? 'REPLACE' : 'INSERT';
 
     my $table = $schema->{model};
     my $cols = [ keys %{ $columns } ];
-    my $sql = "INSERT INTO $table\n";
+    my $sql = "$select_or_replace INTO $table\n";
     $sql .= '(' . join(', ', @{ $cols }) . ')' . "\n" .
             'VALUES (' . join(', ', ('?') x @{ $cols }) . ')' . "\n";

lang/perl/Data-Model/trunk/lib/Data/Model/Driver/DBI/DBD.pm

@@ -13 +13 @@
 sub fetch_last_id {}
 sub bind_param_attributes {}
+sub can_replace { 1 }
 
 1;

lang/perl/Data-Model/trunk/lib/Data/Model/Driver/Memory.pm

@@ -95 +95 @@
     my $data = $self->load_data($schema);
     $data->{records}->{$record_id} = $columns;
+}
+
+sub replace {
+    my($self, $schema, $key, $columns, %args) = @_;
+}
+
+sub update {
+    my($self, $schema, $key, $columns, %args) = @_;
 }