// Copyright (c) 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef NET_SPDY_SPDY_HEADER_INDEXING_H_
#define NET_SPDY_SPDY_HEADER_INDEXING_H_

#include <stdint.h>
#include <memory>
#include <string>
#include <unordered_set>
#include <utility>

#include "base/strings/string_piece.h"
#include "net/base/net_export.h"

namespace net {

namespace test {
class HeaderIndexingPeer;
}

NET_EXPORT_PRIVATE extern int32_t FLAGS_gfe_spdy_indexing_set_bound;
NET_EXPORT_PRIVATE extern int32_t FLAGS_gfe_spdy_tracking_set_bound;

// Maintain two headers sets: Indexing set and tracking
// set. Call ShouldIndex() for each header to decide if to index it. If for some
// connections, we decide to index all headers, we may still want to call
// UpdateSets to log the headers into both sets.
class NET_EXPORT HeaderIndexing {
 public:
  using HeaderSet = std::unordered_set<std::string>;

  HeaderIndexing();
  ~HeaderIndexing();

  void CreateInitIndexingHeaders();

  // Decide if a header should be indexed. We only use |header|. Add |value| to
  // be consistent with HPACK indexing policy interface.
  bool ShouldIndex(base::StringPiece header, base::StringPiece value);

  // Not to make the indexing decision but to update sets.
  void UpdateSets(base::StringPiece header, base::StringPiece value) {
    update_only_header_count_++;
    ShouldIndex(header, value);
  }

  int64_t total_header_count() { return total_header_count_; }
  int64_t update_only_header_count() { return update_only_header_count_; }
  int64_t missed_headers_in_indexing() { return missed_header_in_indexing_; }
  int64_t missed_headers_in_tracking() { return missed_header_in_tracking_; }

 private:
  friend class test::HeaderIndexingPeer;
  void TryInsertHeader(std::string&& header, HeaderSet* set, size_t bound);
  // Headers to index.
  HeaderSet indexing_set_;
  // Headers seen so far.
  HeaderSet tracking_set_;
  const size_t indexing_set_bound_;
  const size_t tracking_set_bound_;
  int64_t total_header_count_ = 0;
  int64_t update_only_header_count_ = 0;
  int64_t missed_header_in_indexing_ = 0;
  int64_t missed_header_in_tracking_ = 0;
};

}  // namespace net

#endif  // NET_SPDY_SPDY_HEADER_INDEXING_H_