abstract_storage.hpp

#pragma once

#include "base/message.hpp"

#include "glog/logging.h"

namespace csci5570 {

/*
 * Implement using the template method and dispatch the SubAdd/SubGet to subclasses.
 */
class AbstractStorage {
 public:
  void Add(Message& msg) {
    CHECK(msg.data.size() == 2);
    auto typed_keys = third_party::SArray<Key>(msg.data[0]);
    SubAdd(typed_keys, msg.data[1]);
  }
  Message Get(Message& msg) {
    CHECK(msg.data.size() == 1);
    auto typed_keys = third_party::SArray<Key>(msg.data[0]);
    Message reply;
    reply.meta.recver = msg.meta.sender;
    reply.meta.sender = msg.meta.recver;
    reply.meta.flag = msg.meta.flag;
    reply.meta.model_id = msg.meta.model_id;
    third_party::SArray<Key> reply_keys(typed_keys);
    third_party::SArray<char> reply_vals = SubGet(reply_keys);
    reply.AddData<Key>(reply_keys);
    reply.AddData<char>(reply_vals);
    return reply;
  }
  
  // Add the typed_keys and typed_vals to kvstore
  virtual void SubAdd(const third_party::SArray<Key>& typed_keys, 
      const third_party::SArray<char>& vals) = 0;

  // Retrieve the vals according to the typed_keys
  virtual third_party::SArray<char> SubGet(const third_party::SArray<Key>& typed_keys) = 0;

  virtual void FinishIter() = 0;
};

}  // namespace csci5570