1 /* 2 * Copyright (C) 2021 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 #ifndef SRC_TRACE_PROCESSOR_SQLITE_CREATE_FUNCTION_H_ 18 #define SRC_TRACE_PROCESSOR_SQLITE_CREATE_FUNCTION_H_ 19 20 #include <sqlite3.h> 21 #include <unordered_map> 22 23 #include "src/trace_processor/sqlite/register_function.h" 24 25 namespace perfetto { 26 namespace trace_processor { 27 28 // Implementation of CREATE_FUNCTION SQL function. 29 // See https://perfetto.dev/docs/analysis/metrics#metric-helper-functions for 30 // usage of this function. 31 struct CreateFunction : public SqlFunction { 32 struct PerFunctionState { 33 ScopedStmt stmt; 34 // void* to avoid leaking state. 35 void* created_functon_context; 36 }; 37 struct NameAndArgc { 38 std::string name; 39 int argc; 40 41 struct Hasher { 42 std::size_t operator()(const NameAndArgc& s) const noexcept; 43 }; 44 bool operator==(const NameAndArgc& other) const { 45 return name == other.name && argc == other.argc; 46 } 47 }; 48 using State = std::unordered_map<NameAndArgc, 49 CreateFunction::PerFunctionState, 50 NameAndArgc::Hasher>; 51 52 struct Context { 53 sqlite3* db; 54 State* state; 55 }; 56 57 static base::Status Run(Context* ctx, 58 size_t argc, 59 sqlite3_value** argv, 60 SqlValue& out, 61 Destructors&); 62 }; 63 64 } // namespace trace_processor 65 } // namespace perfetto 66 67 #endif // SRC_TRACE_PROCESSOR_SQLITE_CREATE_FUNCTION_H_ 68