RMMRings.h

#pragma once
 
#include <chrono>
#include <future>
#include <iostream>
#include <memory>
#include <mutex>
#include <string>
#include <thread>
#include <unordered_map>
#include <vector>
 
#include "RMMRegisterAccess.h"
#include "RMMRing.h"
#include "RMMRingTimestamps.h"
#include "RingId.h"

struct PacketCountsData {
  uint32_t oddCount;                
  uint32_t evenCount;               
  uint32_t oddOverflow;             
  uint32_t evenOverflow;            
  std::map<int, uint32_t> perNode;  
};

using PacketCountMap = std::map<uint8_t, PacketCountsData>;

class RMMRings {
 public:
  RMMRings(RMMRegisterAccess &RMMRegs);
 
  rmmStatus teardownAllRings();
  rmmStatus tearDownRing(RingId ringId, bool waitTime = true);
  rmmStatus bringup(RingId ringId);
  void bringUpAllRings();
  void bringUpAllRingsInSequence();
  ringStatus checkRings();
  ringStatus checkRings(std::vector<int> &failedRings);
  ringStatus checkRing(RingId ringId);
  void resetPacketCounters();
  rmmStatus getPacketCounters(PacketCountMap *packets);
  rmmStatus getSideChannelPacketCounter(uint8_t node, uint32_t *value);
  uint32_t getNumberOfNodes(RingId ringId);
  void initialConfig();
  rmmStatus setRingEnabled(RingId ringId, bool enable);
 
  RMMLog rmmLog;                                   
  static constexpr int FE_DEFAULT_LOCK_TIME = 16;  
 
  rmmStatus setRingLogMask(RingId ringId, int mask);
  std::vector<int> getActiveRings() const;
  RMMRing &getRing(RingId ringId);
  bool isRingActive(RingId ringId);
 
  int getPollTimeout();
  void setPollTimeout(int value);
  void setRingPollTimeout(RingId ringId, int value);
  int getRingPollTimeout(RingId ringId);
 
  RingState getRingStatus(RingId ringId) const;
 
  void addFrontEndToRing(RingId ringId, std::shared_ptr<FrontEndBase> frontend);
 
  void configureAllNodesInRing(RingId ringId);
  rmmStatus getRingLogMsg(std::string *msg, RingId ringId);
 
  rmmStatus startBulkData();
  rmmStatus startRingBulkData(RingId ringId);
 
 private:
  static constexpr int BUCKET_SIZE = 7;
  std::array<std::unique_ptr<RMMRing>, RingId::MAX_NUM_RINGS> rings_;
  std::array<uint32_t, RingId::MAX_NUM_RINGS> pollTimeouts_{};
  std::atomic<uint32_t> pollTimeout_{10};
 
  RMMRegisterAccess &RMMRegs;
  std::mutex beStateMutex_;
 
  std::atomic<bool> _stopRequested{false};
 
  void setDomXcvr(DomFiber val);
 
  bool isInitialConfigDone();
};