daemon: use setproctitle to name process journalpump

This way, the process shows up as journalpump and not as one of many python processes.
Aiven-Open · Mar 1, 2022 · fd67952 · fd67952
1 parent 9f1a540
commit fd67952
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 4 deletions.
diff --git a/debian/control b/debian/control
@@ -15,7 +15,7 @@ Depends: ${misc:Depends}, ${python3:Depends}, adduser,
  python3-elasticsearch, python3-kafka,
  python3-requests, python3-systemd,
  python3-boto3, python3-googleapi,
- python3-websockets, python3-aiohttp-socks, python3-snappy
+ python3-websockets, python3-aiohttp-socks, python3-snappy, python3-setproctitle
 Description: daemon that takes log messages from journald and pumps them
  to a given output.  Currently supported outputs are Elasticsearch, Kafka,
  Websocket, logplex, AWS CloudWatch and Google Cloud Logging. It reads

diff --git a/journalpump.spec b/journalpump.spec
@@ -7,10 +7,10 @@ License:        ASL 2.0
 Source0:        journalpump-rpm-src.tar.gz
 Requires:       python3-kafka, systemd-python3, python3-requests, python3-boto3, python3-google-api-client
 Requires:       python3-oauth2client, python3-geoip2, python3-typing-extensions
-Requires:	python3-websockets, python3-aiohttp-socks, python3-snappy
+Requires:	python3-websockets, python3-aiohttp-socks, python3-snappy python3-setproctitle
 BuildRequires:  python3-kafka, systemd-python3, python3-requests, python3-boto3, python3-google-api-client
 BuildRequires:  python3-devel, python3-pytest, python3-pylint python3-responses
-BuildRequires:	python3-websockets, python3-aiohttp-socks, python3-snappy
+BuildRequires:	python3-websockets, python3-aiohttp-socks, python3-snappy python3-setproctitle
 BuildArch:      noarch
 Obsoletes:      kafkajournalpump
 

diff --git a/journalpump/daemon.py b/journalpump/daemon.py
@@ -14,6 +14,8 @@
 import signal
 import sys
 
+from setproctitle import setproctitle
+
 LOG_FORMAT = "%(asctime)s\t%(name)s\t%(levelname)s\t%(message)s"
 LOG_FORMAT_JOURNAL = "%(name)-20s  %(levelname)-8s  %(message)s"
 LOG_FORMAT_JOURNAL_MULTI_THREAD = "%(name)-20s  %(threadName)-15s %(levelname)-8s  %(message)s"
@@ -111,6 +113,7 @@ def main(cls, args):
         try:
             exe = cls(config_path=args[0])
             daemon.notify("READY=1")
+            setproctitle("journalpump")
             return exe.run()
         except ServiceDaemonError as ex:
             logging.fatal("%s failed to start: %s", cls.__name__, ex)

diff --git a/requirements.txt b/requirements.txt
@@ -8,4 +8,5 @@ google-api-python-client
 oauth2client
 geoip2
 https://github.com/systemd/python-systemd/zipball/master
-typing-extensions
+typing-extensions
+setproctitle
diff --git a/setup.py b/setup.py
@@ -19,6 +19,7 @@
         "oauth2client",
         "geoip2",
         "typing-extensions",
+        "setproctitle",
     ],
     dependency_links=[],
     package_data={},