kylefang/react-native-firebase
1
0
Fork 0
mirror of https://github.com/zhigang1992/react-native-firebase.git synced 2026-04-18 12:12:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[perf][android] implement native methods

Browse Source
This commit is contained in:
Salakar
2019-02-20 07:01:29 +00:00
parent 07f37ad2de
commit 8356cce95c
1 changed files with 124 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

127
packages/perf/android/src/main/java/io/invertase/firebase/perf/ReactNativeFirebasePerfModule.java
View File

@@ -17,24 +17,145 @@ package io.invertase.firebase.perf;
*
*/
import android.app.Activity;
import android.util.SparseArray;
import com.facebook.react.bridge.Arguments;
import com.facebook.react.bridge.Promise;
import com.facebook.react.bridge.ReactApplicationContext;
import com.facebook.react.bridge.ReactMethod;
import com.facebook.react.bridge.ReadableMap;
import com.facebook.react.bridge.ReadableMapKeySetIterator;
import com.google.firebase.perf.FirebasePerformance;
import com.google.firebase.perf.metrics.HttpMetric;
import com.google.firebase.perf.metrics.Trace;
import javax.annotation.Nullable;
import java.util.HashMap;
import java.util.Map;
import io.invertase.firebase.common.ReactNativeFirebaseModule;
public class ReactNativeFirebasePerfModule extends ReactNativeFirebaseModule {
private static final String TAG = "Perf";
private static SparseArray<Trace> traces = new SparseArray<>();
private static SparseArray<HttpMetric> httpMetrics = new SparseArray<>();
ReactNativeFirebasePerfModule(ReactApplicationContext reactContext) {
super(reactContext, TAG);
}
@Override
public void onCatalystInstanceDestroy() {
super.onCatalystInstanceDestroy();
traces.clear();
httpMetrics.clear();
}
@ReactMethod
public void setPerformanceCollectionEnabled(Boolean enabled, Promise promise) {
FirebasePerformance.getInstance().setPerformanceCollectionEnabled(enabled);
promise.resolve(FirebasePerformance.getInstance().isPerformanceCollectionEnabled());
}
@ReactMethod
public void startTrace(int id, String identifier, Promise promise) {
try {
Trace trace = FirebasePerformance.getInstance().newTrace(identifier);
traces.put(id, trace);
trace.start();
promise.resolve(null);
} catch (Exception exception) {
rejectPromiseWithExceptionMap(promise, exception);
}
}
@ReactMethod
public void stopTrace(int id, ReadableMap traceData, Promise promise) {
try {
Trace trace = traces.get(id);
ReadableMap metrics = traceData.getMap("metrics");
ReadableMap attributes = traceData.getMap("attributes");
ReadableMapKeySetIterator metricKeySetIterator = metrics.keySetIterator();
ReadableMapKeySetIterator attrKeySetIterator = attributes.keySetIterator();
while (attrKeySetIterator.hasNextKey()) {
String attrName = attrKeySetIterator.nextKey();
trace.putAttribute(attrName, attributes.getString(attrName));
}
while (metricKeySetIterator.hasNextKey()) {
String metricName = metricKeySetIterator.nextKey();
trace.putMetric(metricName, metrics.getInt(metricName));
}
trace.stop();
traces.remove(id);
promise.resolve(null);
} catch (Exception exception) {
rejectPromiseWithExceptionMap(promise, exception);
}
}
@ReactMethod
public void startHttpMetric(int id, String url, String httpMethod, Promise promise) {
try {
HttpMetric httpMetric = FirebasePerformance.getInstance().newHttpMetric(url, httpMethod);
httpMetrics.put(id, httpMetric);
httpMetric.start();
promise.resolve(null);
} catch (Exception exception) {
rejectPromiseWithExceptionMap(promise, exception);
}
}
@ReactMethod
public void stopHttpMetric(int id, ReadableMap metricData, Promise promise) {
try {
HttpMetric httpMetric = httpMetrics.get(id);
if (metricData.hasKey("httpResponseCode")) {
httpMetric.setHttpResponseCode(metricData.getInt("httpResponseCode"));
}
if (metricData.hasKey("requestPayloadSize")) {
httpMetric.setRequestPayloadSize(metricData.getInt("requestPayloadSize"));
}
if (metricData.hasKey("responsePayloadSize")) {
httpMetric.setResponsePayloadSize(metricData.getInt("responsePayloadSize"));
}
if (metricData.hasKey("responseContentType")) {
httpMetric.setResponseContentType(metricData.getString("responseContentType"));
}
ReadableMap attributes = metricData.getMap("attributes");
ReadableMapKeySetIterator attrKeySetIterator = attributes.keySetIterator();
while (attrKeySetIterator.hasNextKey()) {
String attrName = attrKeySetIterator.nextKey();
httpMetric.putAttribute(attrName, attributes.getString(attrName));
}
httpMetric.stop();
httpMetrics.remove(id);
promise.resolve(null);
} catch (Exception exception) {
rejectPromiseWithExceptionMap(promise, exception);
}
}
@Override
public Map<String, Object> getConstants() {
final Map<String, Object> constants = new HashMap<>();
constants.put(
"isPerformanceCollectionEnabled",
FirebasePerformance.getInstance().isPerformanceCollectionEnabled()
);
return constants;
}
}