[AIRBNB] trigger counter on sync (apache#361)

cong-zhu · Nov 6, 2019 · a6a1353 · a6a1353
1 parent 9e54e19
commit a6a1353
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow/www_rbac/security.py b/airflow/www_rbac/security.py
@@ -27,6 +27,7 @@
 from sqlalchemy.exc import IntegrityError
 
 from airflow import models
+from airflow.settings import Stats
 from airflow.www_rbac.app import appbuilder
 from airflow.utils.db import provide_session
 from airflow.utils.log.logging_mixin import LoggingMixin
@@ -498,6 +499,7 @@ def sync_roles(self):
             logging.info('Failed to clean perms')
 
     def sync_lookup_table(self):
+        Stats.incr('airflow.sync_lookup_table')
         ab_perm_view_role = sqla_models.assoc_permissionview_role
         perm_view = self.permissionview_model
         view_menu = self.viewmenu_model