1 use strict; use warnings;
2 package OpenILS::Utils::CStoreEditor;
3 use OpenILS::Application::AppUtils;
4 use OpenSRF::AppSession;
5 use OpenSRF::EX qw(:try);
6 use OpenILS::Utils::Fieldmapper;
9 use OpenSRF::Utils::JSON;
10 use OpenSRF::Utils::Logger qw($logger);
11 my $U = "OpenILS::Application::AppUtils";
16 # 'biblio.record_entry' => { update => 'UPDATE_MARC' },
17 # 'asset.copy' => { update => 'UPDATE_COPY'},
18 # 'asset.call_number' => { update => 'UPDATE_VOLUME'},
19 # 'action.circulation' => { retrieve => 'VIEW_CIRCULATIONS'},
24 # -----------------------------------------------------------------------------
25 # Export some useful functions
26 # -----------------------------------------------------------------------------
27 use vars qw(@EXPORT_OK %EXPORT_TAGS);
29 use base qw/Exporter/;
30 push @EXPORT_OK, ( 'new_editor', 'new_rstore_editor' );
31 %EXPORT_TAGS = ( funcs => [ qw/ new_editor new_rstore_editor / ] );
33 sub new_editor { return OpenILS::Utils::CStoreEditor->new(@_); }
35 sub new_rstore_editor {
36 my $e = OpenILS::Utils::CStoreEditor->new(@_);
37 $e->app('open-ils.reporter-store');
42 # -----------------------------------------------------------------------------
44 # -----------------------------------------------------------------------------
45 use constant E => 'error';
46 use constant W => 'warn';
47 use constant I => 'info';
48 use constant D => 'debug';
49 use constant A => 'activity';
53 # -----------------------------------------------------------------------------
55 # xact=><true> : creates a storage transaction
56 # authtoken=>$token : the login session key
57 # -----------------------------------------------------------------------------
59 my( $class, %params ) = @_;
60 $class = ref($class) || $class;
61 my $self = bless( \%params, $class );
62 $self->{checked_perms} = {};
68 my( $self, $app ) = @_;
69 $self->{app} = $app if $app;
70 $self->{app} = 'open-ils.cstore' unless $self->{app};
75 # -----------------------------------------------------------------------------
76 # Log the editor metadata along with the log string
77 # -----------------------------------------------------------------------------
79 my( $self, $lev, $str ) = @_;
81 $s .= "0|" unless $self->{xact};
82 $s .= "1|" if $self->{xact};
83 $s .= "0" unless $self->requestor;
84 $s .= $self->requestor->id if $self->requestor;
86 $logger->$lev("$s $str");
89 # -----------------------------------------------------------------------------
90 # Verifies the auth token and fetches the requestor object
91 # -----------------------------------------------------------------------------
94 $self->log(D, "checking auth token ".$self->authtoken);
95 my ($reqr, $evt) = $U->checkses($self->authtoken);
96 $self->event($evt) if $evt;
97 return $self->{requestor} = $reqr;
104 $cache = OpenSRF::Utils::Cache->new('global') unless $cache;
105 $self->log(D, "checking cached auth token ".$self->authtoken);
106 my $user = $cache->get_cache("oils_auth_".$self->authtoken);
107 return $self->{requestor} = $user->{userobj} if $user;
108 $self->event(OpenILS::Event->new('NO_SESSION'));
114 # -----------------------------------------------------------------------------
115 # Returns the last generated event
116 # -----------------------------------------------------------------------------
118 my( $self, $evt ) = @_;
119 $self->{event} = $evt if $evt;
120 return $self->{event};
123 # -----------------------------------------------------------------------------
124 # Destroys the transaction and disconnects where necessary,
125 # then returns the last event that occurred
126 # -----------------------------------------------------------------------------
134 # -----------------------------------------------------------------------------
135 # Clears the last caught event
136 # -----------------------------------------------------------------------------
139 $self->{event} = undef;
143 my( $self, $auth ) = @_;
144 $self->{authtoken} = $auth if $auth;
145 return $self->{authtoken};
150 $self->{timeout} = $to if defined $to;
151 return defined($self->{timeout}) ? $self->{timeout} : 60;
154 # -----------------------------------------------------------------------------
155 # fetches the session, creating if necessary. If 'xact' is true on this
156 # object, a db session is created
157 # -----------------------------------------------------------------------------
159 my( $self, $session ) = @_;
160 $self->{session} = $session if $session;
162 if(!$self->{session}) {
163 $self->{session} = OpenSRF::AppSession->create($self->app);
165 if( ! $self->{session} ) {
166 my $str = "Error creating cstore session with OpenSRF::AppSession->create()!";
168 throw OpenSRF::EX::ERROR ($str);
171 $self->{session}->connect if $self->{xact} or $self->{connect};
172 $self->xact_start if $self->{xact};
174 return $self->{session};
178 # -----------------------------------------------------------------------------
179 # Starts a storage transaction
180 # -----------------------------------------------------------------------------
183 $self->log(D, "starting new db session");
184 my $stat = $self->request($self->app . '.transaction.begin');
185 $self->log(E, "error starting database transaction") unless $stat;
189 # -----------------------------------------------------------------------------
190 # Commits a storage transaction
191 # -----------------------------------------------------------------------------
194 $self->log(D, "comitting db session");
195 my $stat = $self->request($self->app.'.transaction.commit');
196 $self->log(E, "error comitting database transaction") unless $stat;
200 # -----------------------------------------------------------------------------
201 # Rolls back a storage stransaction
202 # -----------------------------------------------------------------------------
205 return unless $self->{session};
206 $self->log(I, "rolling back db session");
207 return $self->request($self->app.".transaction.rollback");
212 # -----------------------------------------------------------------------------
213 # Rolls back the transaction and disconnects
214 # -----------------------------------------------------------------------------
217 $self->xact_rollback if $self->{xact};
218 delete $self->{xact};
224 $self->session->disconnect if $self->{session};
225 delete $self->{session};
229 # -----------------------------------------------------------------------------
230 # commits the db session and destroys the session
231 # -----------------------------------------------------------------------------
234 return unless $self->{xact};
236 $self->session->disconnect;
237 $self->{session} = undef;
240 # -----------------------------------------------------------------------------
241 # clears all object data. Does not commit the db transaction.
242 # -----------------------------------------------------------------------------
246 $$self{$_} = undef for (keys %$self);
250 # -----------------------------------------------------------------------------
252 # -----------------------------------------------------------------------------
261 # -----------------------------------------------------------------------------
262 # Does a simple storage request
263 # -----------------------------------------------------------------------------
265 my( $self, $method, @params ) = @_;
269 my $argstr = __arg_to_string( (scalar(@params)) == 1 ? $params[0] : \@params);
271 $self->log(I, "request $method : $argstr");
273 if( $self->{xact} and
274 $self->session->state != OpenSRF::AppSession::CONNECTED() ) {
275 $logger->error("CStoreEditor lost it's connection!!");
276 #throw OpenSRF::EX::ERROR ("CStoreEditor lost it's connection - transaction cannot continue");
282 my $req = $self->session->request($method, @params);
284 if($self->substream) {
285 $self->log(D,"running in substream mode");
287 while( my $resp = $req->recv(timeout => $self->timeout) ) {
288 push(@$val, $resp->content) if $resp->content;
292 my $resp = $req->recv(timeout => $self->timeout);
293 $val = $resp->content;
300 $self->log(E, "request error $method : $argstr : $err");
308 my( $self, $bool ) = @_;
309 $self->{substream} = $bool if defined $bool;
310 return $self->{substream};
314 # -----------------------------------------------------------------------------
315 # Sets / Returns the requstor object. This is set when checkauth succeeds.
316 # -----------------------------------------------------------------------------
318 my($self, $requestor) = @_;
319 $self->{requestor} = $requestor if $requestor;
320 return $self->{requestor};
325 # -----------------------------------------------------------------------------
326 # Holds the last data received from a storage call
327 # -----------------------------------------------------------------------------
329 my( $self, $data ) = @_;
330 $self->{data} = $data if defined $data;
331 return $self->{data};
335 # -----------------------------------------------------------------------------
336 # True if this perm has already been checked at this org
337 # -----------------------------------------------------------------------------
339 my( $self, $perm, $org ) = @_;
340 $self->{checked_perms}->{$org} = {}
341 unless $self->{checked_perms}->{$org};
342 my $checked = $self->{checked_perms}->{$org}->{$perm};
344 $self->{checked_perms}->{$org}->{$perm} = 1;
352 # -----------------------------------------------------------------------------
353 # Returns true if the requested perm is allowed. If the perm check fails,
354 # $e->event is set and undef is returned
355 # The perm user is $e->requestor->id and perm org defaults to the requestor's
357 # If this perm at the given org has already been verified, true is returned
358 # and the perm is not re-checked
359 # -----------------------------------------------------------------------------
361 my( $self, $perm, $org ) = @_;
362 my $uid = $self->requestor->id;
363 $org ||= $self->requestor->ws_ou;
364 $self->log(I, "checking perms user=$uid, org=$org, perm=$perm");
365 return 1 if $self->perm_checked($perm, $org);
366 return $self->checkperm($uid, $org, $perm);
370 my($self, $userid, $org, $perm) = @_;
371 my $s = $U->storagereq(
372 "open-ils.storage.permission.user_has_perm", $userid, $perm, $org );
375 my $e = OpenILS::Event->new('PERM_FAILURE', ilsperm => $perm, ilspermloc => $org);
385 # -----------------------------------------------------------------------------
386 # checks the appropriate perm for the operation
387 # -----------------------------------------------------------------------------
389 my( $self, $ptype, $action, $org ) = @_;
390 $org ||= $self->requestor->ws_ou;
391 my $perm = $PERMS{$ptype}{$action};
393 return undef if $self->perm_checked($perm, $org);
394 return $self->event unless $self->allowed($perm, $org);
396 $self->log(I, "no perm provided for $ptype.$action");
403 # -----------------------------------------------------------------------------
404 # Logs update actions to the activity log
405 # -----------------------------------------------------------------------------
407 my( $self, $type, $action, $arg ) = @_;
408 my $str = "$type.$action";
409 $str .= _prop_string($arg);
417 my @props = $obj->properties;
420 my $prop = $obj->$_() || "";
421 $prop = substr($prop, 0, 128) . "..." if length $prop > 131;
428 sub __arg_to_string {
430 return "" unless defined $arg;
431 if( UNIVERSAL::isa($arg, "Fieldmapper") ) {
432 return (defined $arg->id) ? $arg->id : '<new object>';
434 return OpenSRF::Utils::JSON->perl2JSON($arg);
439 # -----------------------------------------------------------------------------
440 # This does the actual storage query.
442 # 'search' calls become search_where calls and $arg can be a search hash or
443 # an array-ref of storage search options.
445 # 'retrieve' expects an id
446 # 'update' expects an object
447 # 'create' expects an object
448 # 'delete' expects an object
450 # All methods return true on success and undef on failure. On failure,
451 # $e->event is set to the generated event.
452 # Note: this method assumes that updating a non-changed object and
453 # thereby receiving a 0 from storage, is a successful update.
455 # The method will therefore return true so the caller can just do
456 # $e->update_blah($x) or return $e->event;
457 # The true value returned from storage for all methods will be stored in
458 # $e->data, until the next method is called.
460 # not-found events are generated on retrieve and serach methods.
461 # action=search methods will return [] (==true) if no data is found. If the
462 # caller is interested in the not found event, they can do:
463 # return $e->event unless @$results;
464 # -----------------------------------------------------------------------------
466 my( $self, $action, $type, $arg, $options ) = @_;
470 if( $action eq 'retrieve' ) {
471 if(! defined($arg) ) {
472 $self->log(W,"$action $type called with no ID...");
473 $self->event(_mk_not_found($type, $arg));
475 } elsif( ref($arg) =~ /Fieldmapper/ ) {
476 $self->log(E,"$action $type called with an object.. attempting ID retrieval..");
481 my @arg = ( ref($arg) eq 'ARRAY' ) ? @$arg : ($arg);
482 my $method = $self->app.".direct.$type.$action";
484 if( $action eq 'search' ) {
485 $method .= '.atomic';
487 } elsif( $action eq 'batch_retrieve' ) {
489 @arg = ( { id => $arg } );
490 $method =~ s/batch_retrieve/search/o;
491 $method .= '.atomic';
493 } elsif( $action eq 'retrieve_all' ) {
495 $method =~ s/retrieve_all/search/o;
498 my $fmobj = "Fieldmapper::$tt";
499 @arg = ( { $fmobj->Identity => { '!=' => undef } } );
500 $method .= '.atomic';
503 $method =~ s/search/id_list/o if $options->{idlist};
505 $method =~ s/\.atomic$//o if $self->substream($$options{substream} || 0);
506 $self->timeout($$options{timeout});
508 # remove any stale events
511 if( $action eq 'update' or $action eq 'delete' or $action eq 'create' ) {
513 $logger->error("Attempt to update DB while not in a transaction : $method");
514 throw OpenSRF::EX::ERROR ("Attempt to update DB while not in a transaction : $method");
516 $self->log_activity($type, $action, $arg);
519 if($$options{checkperm}) {
520 my $a = ($action eq 'search') ? 'retrieve' : $action;
521 my $e = $self->_checkperm($type, $a, $$options{permorg});
532 $obj = $self->request($method, @arg);
533 } catch Error with { $err = shift; };
537 $self->log(I, "request returned no data : $method");
539 if( $action eq 'retrieve' ) {
540 $self->event(_mk_not_found($type, $arg));
542 } elsif( $action eq 'update' or
543 $action eq 'delete' or $action eq 'create' ) {
544 my $evt = OpenILS::Event->new(
545 'DATABASE_UPDATE_FAILED', payload => $arg, debug => "$err" );
551 OpenILS::Event->new( 'DATABASE_QUERY_FAILED',
552 payload => $arg, debug => "$err" ));
559 if( $action eq 'create' and $obj == 0 ) {
560 my $evt = OpenILS::Event->new(
561 'DATABASE_UPDATE_FAILED', payload => $arg, debug => "$err" );
566 # If we havn't dealt with the error in a nice way, go ahead and throw it
569 OpenILS::Event->new( 'DATABASE_QUERY_FAILED',
570 payload => $arg, debug => "$err" ));
574 if( $action eq 'search' ) {
575 $self->log(I, "$type.$action : returned ".scalar(@$obj). " result(s)");
576 $self->event(_mk_not_found($type, $arg)) unless @$obj;
579 if( $action eq 'create' ) {
580 $self->log(I, "created a new $type object with ID " . $obj->id);
584 $self->data($obj); # cache the data for convenience
586 return ($obj) ? $obj : 1;
591 my( $type, $arg ) = @_;
592 (my $t = $type) =~ s/\./_/og;
594 return OpenILS::Event->new("${t}_NOT_FOUND", payload => $arg);
599 # utility method for loading
603 $str =~ s/Fieldmapper:://o;
609 # -------------------------------------------------------------
610 # Load up the methods from the FM classes
611 # -------------------------------------------------------------
612 my $map = $Fieldmapper::fieldmap;
613 for my $object (keys %$map) {
614 my $obj = __fm2meth($object,'_');
615 my $type = __fm2meth($object, '.');
617 my $update = "update_$obj";
619 "sub $update {return shift()->runmethod('update', '$type', \@_);}";
622 my $retrieve = "retrieve_$obj";
624 "sub $retrieve {return shift()->runmethod('retrieve', '$type', \@_);}";
627 my $search = "search_$obj";
629 "sub $search {return shift()->runmethod('search', '$type', \@_);}";
632 my $create = "create_$obj";
634 "sub $create {return shift()->runmethod('create', '$type', \@_);}";
637 my $delete = "delete_$obj";
639 "sub $delete {return shift()->runmethod('delete', '$type', \@_);}";
642 my $bretrieve = "batch_retrieve_$obj";
644 "sub $bretrieve {return shift()->runmethod('batch_retrieve', '$type', \@_);}";
647 my $retrieveall = "retrieve_all_$obj";
649 "sub $retrieveall {return shift()->runmethod('retrieve_all', '$type', \@_);}";
654 my( $self, $arg, $options ) = @_;
656 my @arg = ( ref($arg) eq 'ARRAY' ) ? @$arg : ($arg);
657 my $method = $self->app.'.json_query.atomic';
658 $method =~ s/\.atomic$//o if $self->substream($$options{substream} || 0);
664 $obj = $self->request($method, @arg);
665 } catch Error with { $err = shift; };
669 OpenILS::Event->new( 'DATABASE_QUERY_FAILED',
670 payload => $arg, debug => "$err" ));
674 $self->log(I, "json_query : returned ".scalar(@$obj). " result(s)");