Add instance health checks for all gunicorn workers

Fixes https://jira.coreos.com/browse/QS-121

endpoints/verbs/__init__.py

@@ -10,7 +10,7 @@ from auth.auth_context import get_authenticated_user
 from auth.decorators import process_auth
 from auth.permissions import ReadRepositoryPermission
 from data import database
-from endpoints.decorators import anon_protect, route_show_if, parse_repository_name
+from endpoints.decorators import anon_protect, anon_allowed, route_show_if, parse_repository_name
 from endpoints.verbs.models_pre_oci import pre_oci_model as model
 from endpoints.v2.blob import BLOB_DIGEST_ROUTE
 from image.appc import AppCImageFormatter
@@ -394,3 +394,9 @@ def get_tag_torrent(namespace_name, repo_name, digest):
 
   metric_queue.repository_pull.Inc(labelvalues=[namespace_name, repo_name, 'torrent', True])
   return _torrent_for_blob(blob, repo_is_public)
+
+
+@verbs.route('/_internal_ping')
+@anon_allowed
+def internal_ping():
+  return make_response('true', 200)

endpoints/web.py

@@ -67,6 +67,10 @@ STATUS_TAGS = app.config['STATUS_TAGS']
 def index(path, **kwargs):
   return render_page_template_with_routedata('index.html', **kwargs)
 
+@web.route('/_internal_ping')
+@anon_allowed
+def internal_ping():
+  return make_response('true', 200)
 
 @web.route('/500', methods=['GET'])
 def internal_error_display():

health/services.py

@@ -5,27 +5,30 @@ from health.models_pre_oci import pre_oci_model as model
 logger = logging.getLogger(__name__)
 
 
-def _check_registry_gunicorn(app):
-  """ Returns the status of the registry gunicorn workers. """
-  # Compute the URL for checking the registry endpoint. We append a port if and only if the
-  # hostname contains one.
-  client = app.config['HTTPCLIENT']
-  hostname_parts = app.config['SERVER_HOSTNAME'].split(':')
-  port = ''
-  if len(hostname_parts) == 2:
-    port = ':' + hostname_parts[1]
+def _check_gunicorn(endpoint):
+  def fn(app):
+    """ Returns the status of the gunicorn workers. """
+    # Compute the URL for checking the endpoint. We append a port if and only if the
+    # hostname contains one.
+    client = app.config['HTTPCLIENT']
+    hostname_parts = app.config['SERVER_HOSTNAME'].split(':')
+    port = ''
+    if len(hostname_parts) == 2:
+      port = ':' + hostname_parts[1]
 
-  scheme = app.config['PREFERRED_URL_SCHEME']
-  if app.config.get('EXTERNAL_TLS_TERMINATION', False):
-    scheme = 'http'
+    scheme = app.config['PREFERRED_URL_SCHEME']
+    if app.config.get('EXTERNAL_TLS_TERMINATION', False):
+      scheme = 'http'
 
-  registry_url = '%s://localhost%s/v1/_internal_ping' % (scheme, port)
-  try:
-    status_code = client.get(registry_url, verify=False, timeout=2).status_code
-    return (status_code == 200, 'Got non-200 response for registry: %s' % status_code)
-  except Exception as ex:
-    logger.exception('Exception when checking registry health: %s', registry_url)
-    return (False, 'Exception when checking registry health: %s' % registry_url)
+    registry_url = '%s://localhost%s/%s' % (scheme, port, endpoint)
+    try:
+      status_code = client.get(registry_url, verify=False, timeout=2).status_code
+      return (status_code == 200, 'Got non-200 response for worker: %s' % status_code)
+    except Exception as ex:
+      logger.exception('Exception when checking worker health: %s', registry_url)
+      return (False, 'Exception when checking worker health: %s' % registry_url)
+
+  return fn
 
 
 def _check_database(app):
@@ -53,7 +56,9 @@ def _check_auth(app):
 
 
 _SERVICES = {
-  'registry_gunicorn': _check_registry_gunicorn,
+  'registry_gunicorn': _check_gunicorn('v1/_internal_ping'),
+  'web_gunicorn': _check_gunicorn('_internal_ping'),
+  'verbs_gunicorn': _check_gunicorn('c1/_internal_ping'),
   'database': _check_database,
   'redis': _check_redis,
   'storage': _check_storage,