cpp_client_telemetry/lib/offline/MemoryStorage.hpp at 0159528c71402a4c4c1a78a87392307c08f1c1be · microsoft/cpp_client_telemetry · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
//
// Copyright (c) Microsoft Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
//
#ifndef MEMORYSTORAGE_HPP
#define MEMORYSTORAGE_HPP

#pragma once

#include "pal/PAL.hpp"

#include "IOfflineStorage.hpp"

#include "api/IRuntimeConfig.hpp"

#include "ILogManager.hpp"

#include <algorithm>
#include <memory>
#include <mutex>
#include <map>
#include <string>
#include <unordered_set>

namespace MAT_NS_BEGIN {

    class MemoryStorage final : public IOfflineStorage
    {

    public:
        MemoryStorage(ILogManager& logManager, IRuntimeConfig& runtimeConfig);

        virtual void Initialize(IOfflineStorageObserver& observer) override;

        virtual void Shutdown() override;

        virtual void Flush() override;

        virtual bool StoreRecord(StorageRecord const& record) override;

        virtual size_t StoreRecords(std::vector<StorageRecord> & records) override;

        virtual bool GetAndReserveRecords(std::function<bool(StorageRecord&&)> const& consumer, unsigned leaseTimeMs,
            EventLatency minLatency = EventLatency_Unspecified, unsigned maxCount = 0) override;

        virtual bool IsLastReadFromMemory() override;

        virtual unsigned LastReadRecordCount() override;

        virtual void DeleteAllRecords() override;

        virtual void DeleteRecords(const std::map<std::string, std::string> & whereFilter = {}) override;

        virtual void DeleteRecords(std::vector<StorageRecordId> const& ids, HttpHeaders headers, bool& fromMemory) override;

        virtual void ReleaseRecords(std::vector<StorageRecordId> const& ids, bool incrementRetryCount, HttpHeaders headers, bool& fromMemory) override;

        virtual void ReleaseAllRecords() override;

        virtual bool StoreSetting(std::string const& name, std::string const& value) override;

        virtual std::string GetSetting(std::string const& name) override;

        virtual bool DeleteSetting(std::string const& name) override;

        virtual size_t GetSize() override;

        virtual size_t GetRecordCount(EventLatency latency = EventLatency_Unspecified) const override;

        virtual size_t GetRemainingRecordCountForShutdown() const override;

        size_t GetReservedCount();

        virtual std::vector<StorageRecord> GetRecords(bool shutdown = false, EventLatency minLatency = EventLatency_Unspecified, unsigned maxCount = 0) override;

        virtual bool ResizeDb() override;

        virtual ~MemoryStorage() override;

    protected:

        IOfflineStorageObserver*    m_observer;
        IRuntimeConfig&             m_config;
        ILogManager&                m_logManager;

        mutable std::mutex          m_records_lock;
        std::vector<StorageRecord>  m_records[EventLatency_Max+1];

        /// <summary>
        /// Contains reserved (aka in-flight) records.
        /// Current storage interface API requires deletion and release by StorageRecordId.
        /// </summary>
        mutable std::mutex          m_reserved_lock;
        std::map<StorageRecordId, StorageRecord> m_reserved_records;

        size_t                      m_size;

        MATSDK_LOG_DECL_COMPONENT_CLASS();

    private:
        size_t                      m_lastReadCount;

    };

} MAT_NS_END
#endif