HPACKEncoderBase.h

Go to the documentation of this file.

/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */
#pragma once

#include <proxygen/lib/http/codec/compress/HPACKContext.h>
#include <proxygen/lib/http/codec/compress/HPACKEncodeBuffer.h>
#include <proxygen/lib/http/codec/compress/HeaderIndexingStrategy.h>

namespace proxygen {

class HPACKEncoderBase {
 public:

  explicit HPACKEncoderBase(bool huffman)
    : streamBuffer_(kBufferGrowth, huffman) {
    indexingStrat_ = HeaderIndexingStrategy::getDefaultInstance();
  }

  static const uint32_t kBufferGrowth = 4000;

  void setHeaderTableSize(HeaderTable& table, uint32_t size) {
    if (size != table.capacity()) {
      CHECK(table.setCapacity(size));
      pendingContextUpdate_ = true;
    }
  }

  void setHeaderIndexingStrategy(const HeaderIndexingStrategy* indexingStrat) {
    indexingStrat_ = indexingStrat;
  }
  const HeaderIndexingStrategy* getHeaderIndexingStrategy() const {
    return indexingStrat_;
  }

 protected:

  void handlePendingContextUpdate(HPACKEncodeBuffer& buf,
                                  uint32_t tableCapacity);

  const HeaderIndexingStrategy* indexingStrat_;
  HPACKEncodeBuffer streamBuffer_;
  bool pendingContextUpdate_{false};
};


}