OpenSRF/src/libstack/osrf_cache.c

   1 /*
   2 Copyright (C) 2005  Georgia Public Library Service
   3 Bill Erickson <highfalutin@gmail.com>
   4
   5 This program is free software; you can redistribute it and/or
   6 modify it under the terms of the GNU General Public License
   7 as published by the Free Software Foundation; either version 2
   8 of the License, or (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14 */
  15
  16 #include "osrf_cache.h"
  17
  18 struct memcache* __osrfCache = NULL;
  19 time_t __osrfCacheMaxSeconds = -1;
  20
  21 int osrfCacheInit( char* serverStrings[], int size, time_t maxCacheSeconds ) {
  22         if( !(serverStrings && size > 0) ) return -1;
  23
  24         int i;
  25         __osrfCache = mc_new();
  26         __osrfCacheMaxSeconds = maxCacheSeconds;
  27
  28         for( i = 0; i < size && serverStrings[i]; i++ )
  29                 mc_server_add4( __osrfCache, serverStrings[i] );
  30
  31         return 0;
  32 }
  33
  34 int osrfCachePutObject( char* key, const jsonObject* obj, time_t seconds ) {
  35         if( !(key && obj) ) return -1;
  36         char* s = jsonObjectToJSON( obj );
  37         osrfLogInternal("osrfCachePut(): Putting object: %s", s);
  38         if( seconds < 0 ) seconds = __osrfCacheMaxSeconds;
  39
  40         mc_set(__osrfCache, key, strlen(key), s, strlen(s), seconds, 0);
  41         free(s);
  42         return 0;
  43 }
  44
  45 int osrfCachePutString( char* key, const char* value, time_t seconds ) {
  46         if( !(key && value) ) return -1;
  47         if( seconds < 0 ) seconds = __osrfCacheMaxSeconds;
  48         osrfLogInternal("osrfCachePutString(): Putting string: %s", value);
  49         mc_set(__osrfCache, key, strlen(key), value, strlen(value), seconds, 0);
  50         return 0;
  51 }
  52
  53 jsonObject* osrfCacheGetObject( char* key, ... ) {
  54         jsonObject* obj = NULL;
  55         if( key ) {
  56                 VA_LIST_TO_STRING(key);
  57                 char* data = (char*) mc_aget( __osrfCache, VA_BUF, strlen(VA_BUF) );
  58                 if( data ) {
  59                         osrfLogInternal("osrfCacheGetObject(): Returning object: %s", data);
  60                         obj = jsonParseString( data );
  61                         return obj;
  62                 }
  63         }
  64         return NULL;
  65 }
  66
  67 char* osrfCacheGetString( char* key, ... ) {
  68         if( key ) {
  69                 VA_LIST_TO_STRING(key);
  70                 char* data = (char*) mc_aget(__osrfCache, VA_BUF, strlen(VA_BUF) );
  71                 osrfLogInternal("osrfCacheGetObject(): Returning object: %s", data);
  72                 return data;
  73         }
  74         return NULL;
  75 }
  76
  77
  78 int osrfCacheRemove( char* key, ... ) {
  79         if( key ) {
  80                 VA_LIST_TO_STRING(key);
  81                 return mc_delete(__osrfCache, VA_BUF, strlen(VA_BUF), 0 );
  82         }
  83         return -1;
  84 }
  85
  86
  87 int osrfCacheSetExpire( time_t seconds, char* key, ... ) {
  88         if( key ) {
  89                 VA_LIST_TO_STRING(key);
  90                 jsonObject* o = osrfCacheGetObject( VA_BUF );
  91                 osrfCacheRemove(VA_BUF);
  92                 return osrfCachePutObject( VA_BUF, o, seconds );
  93         }
  94         return -1;
  95 }
  96
  97