1 /* 2 * Copyright (c) 2022 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 #ifndef SINGLE_VER_DATA_SYNC_UTIL_H 16 #define SINGLE_VER_DATA_SYNC_UTIL_H 17 18 #include "message.h" 19 #include "single_ver_data_packet.h" 20 #include "single_ver_data_sync.h" 21 #include "single_ver_sync_task_context.h" 22 namespace DistributedDB { 23 class SingleVerDataSyncUtils { 24 public: 25 static int QuerySyncCheck(const SingleVerSyncTaskContext *context, bool &isCheckStatus); 26 27 static int AckMsgErrnoCheck(const SingleVerSyncTaskContext *context, const Message *message); 28 29 static int RequestQueryCheck(const DataRequestPacket *packet, SyncGenericInterface *storage); 30 31 static bool IsPermitLocalDeviceRecvData(const std::string &deviceId, const SecurityOption &remoteSecOption); 32 33 static bool IsPermitRemoteDeviceRecvData(const std::string &deviceId, const SecurityOption &remoteSecOption, 34 SyncGenericInterface *storage); 35 36 static void TransDbDataItemToSendDataItem(const std::string &localHashName, 37 std::vector<SendDataItem> &outData); 38 39 static std::string TransferForeignOrigDevName(const std::string &deviceName, const std::string &localHashName); 40 41 static void TransSendDataItemToLocal(const SingleVerSyncTaskContext *context, 42 const std::string &localHashName, const std::vector<SendDataItem> &data); 43 44 static void TranslateErrCodeIfNeed(int mode, uint32_t version, int &errCode); 45 46 static int RunPermissionCheck(SingleVerSyncTaskContext *context, const SyncGenericInterface* storage, 47 const std::string &label, const DataRequestPacket *packet); 48 49 static int RunPermissionCheck(SingleVerSyncTaskContext *context, const SyncGenericInterface* storage, 50 const std::string &label, int mode); 51 52 static bool CheckPermitReceiveData(const SingleVerSyncTaskContext *context, const ICommunicator *communicator, 53 const SyncGenericInterface *storage); 54 55 static void SetPacketId(DataRequestPacket *packet, SingleVerSyncTaskContext *context, uint32_t version); 56 57 static int GetMessageId(SyncType syncType); 58 59 static void PushAndPullKeyRevokHandle(SingleVerSyncTaskContext *context); 60 61 static int GetReSendMode(int mode, uint32_t sequenceId, SyncType syncType); 62 63 static void FillControlRequestPacket(ControlRequestPacket *packet, SingleVerSyncTaskContext *context); 64 65 static ControlCmdType GetControlCmdType(int mode); 66 67 static int GetModeByControlCmdType(ControlCmdType controlCmd); 68 69 static bool IsNeedTriggerQueryAutoSync(Message *inMsg, QuerySyncObject &query); 70 71 static void ControlAckErrorHandle(const SingleVerSyncTaskContext *context, 72 const std::shared_ptr<SubscribeManager> &subManager); 73 74 static void SetMessageHeadInfo(Message &message, uint16_t inMsgType, 75 const std::string &inTarget, uint32_t inSequenceId, uint32_t inSessionId); 76 77 static bool IsGetDataSuccessfully(int errCode); 78 79 static Timestamp GetMaxSendDataTime(const std::vector<SendDataItem> &inData); 80 81 static SyncTimeRange GetFullSyncDataTimeRange(const std::vector<SendDataItem> &inData, WaterMark localMark, 82 UpdateWaterMark &isUpdate); 83 84 static SyncTimeRange GetQuerySyncDataTimeRange(const std::vector<SendDataItem> &inData, WaterMark localMark, 85 WaterMark deleteLocalMark, UpdateWaterMark &isUpdate); 86 87 static SyncTimeRange ReviseLocalMark(SyncType syncType, const SyncTimeRange &dataTimeRange, 88 UpdateWaterMark updateMark); 89 90 static SyncTimeRange GetRecvDataTimeRange(SyncType syncType, 91 const std::vector<SendDataItem> &data, UpdateWaterMark &isUpdate); 92 93 static SyncTimeRange GetSyncDataTimeRange(SyncType syncType, WaterMark localMark, WaterMark deleteMark, 94 const std::vector<SendDataItem> &inData, UpdateWaterMark &isUpdate); 95 96 static std::pair<TimeOffset, TimeOffset> GetTimeOffsetFromRequestMsg(const Message *message); 97 98 static void RecordClientId(const SingleVerSyncTaskContext &context, const SyncGenericInterface &storage, 99 std::shared_ptr<Metadata> &metadata);; 100 101 static void SetDataRequestCommonInfo(const SingleVerSyncTaskContext &context, 102 const SyncGenericInterface &storage, DataRequestPacket &packet, std::shared_ptr<Metadata> &metadata); 103 104 static int SchemaVersionMatchCheck(const SingleVerSyncTaskContext &context, const DataRequestPacket &packet, 105 std::shared_ptr<Metadata> &metadata); 106 107 static int GetUnsyncTotal(const SingleVerSyncTaskContext *context, const SyncGenericInterface *storage, 108 uint32_t &total); 109 static int GetUnsyncTotal(const SingleVerSyncTaskContext *context, const SyncGenericInterface *storage, 110 SyncTimeRange &waterMarkInfo, uint32_t &total); 111 112 static bool IsSupportRequestTotal(uint32_t version); 113 114 static void UpdateSyncProcess(SingleVerSyncTaskContext *context, const DataRequestPacket *packet); 115 116 static void CacheInitWaterMark(SingleVerSyncTaskContext *context, SingleVerDataSync *dataSync); 117 118 static QuerySyncObject GetQueryFromDataRequest(const DataRequestPacket &packet, 119 const SingleVerSyncTaskContext &context, uint32_t sessionId); 120 private: 121 static int RunPermissionCheckInner(const SingleVerSyncTaskContext *context, const SyncGenericInterface* storage, 122 const std::string &label, const DataRequestPacket *packet, int mode); 123 }; 124 } 125 #endif // SINGLE_VER_DATA_SYNC_UTIL_H