← Back to branch summary

~james-page/charms/trusty/rabbitmq-server/network-splits

« back to all changes in this revision

Viewing changes to hooks/charmhelpers/contrib/openstack/ip.py

  • Committer: james.page at ubuntu
  • Date: 2015-01-23 08:23:05 UTC
  • mfrom: (52.2.27 rabbitmq-server)
  • Revision ID: james.page@ubuntu.com-20150123082305-5uf1uk14iov78hl2
Rebase on next branch

Show diffs side-by-side

added added

removed removed

Lines of Context:
hooks/charmhelpers/contrib/openstack/ip.py
 
1
from charmhelpers.core.hookenv import (
 
2
    config,
 
3
    unit_get,
 
4
)
 
5
from charmhelpers.contrib.network.ip import (
 
6
    get_address_in_network,
 
7
    is_address_in_network,
 
8
    is_ipv6,
 
9
    get_ipv6_addr,
 
10
)
 
11
from charmhelpers.contrib.hahelpers.cluster import is_clustered
 
12
 
 
13
PUBLIC = 'public'
 
14
INTERNAL = 'int'
 
15
ADMIN = 'admin'
 
16
 
 
17
ADDRESS_MAP = {
 
18
    PUBLIC: {
 
19
        'config': 'os-public-network',
 
20
        'fallback': 'public-address'
 
21
    },
 
22
    INTERNAL: {
 
23
        'config': 'os-internal-network',
 
24
        'fallback': 'private-address'
 
25
    },
 
26
    ADMIN: {
 
27
        'config': 'os-admin-network',
 
28
        'fallback': 'private-address'
 
29
    }
 
30
}
 
31
 
 
32
 
 
33
def canonical_url(configs, endpoint_type=PUBLIC):
 
34
    """Returns the correct HTTP URL to this host given the state of HTTPS
 
35
    configuration, hacluster and charm configuration.
 
36
 
 
37
    :param configs: OSTemplateRenderer config templating object to inspect
 
38
                    for a complete https context.
 
39
    :param endpoint_type: str endpoint type to resolve.
 
40
    :param returns: str base URL for services on the current service unit.
 
41
    """
 
42
    scheme = 'http'
 
43
    if 'https' in configs.complete_contexts():
 
44
        scheme = 'https'
 
45
    address = resolve_address(endpoint_type)
 
46
    if is_ipv6(address):
 
47
        address = "[{}]".format(address)
 
48
    return '%s://%s' % (scheme, address)
 
49
 
 
50
 
 
51
def resolve_address(endpoint_type=PUBLIC):
 
52
    """Return unit address depending on net config.
 
53
 
 
54
    If unit is clustered with vip(s) and has net splits defined, return vip on
 
55
    correct network. If clustered with no nets defined, return primary vip.
 
56
 
 
57
    If not clustered, return unit address ensuring address is on configured net
 
58
    split if one is configured.
 
59
 
 
60
    :param endpoint_type: Network endpoing type
 
61
    """
 
62
    resolved_address = None
 
63
    vips = config('vip')
 
64
    if vips:
 
65
        vips = vips.split()
 
66
 
 
67
    net_type = ADDRESS_MAP[endpoint_type]['config']
 
68
    net_addr = config(net_type)
 
69
    net_fallback = ADDRESS_MAP[endpoint_type]['fallback']
 
70
    clustered = is_clustered()
 
71
    if clustered:
 
72
        if not net_addr:
 
73
            # If no net-splits defined, we expect a single vip
 
74
            resolved_address = vips[0]
 
75
        else:
 
76
            for vip in vips:
 
77
                if is_address_in_network(net_addr, vip):
 
78
                    resolved_address = vip
 
79
                    break
 
80
    else:
 
81
        if config('prefer-ipv6'):
 
82
            fallback_addr = get_ipv6_addr(exc_list=vips)[0]
 
83
        else:
 
84
            fallback_addr = unit_get(net_fallback)
 
85
 
 
86
        resolved_address = get_address_in_network(net_addr, fallback_addr)
 
87
 
 
88
    if resolved_address is None:
 
89
        raise ValueError("Unable to resolve a suitable IP address based on "
 
90
                         "charm state and configuration. (net_type=%s, "
 
91
                         "clustered=%s)" % (net_type, clustered))
 
92
 
 
93
    return resolved_address