Newer
Older
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
#!/usr/bin/env python3
import sys
import logging
sys.path.append("lib")
from ops.charm import CharmBase
from ops.framework import StoredState
from ops.main import main
from ops.model import (
ActiveStatus,
MaintenanceStatus,
)
logger = logging.getLogger(__name__)
class NginxK8sCharm(CharmBase):
state = StoredState()
def __init__(self, framework, key):
super().__init__(framework, key)
self.state.set_default(spec=None)
# Observe Charm related events
self.framework.observe(self.on.config_changed, self.on_config_changed)
self.framework.observe(self.on.start, self.on_start)
self.framework.observe(self.on.upgrade_charm, self.on_upgrade_charm)
def _apply_spec(self):
# Only apply the spec if this unit is a leader.
if not self.framework.model.unit.is_leader():
return
new_spec = self.make_pod_spec()
if new_spec == self.state.spec:
return
self.framework.model.pod.set_spec(new_spec)
self.state.spec = new_spec
def make_pod_spec(self):
config = self.framework.model.config
ports = [
{
"name": "port",
"containerPort": config["port"],
"protocol": "TCP",
}
]
spec = {
"version": 2,
"containers": [
{
"name": self.framework.model.app.name,
"image": "{}".format(config["image"]),
"ports": ports,
}
],
}
return spec
def on_config_changed(self, event):
"""Handle changes in configuration"""
unit = self.model.unit
unit.status = MaintenanceStatus("Applying new pod spec")
self._apply_spec()
unit.status = ActiveStatus("Ready")
def on_start(self, event):
"""Called when the charm is being installed"""
unit = self.model.unit
unit.status = MaintenanceStatus("Applying pod spec")
self._apply_spec()
unit.status = ActiveStatus("Ready")
def on_upgrade_charm(self, event):
"""Upgrade the charm."""
unit = self.model.unit
unit.status = MaintenanceStatus("Upgrading charm")
self.on_start(event)
if __name__ == "__main__":
main(NginxK8sCharm)