Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-backend / pithos / workers / glue.py @ 809b482a

History | View | Annotate | Download (4 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
    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 = open(ARCHIPELAGO_CONF_FILE).read()
60
        cfg['SEGMENT_PORTS'] = re.search('SEGMENT_PORTS\s*=\s*\d+',
61
                                         bcfg).group(0).split('=')[1]
62
        cfg['SEGMENT_SIZE'] = re.search('SEGMENT_SIZE\s*=\s*\d+',
63
                                        bcfg).group(0).split('=')[1]
64
        ARCHIPELAGO_SEGMENT_PORTS = int(cfg['SEGMENT_PORTS'])
65
        ARCHIPELAGO_SEGMENT_SIZE = int(cfg['SEGMENT_SIZE'])
66
        ARCHIPELAGO_SEGMENT_ALIGNMENT = 12
67

    
68
        class XsegPool(ObjectPool):
69

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

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

    
95
            def _pool_verify(self, poolobj):
96
                return True
97

    
98
            def _pool_cleanup(self, poolobj):
99
                return False
100

    
101
        WorkerGlue.ioctx_pool = XsegPool()