1 files changed, 87 insertions, 0 deletions
diff --git a/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/event/EventSnippet.java b/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/event/EventSnippet.java
new file mode 100644
index 0000000..4cebb74
--- /dev/null
+++ b/third_party/sl4a/src/main/java/com/google/android/mobly/snippet/event/EventSnippet.java
@@ -0,0 +1,87 @@
+/*
+ * Copyright (C) 2017 Google Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+
+package com.google.android.mobly.snippet.event;
+
+import androidx.annotation.Nullable;
+import com.google.android.mobly.snippet.Snippet;
+import com.google.android.mobly.snippet.rpc.Rpc;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.concurrent.LinkedBlockingDeque;
+import java.util.concurrent.TimeUnit;
+import org.json.JSONException;
+import org.json.JSONObject;
+
+public class EventSnippet implements Snippet {
+    private static class EventSnippetException extends Exception {
+        private static final long serialVersionUID = 1L;
+
+        public EventSnippetException(String msg) {
+            super(msg);
+        }
+    }
+
+    private static final int DEFAULT_TIMEOUT_MILLISECOND = 60 * 1000;
+    private final EventCache mEventCache = EventCache.getInstance();
+
+    @Rpc(
+            description =
+                    "Blocks until an event of a specified type has been received. The returned event is removed from the cache. Default timeout is 60s.")
+    public JSONObject eventWaitAndGet(
+            String callbackId, String eventName, @Nullable Integer timeout)
+            throws InterruptedException, JSONException, EventSnippetException {
+        // The server side should never wait forever, so we'll use a default timeout is one is not
+        // provided.
+        if (timeout == null) {
+            timeout = DEFAULT_TIMEOUT_MILLISECOND;
+        }
+        String qId = EventCache.getQueueId(callbackId, eventName);
+        LinkedBlockingDeque<SnippetEvent> q = mEventCache.getEventDeque(qId);
+        SnippetEvent result = q.pollFirst(timeout, TimeUnit.MILLISECONDS);
+        if (result == null) {
+            throw new EventSnippetException("timeout.");
+        }
+        return result.toJson();
+    }
+
+    @Rpc(
+            description =
+                    "Gets and removes all the events of a certain name that have been received so far. "
+                            + "Non-blocking. Potentially racey since it does not guarantee no event of "
+                            + "the same name will occur after the call.")
+    public List<JSONObject> eventGetAll(String callbackId, String eventName)
+            throws InterruptedException, JSONException {
+        String qId = EventCache.getQueueId(callbackId, eventName);
+        LinkedBlockingDeque<SnippetEvent> q = mEventCache.getEventDeque(qId);
+        ArrayList<JSONObject> results = new ArrayList<>(q.size());
+        ArrayList<SnippetEvent> buffer = new ArrayList<>(q.size());
+        q.drainTo(buffer);
+        for (SnippetEvent snippetEvent : buffer) {
+            results.add(snippetEvent.toJson());
+        }
+        if (results.size() == 0) {
+            return Collections.emptyList();
+        }
+        return results;
+    }
+
+    @Override
+    public void shutdown() {
+        mEventCache.clearAll();
+    }
+}