nlsCppSdk/transport/nlsEventNetWork.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_NETWORK_H
#define NLS_SDK_NETWORK_H

#if defined(_MSC_VER)
#include <windows.h>
#else
#include <pthread.h>
#endif
#include "event2/util.h"
#include "nlsEncoder.h"

namespace AlibabaNls {

class INlsRequest;
class WorkThread;
#ifdef ENABLE_PRECONNECTED_POOL
class ConnectedPool;
#endif

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

  static NlsEventNetWork *_eventClient;
  NlsClientImpl *getInstance() { return _instance; }

  static void DnsLogCb(int w, const char *m);
  static void EventLogCb(int w, const char *m);

  void initEventNetWork(NlsClientImpl *instance, int count, char *aiFamily,
                        char *directIp, bool sysGetAddr,
                        unsigned int syncCallTimeoutMs);
  void destroyEventNetWork();

  int start(INlsRequest *request);
  int sendAudio(INlsRequest *request, const uint8_t *data, size_t dataSize,
                ENCODER_TYPE type);
  int sendText(INlsRequest *request, const char *text);
  int sendPing(INlsRequest *request);
  int sendFlush(INlsRequest *request, const char *parameters);
  int stop(INlsRequest *request);
  int cancel(INlsRequest *request);
  int stControl(INlsRequest *request, const char *message);
  const char *dumpAllInfo(INlsRequest *request);

#ifdef ENABLE_PRECONNECTED_POOL
  int startInner(INlsRequest *request);
  int initPreconnectedPool(unsigned int maxNumber,
                           unsigned int connectedTimeoutMs,
                           unsigned int requestedTimeoutMs);
  int destroyPreconnectedPool();
  ConnectedPool *getPreconnectedPool();
#endif

 private:
  int selectThreadNumber();  //循环选择工作线程

  WorkThread *_workThreadArray;  //工作线程数组
  size_t _workThreadsNumber;     //工作线程数量
  size_t _currentCpuNumber;
  int _addrInFamily;
  char _directIp[64];
  bool _enableSysGetAddr;  //启用getaddrinfo_a接口进行dns解析, 默认false
  unsigned int _syncCallTimeoutMs;  //启用同步接口, 默认0为不启用同步接口
  NlsClientImpl *_instance;

#ifdef ENABLE_PRECONNECTED_POOL
  ConnectedPool *_preconnectedPool;  //预连接池工作线程
  unsigned int _maxPreconnectedNumber;
  unsigned int _preconnectedTimeoutMs;
  unsigned int _prerequestedTimeoutMs;
#endif

#if defined(_MSC_VER)
  static HANDLE _mtxThread;
#else
  static pthread_mutex_t _mtxThread;
#endif
};

}  // namespace AlibabaNls

#endif  // NLS_SDK_NETWORK_H