cache.hpp

Go to the documentation of this file.

// 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 __APPC_PROVISIONER_CACHE_HPP__
#define __APPC_PROVISIONER_CACHE_HPP__

#include <map>
#include <string>

#include <process/owned.hpp>

#include <stout/hashmap.hpp>

#include <mesos/mesos.hpp>

#include <mesos/appc/spec.hpp>

namespace mesos {
namespace internal {
namespace slave {
namespace appc {

class Cache
{
public:
  static Try<process::Owned<Cache>> create(const Path& storeDir);


  Try<Nothing> recover();


  Try<Nothing> add(const std::string& imageId);


  Option<std::string> find(const Image::Appc& image) const;

private:
  struct Key
  {
    Key(const Image::Appc& image);

    Key(const std::string& name,
        const std::map<std::string, std::string>& labels);

    bool operator==(const Key& other) const;

    std::string name;
    std::map<std::string, std::string> labels;
  };

  struct KeyHasher
  {
    size_t operator()(const Key& key) const;
  };

  Cache(const Path& imagesDir);

  const Path storeDir;

  // Mappings: Key -> image id.
  hashmap<Cache::Key, std::string, KeyHasher> imageIds;
};

} // namespace appc {
} // namespace slave {
} // namespace internal {
} // namespace mesos {

#endif // __APPC_PROVISIONER_CACHE_HPP__