nlsCppSdk/transport/nodeManager.h

/*
 * Copyright 2021 Alibaba Group Holding Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef NLS_SDK_NODE_MANAGER_H
#define NLS_SDK_NODE_MANAGER_H

#if defined(_MSC_VER)
#include <windows.h>
#else
#include <pthread.h>
#endif
#include <map>
#include <vector>

namespace AlibabaNls {

/* Node处于的最新运行状态 */
enum NodeStatus {
  NodeStatusInvalid = 0,
  NodeStatusCreated, /* 新建node */
  NodeStatusInvoking, /* 刚调用start的过程, 向notifyEventCallback发送c指令 */
  NodeStatusInvoked,    /* 调用start的过程, 在notifyEventCallback完成 */
  NodeStatusConnecting, /* 正在dns解析, 在dnsProcess中设置 */
  NodeStatusConnected,  /* socket链接成功 */
  NodeStatusHandshaking, /* ssl握手中 */
  NodeStatusHandshaked,  /* 握手成功 */
  NodeStatusRunning,     /* 运行中 */
  NodeStatusCancelling,  /* 调用cancel, 正在cancel过程中 */
  NodeStatusClosing,     /* 正在关闭ssl */
  NodeStatusClosed,      /* 已经调用完closed回调 */
  NodeStatusReleasing,
  NodeStatusReleased, /* 已销毁node */
};

typedef struct {
  void* request;
  void* node;
  void* instance;
  std::string uuid;
  int status;
} NodeInfo;

class NlsNodeManager {
 public:
  NlsNodeManager();
  virtual ~NlsNodeManager();

  int addRequestIntoInfoWithInstance(void* request, void* instance);
  int checkRequestWithInstance(void* request, void* instance);
  int removeInstanceFromInfo(void* instance);
  int removeRequestFromInfo(void* request, bool wait);

  int checkRequestExist(void* request, int* status);
  int checkNodeExist(void* node, int* status);
  int updateNodeStatus(void* node, int status);
  std::string getNodeStatusString(int status);

 private:
  enum NodeManagerConstValue {
    StepSleepMs = 500,
    DefaultRemoveTimeout = 2000,
    DefaultMaxUnalignedItemSize = 256,
    DefaultMaxUnalignedArrayLen = 2048,
  };

#ifdef _MSC_VER
  HANDLE _mtxNodeManager;
#else
  pthread_mutex_t _mtxNodeManager;
#endif

  std::map<void*, void*> _requestListByNode;
  std::map<void*, NodeInfo> _infoByRequest;
  int _timeout_ms;
};

}  // namespace AlibabaNls

#endif  // NLS_SDK_NODE_MANAGER_H