some fixes on services

PowerDataHub · Mar 26, 2019 · 22b7699 · 22b7699
1 parent 7597f36
commit 22b7699
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -37,6 +37,7 @@ The Airflow service runs under systemd, so logs are available through journalctl
 - [x] Provide a way to pass a custom requirements.txt files on provision step
 - [ ] Provide a way to pass a custom packages.txt files on provision step
 - [ ] RBAC
+- [ ] Flower
 - [ ] Provide a way inject envinronment variables to airflow
 - [ ] Auto Scalling for workers
 - [ ] Use SPOT instances for workers

diff --git a/files/airflow.service b/files/airflow.service
@@ -1,14 +1,16 @@
 [Unit]
-Description=Airflow Webserver Service
+Description=Airflow daemon
+After=network.target
 
 [Service]
 EnvironmentFile=/etc/sysconfig/airflow
 User=ubuntu
 Group=ubuntu
-ExecStart=/usr/bin/terraform-aws-airflow
+Type=simple
 Restart=always
+ExecStart=/usr/bin/terraform-aws-airflow
 RestartSec=5s
-KillMode=mixed
-TimeoutStopSec=24h
+PrivateTmp=true
+
 [Install]
 WantedBy=multi-user.target
diff --git a/files/cloud-init.sh b/files/cloud-init.sh
@@ -72,7 +72,7 @@ function setup_airflow() {
 	sudo tee -a /usr/bin/terraform-aws-airflow <<EOL
 #!/usr/bin/env bash
 if [ "\$AIRFLOW_ROLE" == "SCHEDULER" ]
-then exec airflow scheduler
+then exec airflow scheduler -n 10
 elif [ "\$AIRFLOW_ROLE" == "WEBSERVER" ]
 then exec airflow webserver
 elif [ "\$AIRFLOW_ROLE" == "WORKER" ]