← Back to branch summary

~canonical-ci-engineering/uci-engine/trunk

« back to all changes in this revision

Viewing changes to charms/precise/wsgi-app/hooks/intercom.py

  • Committer: Ubuntu CI Bot
  • Author(s): Celso Providelo
  • Date: 2015-01-07 14:33:28 UTC
  • mfrom: (911.1.8 charms-cleanup)
  • Revision ID: ubuntu_ci_bot-20150107143328-q0d037ojtopm3n9d
Removing the charms from UCI-E since they live in isolated branches now and also adjusting the run-tests to stop loading/running charm tests.

Updating local j-d copy to trunk (r126) for proper support for charm branch pinning. [r=Evan Dandrea]

Show diffs side-by-side

added added

removed removed

Lines of Context:
charms/precise/wsgi-app/hooks/intercom.py
1
 
#!/usr/bin/env python
2
 
# Ubuntu CI Engine
3
 
# Copyright 2014 Canonical Ltd.
4
 
 
5
 
# This program is free software: you can redistribute it and/or modify it
6
 
# under the terms of the GNU Affero General Public License version 3, as
7
 
# published by the Free Software Foundation.
8
 
 
9
 
# This program is distributed in the hope that it will be useful, but
10
 
# WITHOUT ANY WARRANTY; without even the implied warranties of
11
 
# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
12
 
# PURPOSE.  See the GNU Affero General Public License for more details.
13
 
 
14
 
# You should have received a copy of the GNU Affero General Public License
15
 
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
16
 
import json
17
 
import os
18
 
import sys
19
 
 
20
 
from charmhelpers.core import hookenv
21
 
 
22
 
hooks = hookenv.Hooks()
23
 
log = hookenv.log
24
 
 
25
 
 
26
 
def get_service_dir():
27
 
    unit_name = hookenv.local_unit()
28
 
    sanitized_unit_name = unit_name.split('/')[0]
29
 
    return os.path.join('/srv', sanitized_unit_name)
30
 
 
31
 
 
32
 
@hooks.hook('intercom-relation-changed')
33
 
@hooks.hook('intercom-relation-joined')
34
 
def update_intercom(remove=False):
35
 
    """Update the inter-components map.
36
 
 
37
 
    It maintains a map of <unit> -> (<hostname>, <port>) components
38
 
    that may exchange data directly.
39
 
 
40
 
    It consumes 'http' (hostname:port) interface.
41
 
 
42
 
    The inter-components map is saved as 'intercom.json' in the
43
 
    application DIR in JSON format.
44
 
    """
45
 
    intercom_path = os.path.join(get_service_dir(), 'intercom.json')
46
 
 
47
 
    if os.path.exists(intercom_path):
48
 
        registry = json.load(open(intercom_path))
49
 
    else:
50
 
        registry = {}
51
 
 
52
 
    relation_id = hookenv.relation_id()
53
 
    unit = hookenv.remote_unit()
54
 
    hostname = hookenv.relation_get('hostname')
55
 
    port = hookenv.relation_get('port')
56
 
 
57
 
    if remove:
58
 
        log('Intercom unit \'{}\' ({}) departed.'.format(unit, hostname))
59
 
        if registry.get(relation_id) is not None:
60
 
            del registry[relation_id]
61
 
    else:
62
 
        log('Intercom unit \'{}\' ({}) joined.'.format(unit, hostname))
63
 
        registry[relation_id] = {
64
 
            'unit': unit,
65
 
            'hostname': hostname,
66
 
            'port': port,
67
 
        }
68
 
 
69
 
    with open(intercom_path, 'w') as fd:
70
 
        json.dump(registry, fd)
71
 
 
72
 
 
73
 
@hooks.hook()
74
 
def intercom_relation_broken():
75
 
    update_intercom(remove=True)
76
 
 
77
 
 
78
 
if __name__ == "__main__":
79
 
    hooks.execute(sys.argv)