Line data    Source code

       1              : /**
       2              :  * Copyright (c) 2025 Huawei Technologies Co., Ltd.
       3              :  * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
       4              :  * CANN Open Software License Agreement Version 2.0 (the "License").
       5              :  * Please refer to the License for details. You may not use this file except in compliance with the License.
       6              :  * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
       7              :  * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
       8              :  * See LICENSE in the root of the software repository for the full text of the License.
       9              :  */
      10              : 
      11              : #ifndef DGW_GROUP_ENTITY_H
      12              : #define DGW_GROUP_ENTITY_H
      13              : 
      14              : #include "entity.h"
      15              : #include "dgw_client.h"
      16              : 
      17              : namespace dgw {
      18              : 
      19              : struct GroupEntityInfo {
      20              :     int32_t groupId;               // group id
      21              :     bqs::GroupPolicy groupPolicy;  // group policy
      22              :     int64_t timeout;               // timeout interval, us
      23              :     uint64_t lastTransId;          // last route label
      24              :     uint64_t lastTimestamp;        // timestamp of last data migration, us
      25              :     uint32_t peerInstanceNum;
      26              :     uint32_t localInstanceIndex;
      27              : };
      28              : 
      29              : class GroupEntity : public Entity {
      30              : public:
      31              :     explicit GroupEntity(const EntityMaterial &material, const uint32_t resIndex);
      32           27 :     virtual ~GroupEntity() = default;
      33              :     GroupEntity(const GroupEntity &) = delete;
      34              :     GroupEntity(const GroupEntity &&) = delete;
      35              :     GroupEntity &operator = (const GroupEntity &) = delete;
      36              :     GroupEntity &operator = (GroupEntity &&) = delete;
      37              : 
      38              :     FsmStatus Dequeue() override;
      39              :     void SelectDstEntities(const uint64_t key, std::vector<Entity*> &toPushDstEntities,
      40              :         std::vector<Entity*> &reprocessDstEntities, std::vector<Entity*> &abnormalDstEntities) override;
      41              :     void ReprocessInTryPush(const Entity &srcEntity, DynamicRequestPtr &dynamicRequest, uint32_t &schedCfgKey) override;
      42              :     FsmStatus AbProcessInTryPush() override;
      43              :     FsmStatus PauseSubscribe(const Entity &fullEntity) override;
      44              :     FsmStatus ResumeSubscribe(const Entity &notFullEntity) override;
      45              :     FsmStatus ClearQueue() override;
      46              :     FsmStatus MakeSureOutputCompletion() override;
      47              :     uint32_t GetMbufDeviceId() const override;
      48              :     uint32_t GetMbufQueueType() const override;
      49              : 
      50              : private:
      51              :     void SetGroupInfo(const uint64_t lastTransId, const uint64_t lastTimestamp);
      52              :     EntityPtr SelectSrcEntity(FsmStatus &status);
      53              :     bool Match(const Entity &entity, const uint64_t waitTransId, bool exactlyMatch) const;
      54              :     FsmStatus PeekFromEntityInGroup(Entity &entity, const uint64_t waitTransId) const;
      55              :     bool CheckTimeout(const uint64_t waitTransId) const;
      56              :     EntityPtr SelectEntityWithMinTransId(const std::vector<EntityPtr> &entities) const;
      57              : 
      58              :     GroupEntityInfo groupInfo_;
      59              :     uint32_t mbufDeviceId_;
      60              :     uint32_t mbufQueueType_;
      61              : };
      62              : }
      63              : #endif