Skip to content

Set TrafficStats tag in FirebaseInstallationServiceClient.createFirebaseInstallation #2270

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Jan 11, 2021
Merged

Set TrafficStats tag in FirebaseInstallationServiceClient.createFirebaseInstallation #2270

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
1d9d0b6
Set TrafficStats tag in FirebaseInstallationServiceClient.createFireb…
regisd Dec 17, 2020
78fe3d4
Add TrafficStats tags for Create isntallation, Delete installation an…
regisd Jan 11, 2021
537e2b5
Rename tags to be prefixed with TRAFFIC_STATS
regisd Jan 11, 2021
d215302
Update copyright header for 2021.
regisd Jan 11, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 20 additions & 2 deletions ...main/java/com/google/firebase/installations/remote/FirebaseInstallationServiceClient.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Copyright 2019 Google LLC
// Copyright 2019-2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
Expand All @@ -20,6 +20,7 @@

import android.content.Context;
import android.content.pm.PackageManager;
import android.net.TrafficStats;
import android.text.TextUtils;
import android.util.JsonReader;
import android.util.Log;
Expand Down Expand Up @@ -56,6 +57,16 @@
* @hide
*/
public class FirebaseInstallationServiceClient {

// TrafficStats tags must be kept in sync with java/com/google/android/gms/libs/punchclock
private static final int TRAFFIC_STATS_FIREBASE_INSTALLATIONS_TAG = 0x00008000;
private static final int TRAFFIC_STATS_CREATE_INSTALLATION_TAG =
TRAFFIC_STATS_FIREBASE_INSTALLATIONS_TAG | 0x1;
private static final int TRAFFIC_STATS_DELETE_INSTALLATION_TAG =
TRAFFIC_STATS_FIREBASE_INSTALLATIONS_TAG | 0x2;
private static final int TRAFFIC_STATS_GENERATE_AUTH_TOKEN_TAG =
TRAFFIC_STATS_FIREBASE_INSTALLATIONS_TAG | 0x3;

private static final String FIREBASE_INSTALLATIONS_API_DOMAIN =
"firebaseinstallations.googleapis.com";
private static final String CREATE_REQUEST_RESOURCE_NAME_FORMAT = "projects/%s/installations";
Expand Down Expand Up @@ -153,6 +164,7 @@ public InstallationResponse createFirebaseInstallation(
URL url = getFullyQualifiedRequestUri(resourceName);
for (int retryCount = 0; retryCount <= MAX_RETRIES; retryCount++) {

TrafficStats.setThreadStatsTag(TRAFFIC_STATS_CREATE_INSTALLATION_TAG);
HttpURLConnection httpURLConnection = openHttpURLConnection(url, apiKey);

try {
Expand Down Expand Up @@ -194,6 +206,7 @@ public InstallationResponse createFirebaseInstallation(
continue;
} finally {
httpURLConnection.disconnect();
TrafficStats.clearThreadStatsTag();
}
}

Expand Down Expand Up @@ -309,6 +322,7 @@ public void deleteFirebaseInstallation(

int retryCount = 0;
while (retryCount <= MAX_RETRIES) {
TrafficStats.setThreadStatsTag(TRAFFIC_STATS_DELETE_INSTALLATION_TAG);
HttpURLConnection httpURLConnection = openHttpURLConnection(url, apiKey);
try {
httpURLConnection.setRequestMethod("DELETE");
Expand All @@ -335,6 +349,7 @@ public void deleteFirebaseInstallation(
retryCount++;
} finally {
httpURLConnection.disconnect();
TrafficStats.clearThreadStatsTag();
}
}

Expand Down Expand Up @@ -390,6 +405,7 @@ public TokenResult generateAuthToken(
URL url = getFullyQualifiedRequestUri(resourceName);
for (int retryCount = 0; retryCount <= MAX_RETRIES; retryCount++) {

TrafficStats.setThreadStatsTag(TRAFFIC_STATS_GENERATE_AUTH_TOKEN_TAG);
HttpURLConnection httpURLConnection = openHttpURLConnection(url, apiKey);
try {
httpURLConnection.setRequestMethod("POST");
Expand Down Expand Up @@ -430,6 +446,7 @@ public TokenResult generateAuthToken(
continue;
} finally {
httpURLConnection.disconnect();
TrafficStats.clearThreadStatsTag();
}
}
throw new FirebaseInstallationsException(
Expand Down Expand Up @@ -619,7 +636,8 @@ private static String readErrorResponse(HttpURLConnection conn) {
response.append(input).append('\n');
}
return String.format(
"Error when communicating with the Firebase Installations server API. HTTP response: [%d %s: %s]",
"Error when communicating with the Firebase Installations server API. HTTP response: [%d"
+ " %s: %s]",
conn.getResponseCode(), conn.getResponseMessage(), response);
} catch (IOException ignored) {
return null;
Expand Down