1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
# Copyright 2015 Canonical Ltd.
#
# This file is part of the Cassandra Charm for Juju.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranties of
# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
# PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from charmhelpers.core import hookenv
from charmhelpers.core import services
import actions
import helpers
import relations
def get_service_definitions():
# This looks like it could be a module level global list, but
# unfortunately that makes the module unimportable outside of a
# hook context. The main culprit is RelationContext, which invokes
# relation-get from its constructor. By wrapping the service
# definition list in this function, we can defer constructing it
# until we have constructed enough of a mock context and perform
# basic tests.
config = hookenv.config()
return [
# Prepare for the Cassandra service.
dict(service='install',
data_ready=[actions.set_proxy,
actions.preinstall,
actions.emit_meminfo,
actions.revert_unchangeable_config,
actions.store_unit_private_ip,
actions.add_implicit_package_signing_keys,
actions.configure_sources,
actions.swapoff,
actions.reset_sysctl,
actions.reset_limits,
actions.install_oracle_jre,
actions.install_cassandra_packages,
actions.emit_java_version,
actions.ensure_cassandra_package_status],
start=[], stop=[]),
# Get Cassandra running.
dict(service=helpers.get_cassandra_service(),
# Open access to client and replication ports. Client
# protocols require password authentication. Access to
# the unauthenticated replication ports is protected via
# ufw firewall rules. We do not open the JMX port, although
# we could since it is similarly protected by ufw.
ports=[config['rpc_port'], # Thrift clients
config['native_transport_port'], # Native clients.
config['storage_port'], # Plaintext replication
config['ssl_storage_port']], # Encrypted replication.
required_data=[relations.StorageRelation(),
relations.PeerRelation()],
provided_data=[relations.StorageRelation()],
data_ready=[actions.configure_firewall,
actions.update_etc_hosts,
actions.maintain_seeds,
actions.configure_cassandra_yaml,
actions.configure_cassandra_env,
actions.configure_cassandra_rackdc,
actions.reset_all_io_schedulers,
actions.maybe_restart,
actions.request_unit_superuser,
actions.reset_default_password],
start=[services.open_ports],
stop=[actions.stop_cassandra, services.close_ports]),
# Actions that must be done while Cassandra is running.
dict(service='post',
required_data=[RequiresLiveNode()],
data_ready=[actions.post_bootstrap,
actions.create_unit_superusers,
actions.reset_auth_keyspace_replication,
actions.publish_database_relations,
actions.publish_database_admin_relations,
actions.install_maintenance_crontab,
actions.nrpe_external_master_relation,
actions.emit_cluster_info,
actions.set_active],
start=[], stop=[])]
class RequiresLiveNode:
def __bool__(self):
is_live = self.is_live()
hookenv.log('Requirement RequiresLiveNode: {}'.format(is_live),
hookenv.DEBUG)
return is_live
def is_live(self):
if helpers.is_decommissioned():
hookenv.log('Node is decommissioned')
return False
if helpers.is_cassandra_running():
hookenv.log('Cassandra is running')
auth = hookenv.config()['authenticator']
if auth == 'AllowAllAuthenticator':
return True
elif hookenv.local_unit() in helpers.get_unit_superusers():
hookenv.log('Credentials created')
return True
else:
hookenv.log('Credentials have not been created')
return False
else:
hookenv.log('Cassandra is not running')
return False
def get_service_manager():
return services.ServiceManager(get_service_definitions())
|
