lib/cpp/src/thrift/async/TEvhttpClientChannel.h

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you 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 _THRIFT_TEVHTTP_CLIENT_CHANNEL_H_
#define _THRIFT_TEVHTTP_CLIENT_CHANNEL_H_ 1

#include <queue>
#include <string>
#include <utility>
#include <memory>
#include <thrift/async/TAsyncChannel.h>

struct event_base;
struct evdns_base;
struct evhttp_connection;
struct evhttp_request;

namespace apache {
namespace thrift {
namespace transport {
class TMemoryBuffer;
}
}
}

namespace apache {
namespace thrift {
namespace async {

class TEvhttpClientChannel : public TAsyncChannel {
public:
  using TAsyncChannel::VoidCallback;

  TEvhttpClientChannel(const std::string& host,
                       const std::string& path,
                       const char* address,
                       int port,
                       struct event_base* eb,
                       struct evdns_base *dnsbase = nullptr);
  ~TEvhttpClientChannel() override;

  void sendAndRecvMessage(const VoidCallback& cob,
                                  apache::thrift::transport::TMemoryBuffer* sendBuf,
                                  apache::thrift::transport::TMemoryBuffer* recvBuf) override;

  void sendMessage(const VoidCallback& cob,
                           apache::thrift::transport::TMemoryBuffer* message) override;
  void recvMessage(const VoidCallback& cob,
                           apache::thrift::transport::TMemoryBuffer* message) override;

  void finish(struct evhttp_request* req);

  // XXX
  bool good() const override { return true; }
  bool error() const override { return false; }
  bool timedOut() const override { return false; }

private:
  static void response(struct evhttp_request* req, void* arg);

  std::string host_;
  std::string path_;
  typedef std::pair<VoidCallback, apache::thrift::transport::TMemoryBuffer*> Completion;
  typedef std::queue<Completion> CompletionQueue;
  CompletionQueue completionQueue_;
  struct evhttp_connection* conn_;
};
}
}
} // apache::thrift::async

#endif // #ifndef _THRIFT_TEVHTTP_CLIENT_CHANNEL_H_