Check Size of Object in memcached

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import memcache

try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO

try:
    import cPickle as pickle
except ImportError:
    import pickle

def data_size(data):
    """Returns the size in bytes of the data passed in."""
    fp = StringIO()
    pickler = pickle.Pickler(fp, protocol=0)
    pickler.dump(data)
    val = fp.getvalue()
    return len(val)

def data_too_large(data):
    """Returns True if the data passed in is too large for memcached."""
    return data_size(data) >= memcache.SERVER_MAX_VALUE_LENGTH

Comments

(Forgotten your password?)

You may use Markdown syntax here, but raw HTML will be removed.