responder_lookup.h

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the Apache 2.0 License.
#pragma once
 
#include "ccf/odata_error.h"
#include "ccf/pal/locking.h"
#include "http2_types.h"
 
#include <unordered_map>
 
namespace http
{
  class ResponderLookup
  {
  protected:
    using ByStream = std::
      unordered_map<http2::StreamId, std::shared_ptr<ccf::http::HTTPResponder>>;
 
    std::unordered_map<::tcp::ConnID, ByStream> all_responders;
 
    // Responder lookup is shared by all HTTP sessions
    ccf::pal::Mutex lock;
 
  public:
    std::shared_ptr<ccf::http::HTTPResponder> lookup_responder(
      ::tcp::ConnID session_id, http2::StreamId stream_id)
    {
      std::unique_lock<ccf::pal::Mutex> guard(lock);
      auto conn_it = all_responders.find(session_id);
      if (conn_it != all_responders.end())
      {
        auto& by_stream = conn_it->second;
        auto stream_it = by_stream.find(stream_id);
        if (stream_it != by_stream.end())
        {
          return stream_it->second;
        }
      }
 
      return nullptr;
    }
 
    void add_responder(
      ::tcp::ConnID session_id,
      http2::StreamId stream_id,
      std::shared_ptr<ccf::http::HTTPResponder> responder)
    {
      std::unique_lock<ccf::pal::Mutex> guard(lock);
      all_responders[session_id][stream_id] = responder;
    }
 
    void cleanup_responders(::tcp::ConnID session_id)
    {
      std::unique_lock<ccf::pal::Mutex> guard(lock);
      all_responders.erase(session_id);
    }
  };
}