1 /*
2 * Copyright (c) 2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "call_reporter.h"
17
18 #include <cinttypes>
19 #include "datashare_log.h"
20 #include "datashare_string_utils.h"
21
22 namespace OHOS {
23 namespace DataShare {
Count(const std::string & funcName,const std::string & uri)24 void DataShareCallReporter::Count(const std::string &funcName, const std::string &uri)
25 {
26 int overCount = 0;
27 int64_t firstCallTime = 0;
28 callCounts.Compute(funcName, [&overCount, &firstCallTime](auto &key, CallInfo &callInfo) {
29 auto callCount = callInfo.count;
30 if (callCount == 0) {
31 callInfo.firstTime = std::chrono::system_clock::now();
32 }
33 if (++callCount % RESET_COUNT_THRESHOLD == 0) {
34 int64_t first = std::chrono::duration_cast<std::chrono::milliseconds>(
35 callInfo.firstTime.time_since_epoch()).count();
36 int64_t now = std::chrono::duration_cast<std::chrono::milliseconds>(
37 std::chrono::system_clock::now().time_since_epoch()).count();
38 ++overCount;
39 firstCallTime = first;
40 if (now - first <= TIME_THRESHOLD.count()) {
41 ++overCount;
42 }
43 callCount = 0;
44 }
45 callInfo.count = callCount;
46 return true;
47 });
48 int64_t now = std::chrono::duration_cast<std::chrono::milliseconds>(
49 std::chrono::system_clock::now().time_since_epoch()).count();
50 if (overCount > 0) {
51 LOG_WARN("Call the threshold, func: %{public}s, first:%{public}" PRIi64 "ms, now:%{public}" PRIi64
52 "ms, uri:%{public}s", funcName.c_str(), firstCallTime, now, DataShareStringUtils::Anonymous(uri).c_str());
53 }
54 if (overCount > 1) {
55 LOG_WARN("Call too frequently, func: %{public}s, first:%{public}" PRIi64 "ms, now:%{public}" PRIi64
56 "ms, uri:%{public}s", funcName.c_str(), firstCallTime, now, DataShareStringUtils::Anonymous(uri).c_str());
57 }
58 }
59 } // namespace DataShare
60 } // namespace OHOS