Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / workers / glue.py @ 1c6f4d42

History | View | Annotate | Download (4.1 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 ConfigParser
37

    
38

    
39
class WorkerGlue(object):
40

    
41
    pmap = {}
42
    worker_id = None
43
    ioctx_pool = None
44
    ArchipelagoConfFile = '/etc/archipelago/archipelago.conf'
45

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

    
51
    @classmethod
52
    def setupXsegPool(cls, ObjectPool, Segment, Xseg_ctx,
53
                      cfile=ArchipelagoConfFile, pool_size=8):
54
        worker_id = WorkerGlue.worker_id
55
        WorkerGlue.ArchipelagoConfFile = ARCHIPELAGO_CONF_FILE = cfile
56
        ARCHIPELAGO_SEGMENT_TYPE = 'segdev'
57
        ARCHIPELAGO_SEGMENT_NAME = 'xsegbd'
58
        cfg = {}
59
        bcfg = ConfigParser.ConfigParser()
60
        bcfg.readfp(open(ARCHIPELAGO_CONF_FILE))
61
        cfg['SEGMENT_PORTS'] = bcfg.getint('XSEG', 'SEGMENT_PORTS')
62
        cfg['SEGMENT_DYNPORTS'] = bcfg.getint('XSEG', 'SEGMENT_DYNPORTS')
63
        cfg['SEGMENT_SIZE'] = bcfg.getint('XSEG', 'SEGMENT_SIZE')
64
        ARCHIPELAGO_SEGMENT_DYNPORTS = int(cfg['SEGMENT_DYNPORTS'])
65
        ARCHIPELAGO_SEGMENT_PORTS = int(cfg['SEGMENT_PORTS'])
66
        ARCHIPELAGO_SEGMENT_SIZE = int(cfg['SEGMENT_SIZE'])
67
        ARCHIPELAGO_SEGMENT_ALIGNMENT = 12
68

    
69
        class XsegPool(ObjectPool):
70

    
71
            def __init__(self):
72
                super(XsegPool, self).__init__(size=pool_size)
73
                self.segment = Segment(ARCHIPELAGO_SEGMENT_TYPE,
74
                                       ARCHIPELAGO_SEGMENT_NAME,
75
                                       ARCHIPELAGO_SEGMENT_DYNPORTS,
76
                                       ARCHIPELAGO_SEGMENT_PORTS,
77
                                       ARCHIPELAGO_SEGMENT_SIZE,
78
                                       ARCHIPELAGO_SEGMENT_ALIGNMENT)
79
                self.worker_id = worker_id
80
                self.cnt = 1
81

    
82
            def _pool_create(self):
83
                if self.worker_id == 1:
84
                    ioctx = Xseg_ctx(self.segment, self.worker_id + self.cnt)
85
                    self.cnt += 1
86
                    return ioctx
87
                elif self.worker_id > 1:
88
                    ioctx = Xseg_ctx(self.segment,
89
                                     (self.worker_id - 1) * pool_size + 2 +
90
                                     self.cnt)
91
                    self.cnt += 1
92
                    return ioctx
93
                elif self.worker_id is None:
94
                    ioctx = Xseg_ctx(self.segment, 200)
95
                    return ioctx
96

    
97
            def _pool_verify(self, poolobj):
98
                return True
99

    
100
            def _pool_cleanup(self, poolobj):
101
                return False
102

    
103
        WorkerGlue.ioctx_pool = XsegPool()