Count flakiness pipeline issue tracker requests

appengine/chromium_try_flakes/issue_tracker/issue_tracker_api.py

 """Provides API wrapper for the codesite issue tracker"""
 
 import httplib2
 import logging
 import time
 
 from apiclient import discovery
 from apiclient.errors import HttpError
 from issue_tracker.issue import Issue
 from issue_tracker.comment import Comment
 from oauth2client.appengine import AppAssertionCredentials
 
+import gae_ts_mon
+
 
 # TODO(akuegel): Do we want to use a different timeout? Do we want to use a
 # cache? See documentation here:
 # https://github.com/jcgregorio/httplib2/blob/master/python2/httplib2/__init__.py#L1142
 def _createHttpObject(scope):  # pragma: no cover
   credentials = AppAssertionCredentials(scope=scope)
   return credentials.authorize(httplib2.Http())
 
 
 def _buildClient(api_name, api_version, http,
@@ skipping 21 matching lines @@
         logging.exception(
             'apiclient.discovery.build() failed for %s too many times.',
             api_name)
         raise e
   return client
 
 
 class IssueTrackerAPI(object):  # pragma: no cover
   CAN_ALL = 'all'
 
+  issue_tracker_requests = gae_ts_mon.CounterMetric(
+      'flakiness_pipeline/issue_tracker_requests',
+      description='Number of requests to the issue tracker')
+
   """A wrapper around the issue tracker api."""
   def __init__(self, project_name):
     self.project_name = project_name
 
     self.client = _buildClient(
         'monorail', 'v1',
         _createHttpObject('https://www.googleapis.com/auth/userinfo.email'),
         'https://monorail-prod.appspot.com/_ah/api/discovery/v1/apis/{api}/'
         '{apiVersion}/rest')
 
@@ skipping 21 matching lines @@
       body['owner'] = {'name': issue.owner}
     if issue.labels:
       body['labels'] = issue.labels
     if issue.components:
       body['components'] = issue.components
     if issue.cc:
       body['cc'] = [{'name': user} for user in issue.cc]
     request = self.client.issues().insert(
         projectId=self.project_name, sendEmail=send_email, body=body)
     tmp = self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'issues_insert'})
     issue.id = int(tmp['id'])
     issue.dirty = False
     return issue
 
   def update(self, issue, comment=None, send_email=True):
     if not issue.dirty and not comment:
       return issue
 
     updates = {}
     if 'summary' in issue.changed:
@@ skipping 18 matching lines @@
     body = {'id': issue.id,
             'updates': updates}
 
     if comment:
       body['content'] = comment
 
     request = self.client.issues().comments().insert(
         projectId=self.project_name, issueId=issue.id, sendEmail=send_email,
         body=body)
     self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'comments_insert'})
 
     if issue.owner == '----':
       issue.owner = ''
 
     issue.dirty = False
     return issue
 
   def addComment(self, issue_id, comment, send_email=True):
     issue = self.getIssue(issue_id)
     self.update(issue, comment, send_email)
 
   def getCommentCount(self, issue_id):
     request = self.client.issues().comments().list(
         projectId=self.project_name, issueId=issue_id, startIndex=1,
         maxResults=0)
     feed = self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'comments_list'})
     return feed.get('totalResults', '0')
 
   def getComments(self, issue_id):
     rtn = []
 
     request = self.client.issues().comments().list(
         projectId=self.project_name, issueId=issue_id)
     feed = self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'comments_list'})
     rtn.extend([Comment(entry) for entry in feed['items']])
     total_results = feed['totalResults']
     if not total_results:
       return rtn
 
     while len(rtn) < total_results:
       request = self.client.issues().comments().list(
           projectId=self.project_name, issueId=issue_id, startIndex=len(rtn))
       feed = self._retry_api_call(request)
+      self.issue_tracker_requests.increment(
+          {'source': 'chromium-try-flakes', 'operation': 'comments_list'})
       rtn.extend([Comment(entry) for entry in feed['items']])
 
     return rtn
 
   def getFirstComment(self, issue_id):
     request = self.client.issues().comments().list(
         projectId=self.project_name, issueId=issue_id, startIndex=0,
         maxResults=1)
     feed = self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'comments_list'})
     if 'items' in feed and len(feed['items']) > 0:
       return Comment(feed['items'][0])
     return None
 
   def getLastComment(self, issue_id):
     total_results = self.getCommentCount(issue_id)
     request = self.client.issues().comments().list(
         projectId=self.project_name, issueId=issue_id,
         startIndex=total_results-1, maxResults=1)
     feed = self._retry_api_call(request)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'comments_list'})
     if 'items' in feed and len(feed['items']) > 0:
       return Comment(feed['items'][0])
     return None
 
   def getIssue(self, issue_id):
     """Retrieve a set of issues in a project."""
     request = self.client.issues().get(
         projectId=self.project_name, issueId=issue_id)
+    self.issue_tracker_requests.increment(
+        {'source': 'chromium-try-flakes', 'operation': 'issues_get'})
     entry = self._retry_api_call(request)
     return Issue(entry)