code.grnet.gr Git - pithos/blob - pithos/backends/lib_alchemy/hashfiler/mapper.py

   1 # Copyright 2011 GRNET S.A. All rights reserved.
   2 #
   3 # Redistribution and use in source and binary forms, with or
   4 # without modification, are permitted provided that the following
   5 # conditions are met:
   6 #
   7 #   1. Redistributions of source code must retain the above
   8 #      copyright notice, this list of conditions and the following
   9 #      disclaimer.
  10 #
  11 #   2. Redistributions in binary form must reproduce the above
  12 #      copyright notice, this list of conditions and the following
  13 #      disclaimer in the documentation and/or other materials
  14 #      provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  17 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  20 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  23 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  24 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  26 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  27 # POSSIBILITY OF SUCH DAMAGE.
  28 #
  29 # The views and conclusions contained in the software and
  30 # documentation are those of the authors and should not be
  31 # interpreted as representing official policies, either expressed
  32 # or implied, of GRNET S.A.
  33
  34 from os.path import realpath, join, exists, isdir
  35 from os import makedirs, unlink
  36 from errno import ENOENT
  37
  38 from context_file import ContextFile
  39
  40
  41 class Mapper(object):
  42     """Mapper.
  43        Required contstructor parameters: mappath, namelen.
  44     """
  45
  46     mappath = None
  47     namelen = None
  48
  49     def __init__(self, **params):
  50         self.params = params
  51         self.namelen = params['namelen']
  52         mappath = realpath(params['mappath'])
  53         if not isdir(mappath):
  54             if not exists(mappath):
  55                 makedirs(mappath)
  56             else:
  57                 raise ValueError("Variable mappath '%s' is not a directory" % (mappath,))
  58         self.mappath = mappath
  59
  60     def get_rear_map(self, name, create=0):
  61         name = join(self.mappath, hex(int(name)))
  62         return ContextFile(name, create)
  63
  64     def delete_rear_map(self, name):
  65         name = join(self.mappath, hex(int(name)))
  66         try:
  67             unlink(name)
  68             return 1
  69         except OSError, e:
  70             if e.errno != ENOENT:
  71                 raise
  72         return 0
  73
  74     def map_retr(self, name, blkoff=0, nr=100000000000000):
  75         """Return as a list, part of the hashes map of an object
  76            at the given block offset.
  77            By default, return the whole hashes map.
  78         """
  79         namelen = self.namelen
  80         hashes = ()
  81
  82         with self.get_rear_map(name, 0) as rmap:
  83             if rmap:
  84                 hashes = list(rmap.sync_read_chunks(namelen, nr, blkoff))
  85         return hashes
  86
  87     def map_stor(self, name, hashes=(), blkoff=0, create=1):
  88         """Store hashes in the given hashes map, replacing the old ones."""
  89         namelen = self.namelen
  90         with self.get_rear_map(name, 1) as rmap:
  91             rmap.sync_write_chunks(namelen, blkoff, hashes, None)
  92
  93 #     def map_copy(self, src, dst):
  94 #         """Copy a hashes map to another one, replacing it."""
  95 #         with self.get_rear_map(src, 0) as rmap:
  96 #             if rmap:
  97 #                 rmap.copy_to(dst)
  98
  99     def map_remv(self, name):
 100         """Remove a hashes map. Returns true if the map was found and removed."""
 101         return self.delete_rear_map(name)
 102