Bug 12508: adding an error message if a contract cannot be removed
[koha_fer] / Koha / Cache.pm
index 740a133..1638324 100644 (file)
@@ -1,6 +1,7 @@
 package Koha::Cache;
 
 # Copyright 2009 Chris Cormack and The Koha Dev Team
+# Parts copyright 2012-2013 C & P Bibliography Services
 #
 # This file is part of Koha.
 #
@@ -23,117 +24,452 @@ Koha::Cache - Handling caching of html and Objects for Koha
 
 =head1 SYNOPSIS
 
-  use Koha::Cache (cache_type => $cache_type, %params );
+  use Koha::Cache;
+  my $cache = Koha::Cache->new({cache_type => $cache_type, %params});
 
 =head1 DESCRIPTION
 
-Base class for Koha::Cache::X. Subclasses must provide the following methods
-
-B<_cache_handle ($params_hr)> - cache handle creator
-
-Subclasses may override the following methods if they are not using a
-CHI-derived cache
-
-B<set_in_cache ($key, $value, $expiry)>
-
-B<get_from_cache ($key)>
-
-B<clear_from_cache ($key)>
-
-B<flush_all ()>
+Koha caching routines. This class provides two interfaces for cache access.
+The first, traditional OO interface provides the following functions:
 
 =head1 FUNCTIONS
 
 =cut
-
 use strict;
 use warnings;
 use Carp;
 use Module::Load::Conditional qw(can_load);
-use Module::Load;
+use Koha::Cache::Object;
 
-my $have_chi = 0;
+use base qw(Class::Accessor);
 
