Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / workers / glue.py @ b5636704

History | View | Annotate | Download (3.7 kB)

1
# -*- coding: utf-8 -
2
#
3
# Copyright 2013 GRNET S.A. All rights reserved.
4
#
5
# Redistribution and use in source and binary forms, with or
6
# without modification, are permitted provided that the following
7
# conditions are met:
8
#
9
#   1. Redistributions of source code must retain the above
10
#      copyright notice, this list of conditions and the following
11
#      disclaimer.
12
#
13
#   2. Redistributions in binary form must reproduce the above
14
#      copyright notice, this list of conditions and the following
15
#      disclaimer in the documentation and/or other materials
16
#      provided with the distribution.
17
#
18
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
19
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
22
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
25
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29
# POSSIBILITY OF SUCH DAMAGE.
30
#
31
# The views and conclusions contained in the software and
32
# documentation are those of the authors and should not be
33
# interpreted as representing official policies, either expressed
34
# or implied, of GRNET S.A.
35

    
36
import re
37

    
38

    
39
class WorkerGlue(object):
40

    
41
    pmap = {}
42
    worker_id = None
43
    ioctx_pool = None
44

    
45
    @classmethod
46
    def setmap(cls, pid, index):
47
        WorkerGlue.pmap[pid] = index
48
        WorkerGlue.worker_id = index
49

    
50
    @classmethod
51
    def setupXsegPool(cls, ObjectPool, Segment, Xseg_ctx, cfile, pool_size=8):
52
        worker_id = WorkerGlue.worker_id
53
        ARCHIPELAGO_CONF_FILE = cfile
54
        ARCHIPELAGO_SEGMENT_TYPE = 'segdev'
55
        ARCHIPELAGO_SEGMENT_NAME = 'xsegbd'
56
        cfg = {}
57
        bcfg = open(ARCHIPELAGO_CONF_FILE).read()
58
        cfg['SEGMENT_PORTS'] = re.search('SEGMENT_PORTS\s*=\s*\d+',
59
                                         bcfg).group(0).split('=')[1]
60
        cfg['SEGMENT_SIZE'] = re.search('SEGMENT_SIZE\s*=\s*\d+',
61
                                        bcfg).group(0).split('=')[1]
62
        ARCHIPELAGO_SEGMENT_PORTS = int(cfg['SEGMENT_PORTS'])
63
        ARCHIPELAGO_SEGMENT_SIZE = int(cfg['SEGMENT_SIZE'])
64
        ARCHIPELAGO_SEGMENT_ALIGNMENT = 12
65

    
66
        class XsegPool(ObjectPool):
67

    
68
            def __init__(self):
69
                super(XsegPool, self).__init__(size=pool_size)
70
                self.segment = Segment(ARCHIPELAGO_SEGMENT_TYPE,
71
                                       ARCHIPELAGO_SEGMENT_NAME,
72
                                       ARCHIPELAGO_SEGMENT_PORTS,
73
                                       ARCHIPELAGO_SEGMENT_SIZE,
74
                                       ARCHIPELAGO_SEGMENT_ALIGNMENT)
75
                self.worker_id = worker_id
76
                self.cnt = 1
77

    
78
            def _pool_create(self):
79
                if self.worker_id == 1:
80
                    ioctx = Xseg_ctx(self.segment, self.worker_id + self.cnt)
81
                    self.cnt += 1
82
                    return ioctx
83
                else:
84
                    ioctx = Xseg_ctx(self.segment,
85
                                     (self.worker_id - 1) * pool_size + 2 +
86
                                     self.cnt)
87
                    self.cnt += 1
88
                    return ioctx
89

    
90
            def _pool_verify(self, poolobj):
91
                return True
92

    
93
            def _pool_cleanup(self, poolobj):
94
                return False
95

    
96
        WorkerGlue.ioctx_pool = XsegPool()