dataconnect.py: add "X-Firebase-Sqlconnect-Affinity" header for GSLB soft stickiness (#972)

Author: dconeybe

firebase_admin/dataconnect.py

@@ -361,6 +361,9 @@ def _get_headers(self) -> Dict[str, str]:
             "X-Firebase-Client": f"fire-admin-python/{firebase_admin.__version__}",
             "x-goog-api-client": _utils.get_metrics_header(),
             "X-Client-Version": f"python/{firebase_admin.__version__}",
+            "X-Firebase-Sqlconnect-Affinity": (
+                f"{self._project_id}{self._connector_config.service_id}"
+            ),
         }
 
     @staticmethod

tests/test_data_connect.py

@@ -741,6 +741,7 @@ def test_get_headers(self):
         assert headers.get("X-Firebase-Client") == f"fire-admin-python/{firebase_admin.__version__}"
         assert headers.get("x-goog-api-client") == _utils.get_metrics_header()
         assert headers.get("X-Client-Version") == f"python/{firebase_admin.__version__}"
+        assert headers.get("X-Firebase-Sqlconnect-Affinity") == "test-projectstarterproject"
 
 
 class TestDataConnectApiClientMakeGqlRequest: