1 /*
2  * Copyright (C) 2019 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <aidl/android/os/IPullAtomCallback.h>
20 #include "StatsPuller.h"
21 
22 using aidl::android::os::IPullAtomCallback;
23 using std::shared_ptr;
24 
25 namespace android {
26 namespace os {
27 namespace statsd {
28 
29 class StatsCallbackPuller : public StatsPuller {
30 public:
31     explicit StatsCallbackPuller(int tagId, const shared_ptr<IPullAtomCallback>& callback,
32                                  const int64_t coolDownNs, const int64_t timeoutNs,
33                                  const std::vector<int> additiveFields);
34 
35 private:
36     bool PullInternal(vector<std::shared_ptr<LogEvent>>* data) override;
37     const shared_ptr<IPullAtomCallback> mCallback;
38 
39     FRIEND_TEST(StatsCallbackPullerTest, PullFail);
40     FRIEND_TEST(StatsCallbackPullerTest, PullSuccess);
41     FRIEND_TEST(StatsCallbackPullerTest, PullTimeout);
42 };
43 
44 }  // namespace statsd
45 }  // namespace os
46 }  // namespace android
47