Skip to content

Commit

Permalink
fix(gazer): Handled for empty config.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
isala404 committed Feb 15, 2022
1 parent 3361804 commit 0a0e0d0
Show file tree
Hide file tree
Showing 4 changed files with 41 additions and 38 deletions.
11 changes: 7 additions & 4 deletions gazer/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,13 @@ def __init__(self, **kargs):
self.read_config()

def read_config(self):
with open(r'config/config.yaml') as file:
self.config = yaml.load(file, Loader=yaml.FullLoader)
self.config = dict(filter(lambda elem: elem[1]['isService'] or elem[1]['node'] == NODE_NAME,
self.config.items()))
try:
with open(r'config/config.yaml') as file:
self.config = yaml.load(file, Loader=yaml.FullLoader)
self.config = dict(filter(lambda elem: elem[1]['isService'] or elem[1]['node'] == NODE_NAME,
self.config.items()))
except:
self.config = None

def process_IN_CLOSE_WRITE(self, evt):
self.read_config()
Expand Down
4 changes: 2 additions & 2 deletions gazer/console.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ def draw_menu(stdscr: curses.window):
statusbarstr = "Press 'q' to exit | STATUS BAR | Pos: {}, {}".format(cursor_x, cursor_y)

stdscr.addstr(0, 0, "Requests", curses.color_pair(1))
stdscr.addstr(2, 0, gazer.request_log_text(), curses.color_pair(1))
stdscr.addstr(15, 0, gazer.syn_backlog_text(), curses.color_pair(1))
stdscr.addstr(2, 0, "f\n"*25, curses.color_pair(1))
stdscr.addstr(28, 0, gazer.syn_backlog_text(), curses.color_pair(1))

# Render status bar
stdscr.attron(curses.color_pair(3))
Expand Down
38 changes: 19 additions & 19 deletions gazer/deployment.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,15 @@
apiVersion: v1
kind: Namespace
metadata:
name: lazy-koala
---
apiVersion: v1
kind: ConfigMap
metadata:
name: gazer-config
namespace: lazy-koala
data:
config.yaml: |
#apiVersion: v1
#kind: Namespace
#metadata:
# name: lazy-koala
#---
#apiVersion: v1
#kind: ConfigMap
#metadata:
# name: gazer-config
# namespace: lazy-koala
#data:
# config.yaml: |
---
apiVersion: apps/v1
kind: DaemonSet
Expand Down Expand Up @@ -80,7 +80,7 @@ spec:
readOnly: true
hostNetwork: true
hostPID: true
serviceAccountName: lazy-koala
serviceAccountName: gazer
initContainers:
- name: init-headers
image: ghcr.io/mrsupiri/lazy-koala/gazer:init
Expand Down Expand Up @@ -149,7 +149,7 @@ spec:
labels:
app: prometheus
spec:
serviceAccountName: lazy-koala
serviceAccountName: gazer
containers:
- name: prometheus
image: prom/prometheus:v2.33.1
Expand All @@ -168,7 +168,7 @@ spec:
apiVersion: rbac.authorization.k8s.io/v1
kind: ClusterRole
metadata:
name: lazy-koala
name: gazer
rules:
- apiGroups:
- ""
Expand Down Expand Up @@ -209,20 +209,20 @@ rules:
apiVersion: rbac.authorization.k8s.io/v1
kind: ClusterRoleBinding
metadata:
name: lazy-koala
name: gazer
roleRef:
apiGroup: rbac.authorization.k8s.io
kind: ClusterRole
name: lazy-koala
name: gazer
subjects:
- kind: ServiceAccount
name: lazy-koala
name: gazer
namespace: lazy-koala
---
apiVersion: v1
kind: ServiceAccount
metadata:
name: lazy-koala
name: gazer
namespace: lazy-koala
---
apiVersion: v1
Expand Down
26 changes: 13 additions & 13 deletions gazer/gazer.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,9 @@ def ipv4_request_event(self, cpu, data, size):
rpod = config_watcher.config[event['RADDR']]
request_received.labels(rpod['namespace'], rpod['serviceName'], pod['name']).inc()

if self.console_mode:
self.request_df = self.request_df.append(event, ignore_index=True)
self.request_df = self.request_df[-10:]
if self.console_mode:
self.request_df = self.request_df.append(event, ignore_index=True)
self.request_df = self.request_df[-25:]

def poll_requests(self):
while True:
Expand Down Expand Up @@ -122,15 +122,15 @@ def poll_syn_backlog(self):

backlog.labels(pod['namespace'], pod['serviceName'], pod['name'], row[0].slot).set(row[1].value)

if self.console_mode:
self.syn_df = self.syn_df.append({
"backlog": row[0].backlog,
"slot": row[0].slot,
"saddr": inet_ntop(AF_INET, pack("I", row[0].saddr)),
"lport": row[0].lport,
"value": row[1].value,
"outdated": False,
}, ignore_index=True)
if self.console_mode:
self.syn_df = self.syn_df.append({
"backlog": row[0].backlog,
"slot": row[0].slot,
"saddr": inet_ntop(AF_INET, pack("I", row[0].saddr)),
"lport": row[0].lport,
"value": row[1].value,
"outdated": False,
}, ignore_index=True)
self.syn_backlog_buffer.clear()
time.sleep(5)

Expand Down Expand Up @@ -159,7 +159,7 @@ def syn_backlog_text(self):
def request_log_text(self):
if self.request_df.empty:
return ""
return self.request_df.tail(10).__str__()
return self.request_df.tail(25).__str__()

def poll_data_in_bg(self):
poll_syn_backlog = threading.Thread(target=self.poll_syn_backlog, args=())
Expand Down

0 comments on commit 0a0e0d0

Please sign in to comment.