-BEGIN: {
-    if ( can_load( modules => { CHI => undef } ) ) {
-        $have_chi = 1;
-    }
+__PACKAGE__->mk_ro_accessors(
+    qw( cache memcached_cache fastmmap_cache memory_cache ));
+
+=head2 get_instance
+
+    my $cache = Koha::Cache->get_instance();
+
+This gets a shared instance of the cache, set up in a very default way. This is
+the recommended way to fetch a cache object. If possible, it'll be
+persistent across multiple instances.
+
+=cut
+
+our $singleton_cache;
+sub get_instance {
+    my ($class) = @_;
+    $singleton_cache = $class->new() unless $singleton_cache;
+    return $singleton_cache;
 }
 
-use base qw(Class::Accessor);
+=head2 new
+
+Create a new Koha::Cache object. This is required for all cache-related functionality.
 
-__PACKAGE__->mk_ro_accessors(qw( cache ));
+=cut
 
 sub new {
-    my $class = shift;
-    my $param = shift;
-    my $cache_type =
-         $ENV{CACHING_SYSTEM}
-      || $param->{cache_type}
+    my ( $class, $self ) = @_;
+    $self->{'default_type'} =
+         $self->{cache_type}
+      || $ENV{CACHING_SYSTEM}
       || 'memcached';
-    my $subclass = __PACKAGE__ . "::" . ucfirst($cache_type);
-    $param->{have_chi} = $have_chi;
-    unless ( can_load( modules => { $subclass => undef } ) ) {
-        $subclass = __PACKAGE__ . "::" . ucfirst('Null');
-        load $subclass;
+
+    $ENV{DEBUG} && carp "Default caching system: $self->{'default_type'}";
+
+    $self->{'timeout'}   ||= 0;
+    $self->{'namespace'} ||= $ENV{MEMCACHED_NAMESPACE} || 'koha';
+
+    if ( can_load( modules => { 'Cache::Memcached::Fast' => undef } ) ) {
+        _initialize_memcached($self);
+        if ( $self->{'default_type'} eq 'memcached'
+            && defined( $self->{'memcached_cache'} ) )
+        {
+            $self->{'cache'} = $self->{'memcached_cache'};
+        }
+    }
+
+    if ( can_load( modules => { 'Cache::FastMmap' => undef } ) ) {
+        _initialize_fastmmap($self);
+        if ( $self->{'default_type'} eq 'fastmmap'
+            && defined( $self->{'fastmmap_cache'} ) )
+        {
+            $self->{'cache'} = $self->{'fastmmap_cache'};
+        }
     }
-    my $cache = $subclass->_cache_handle($param);
+
+    if ( can_load( modules => { 'Cache::Memory' => undef } ) ) {
+        _initialize_memory($self);
+        if ( $self->{'default_type'} eq 'memory'
+            && defined( $self->{'memory_cache'} ) )
+        {
+            $self->{'cache'} = $self->{'memory_cache'};
+        }
+    }
+
+    # Unless a default has already been picked, we go through in best-to-
+    # least-best order, looking for something we can use. fastmmap_cache
+    # is excluded because it doesn't support expiry in a useful way.
+    unless ( defined( $self->{'cache'} ) ) {
+        foreach my $cachemember (qw(memcached_cache memory_cache )) {
+            if ( defined( $self->{$cachemember} ) ) {
+                $self->{'cache'} = $self->{$cachemember};
+                last;
+            }
+        }
+    }
+
+    $ENV{DEBUG} && carp "Selected caching system: " . ($self->{'cache'} // 'none');
+
     return
-      bless $class->SUPER::new( { cache => $cache, have_chi => $have_chi } ),
-      $subclass;
+      bless $self,
+      $class;
+}
+
+sub _initialize_memcached {
+    my ($self) = @_;
+    my @servers =
+      split /,/, $self->{'cache_servers'}
+      ? $self->{'cache_servers'}
+      : ($ENV{MEMCACHED_SERVERS} || '');
+    return if !@servers;
+
+    $ENV{DEBUG}
+      && carp "Memcached server settings: "
+      . join( ', ', @servers )
+      . " with "
+      . $self->{'namespace'};
+    # Cache::Memcached::Fast doesn't allow a default expire time to be set
+    # so we force it on setting.
+    my $memcached = Cache::Memcached::Fast->new(
+        {
+            servers            => \@servers,
+            compress_threshold => 10_000,
+            namespace          => $self->{'namespace'},
+            utf8               => 1,
+        }
+    );
+    # Ensure we can actually talk to the memcached server
+    my $ismemcached = $memcached->set('ismemcached','1');
+    return $self unless $ismemcached;
+    $self->{'memcached_cache'} = $memcached;
+    return $self;
+}
+
+sub _initialize_fastmmap {
+    my ($self) = @_;
+
+    $self->{'fastmmap_cache'} = Cache::FastMmap->new(
+        'share_file'  => "/tmp/sharefile-koha-$self->{'namespace'}",
+        'expire_time' => $self->{'timeout'},
+        'unlink_on_exit' => 0,
+    );
+    return $self;
 }
 
+sub _initialize_memory {
+    my ($self) = @_;
+
+    # Default cache time for memory is _always_ short unless it's specially
+    # defined, to allow it to work reliably in a persistent environment.
+    my $cache = Cache::Memory->new(
+        'namespace'       => $self->{'namespace'},
+        'default_expires' => "$self->{'timeout'} sec" || "10 sec",
+    );
+    $self->{'memory_cache'} = $cache;
+    # Memory cache can't handle complex types for some reason, so we use its
+    # freeze and thaw functions.
+    $self->{ref($cache) . '_set'} = sub {
+        my ($key, $val, $exp) = @_;
+        # Refer to set_expiry in Cache::Entry for why we do this 'sec' thing.
+        $exp = "$exp sec" if defined $exp;
+        # Because we need to use freeze, it must be a reference type.
+        $cache->freeze($key, [$val], $exp);
+    };
+    $self->{ref($cache) . '_get'} = sub {
+        my $res = $cache->thaw(shift);
+        return unless defined $res;
+        return $res->[0];
+    };
+    return $self;
+}
+
+=head2 is_cache_active
+
+Routine that checks whether or not a default caching method is active on this
+object.
+
+=cut
+
 sub is_cache_active {
-    return $ENV{CACHING_SYSTEM} ? '1' : '';
+    my $self = shift;
+    return $self->{'cache'} ? 1 : 0;
 }
 
+=head2 set_in_cache
+
+    $cache->set_in_cache($key, $value, [$options]);
+
+Save a value to the specified key in the cache. A hashref of options may be
+specified.
+
+The possible options are:
+
+=over
+
+=item expiry
+
+Expiry time of this cached entry in seconds.
+
+=item deepcopy
+
+If set, this will perform a deep copy of the item when it's retrieved. This
+means that it'll be safe if something later modifies the result of the
+function. Will be ignored in situations where the same behaviour comes from
+the caching layer anyway.
+
+=item cache
+
+The cache object to use if you want to provide your own. It should be an
+instance of C<Cache::*> and follow the same interface as L<Cache::Memcache>.
+
+=back
+
+=cut
+
 sub set_in_cache {
-    my ( $self, $key, $value, $expiry ) = @_;
-    croak "No key" unless $key;
-    $ENV{DEBUG} && warn "set_in_cache for $key";
+    my ( $self, $key, $value, $options, $_cache) = @_;
+    # This is a bit of a hack to support the old API in case things still use it
+    if (defined $options && (ref($options) ne 'HASH')) {
+        my $new_options;
+        $new_options->{expiry} = $options;
+        $new_options->{cache} = $_cache if defined $_cache;
+        $options = $new_options;
+    }
 
-    return unless $self->{cache};
-    return unless $self->{have_chi};
+    # the key mustn't contain whitespace (or control characters) for memcache
+    # but shouldn't be any harm in applying it globally.
+    $key =~ s/[\x00-\x20]/_/g;
 
-    if ( defined $expiry ) {
-        return $self->{cache}->set( $key, $value, $expiry );
+    my $cache = $options->{cache} || 'cache';
+    croak "No key" unless $key;
+    $ENV{DEBUG} && carp "set_in_cache for $key";
+
+    return unless ( $self->{$cache} && ref( $self->{$cache} ) =~ m/^Cache::/ );
+    my $expiry = $options->{expiry};
+    $expiry //= $self->{timeout};
+    my $set_sub = $self->{ref($self->{$cache}) . "_set"};
+    # We consider an expiry of 0 to be inifinite
+    if ( $expiry ) {
+        return $set_sub
+          ? $set_sub->( $key, $value, $expiry )
+          : $self->{$cache}->set( $key, $value, $expiry );
     }
     else {
-        return $self->{cache}->set( $key, $value );
+        return $set_sub
+          ? $set_sub->( $key, $value )
+          : $self->{$cache}->set( $key, $value );
     }
 }
 
+=head2 get_from_cache
+
+    my $value = $cache->get_from_cache($key);
+
+Retrieve the value stored under the specified key in the default cache.
+
+=cut
+
 sub get_from_cache {
-    my ( $self, $key ) = @_;
+    my ( $self, $key, $cache ) = @_;
+    $key =~ s/[\x00-\x20]/_/g;
+    $cache ||= 'cache';
     croak "No key" unless $key;
-    $ENV{DEBUG} && warn "get_from_cache for $key";
-    return unless $self->{cache};
-    return unless $self->{have_chi};
-    return $self->{cache}->get($key);
+    $ENV{DEBUG} && carp "get_from_cache for $key";
+    return unless ( $self->{$cache} && ref( $self->{$cache} ) =~ m/^Cache::/ );
+    my $get_sub = $self->{ref($self->{$cache}) . "_get"};
+    return $get_sub ? $get_sub->($key) : $self->{$cache}->get($key);
 }
 
+=head2 clear_from_cache
+
+    $cache->clear_from_cache($key);
+
+Remove the value identified by the specified key from the default cache.
+
+=cut
+
 sub clear_from_cache {
-    my ( $self, $key ) = @_;
+    my ( $self, $key, $cache ) = @_;
+    $key =~ s/[\x00-\x20]/_/g;
+    $cache ||= 'cache';
     croak "No key" unless $key;
-    return unless $self->{cache};
-    return unless $self->{have_chi};
-    return $self->{cache}->remove($key);
+    return unless ( $self->{$cache} && ref( $self->{$cache} ) =~ m/^Cache::/ );
+    return $self->{$cache}->delete($key)
+      if ( ref( $self->{$cache} ) =~ m'^Cache::Memcached' );
+    return $self->{$cache}->remove($key);
 }
 
+=head2 flush_all
+
+    $cache->flush_all();
+
+Clear the entire default cache.
+
+=cut
+
 sub flush_all {
-    my $self = shift;
-    return unless $self->{cache};
-    return unless $self->{have_chi};
-    return $self->{cache}->clear();
+    my ( $self, $cache ) = shift;
+    $cache ||= 'cache';
+    return unless ( $self->{$cache} && ref( $self->{$cache} ) =~ m/^Cache::/ );
+    return $self->{$cache}->flush_all()
+      if ( ref( $self->{$cache} ) =~ m'^Cache::Memcached' );
+    return $self->{$cache}->clear();
+}
+
+=head1 TIED INTERFACE
+
+Koha::Cache also provides a tied interface which enables users to provide a
+constructor closure and (after creation) treat cached data like normal reference
+variables and rely on the cache Just Working and getting updated when it
+expires, etc.
+
+    my $cache = Koha::Cache->new();
+    my $data = 'whatever';
+    my $scalar = Koha::Cache->create_scalar(
+        {
+            'key'         => 'whatever',
+            'timeout'     => 2,
+            'constructor' => sub { return $data; },
+        }
+    );
+    print "$$scalar\n"; # Prints "whatever"
+    $data = 'somethingelse';
+    print "$$scalar\n"; # Prints "whatever" because it is cached
+    sleep 2; # Wait until the cache entry has expired
+    print "$$scalar\n"; # Prints "somethingelse"
+
+    my $hash = Koha::Cache->create_hash(
+        {
+            'key'         => 'whatever',
+            'timeout'     => 2,
+            'constructor' => sub { return $data; },
+        }
+    );
+    print "$$variable\n"; # Prints "whatever"
+
+The gotcha with this interface, of course, is that the variable returned by
+create_scalar and create_hash is a I<reference> to a tied variable and not a
+tied variable itself.
+
+The tied variable is configured by means of a hashref passed in to the
+create_scalar and create_hash methods. The following parameters are supported:
+
+=over
+
+=item I<key>
+
+Required. The key to use for identifying the variable in the cache.
+
+=item I<constructor>
+
+Required. A closure (or reference to a function) that will return the value that
+needs to be stored in the cache.
+
+=item I<preload>
+
+Optional. A closure (or reference to a function) that gets run to initialize
+the cache when creating the tied variable.
+
+=item I<arguments>
+
+Optional. Array reference with the arguments that should be passed to the
+constructor function.
+
+=item I<timeout>
+
+Optional. The cache timeout in seconds for the variable. Defaults to 600
+(ten minutes).
+
+=item I<cache_type>
+
+Optional. Which type of cache to use for the variable. Defaults to whatever is
+set in the environment variable CACHING_SYSTEM. If set to 'null', disables
+caching for the tied variable.
+
+=item I<allowupdate>
+
+Optional. Boolean flag to allow the variable to be updated directly. When this
+is set and the variable is used as an l-value, the cache will be updated
+immediately with the new value. Using this is probably a bad idea on a
+multi-threaded system. When I<allowupdate> is not set to true, using the
+tied variable as an l-value will have no effect.
+
+=item I<destructor>
+
+Optional. A closure (or reference to a function) that should be called when the
+tied variable is destroyed.
+
+=item I<unset>
+
+Optional. Boolean flag to tell the object to remove the variable from the cache
+when it is destroyed or goes out of scope.
+
+=item I<inprocess>
+
+Optional. Boolean flag to tell the object not to refresh the variable from the
+cache every time the value is desired, but rather only when the I<local> copy
+of the variable is older than the timeout.
+
+=back
+
+=head2 create_scalar
+
+    my $scalar = Koha::Cache->create_scalar(\%params);
+
+Create scalar tied to the cache.
+
+=cut
+
+sub create_scalar {
+    my ( $self, $args ) = @_;
+
+    $self->_set_tied_defaults($args);
+
+    tie my $scalar, 'Koha::Cache::Object', $args;
+    return \$scalar;
+}
+
+sub create_hash {
+    my ( $self, $args ) = @_;
+
+    $self->_set_tied_defaults($args);
+
+    tie my %hash, 'Koha::Cache::Object', $args;
+    return \%hash;
+}
+
+sub _set_tied_defaults {
+    my ( $self, $args ) = @_;
+
+    $args->{'timeout'}   = '600' unless defined( $args->{'timeout'} );
+    $args->{'inprocess'} = '0'   unless defined( $args->{'inprocess'} );
+    unless ( $args->{cache_type} and lc( $args->{cache_type} ) eq 'null' ) {
+        $args->{'cache'} = $self;
+        $args->{'cache_type'} ||= $ENV{'CACHING_SYSTEM'};
+    }
+
+    return $args;
 }
 
-=head2 EXPORT
+=head1 EXPORT
 
 None by default.
 
 =head1 SEE ALSO
 
-Koha::Cache::Memcached
+Koha::Cache::Object
 
 =head1 AUTHOR