OperationSetState.cc

Go to the documentation of this file.

/*
 * Copyright (C) 2007-2015 Hypertable, Inc.
 *
 * This file is part of Hypertable.
 *
 * Hypertable is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; version 3 of the
 * License, or any later version.
 *
 * Hypertable is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

#include <Common/Compat.h>

#include "DispatchHandlerOperationSetState.h"
#include "OperationSetState.h"
#include "Utility.h"

#include <Hypertable/Lib/Key.h>

#include <Hyperspace/Session.h>

#include <Common/Error.h>
#include <Common/FailureInducer.h>
#include <Common/ScopeGuard.h>
#include <Common/Serialization.h>

#include <boost/algorithm/string.hpp>

using namespace Hypertable;
using namespace Hypertable::Lib;

OperationSetState::OperationSetState(ContextPtr &context)
  : Operation(context, MetaLog::EntityType::OPERATION_SET) {
  m_exclusivities.insert("SET VARIABLES");
}

OperationSetState::OperationSetState(ContextPtr &context,
                                     const MetaLog::EntityHeader &header_)
  : Operation(context, header_) {
}

OperationSetState::OperationSetState(ContextPtr &context, EventPtr &event)
  : Operation(context, event, MetaLog::EntityType::OPERATION_SET) {
  const uint8_t *ptr = event->payload;
  size_t remaining = event->payload_len;
  m_params.decode(&ptr, &remaining);
  m_context->system_state->admin_set(m_params.specs());
  m_exclusivities.insert("SET VARIABLES");
}

void OperationSetState::execute() {
  int32_t state = get_state();
  DispatchHandlerOperationSetState dispatch_handler(m_context);

  HT_INFOF("Entering SetState-%lld() state=%s",
           (Lld)header.id, OperationState::get_text(state));

  switch (state) {

  case OperationState::INITIAL:

    // Check for and send notifications
    {
      std::vector<NotificationMessage> notifications;
      if (m_context->system_state->get_notifications(notifications)) {
        for (auto &msg : notifications)
          m_context->notification_hook(msg.subject, msg.body);
      }
    }

    m_context->system_state->get(m_specs, &m_generation);

    HT_INFOF("specs = %s", SystemVariable::specs_to_string(m_specs).c_str());

    set_state(OperationState::STARTED);
    {
      std::vector<MetaLog::EntityPtr> entities;
      entities.push_back(shared_from_this());
      entities.push_back(m_context->system_state);
      m_context->mml_writer->record_state(entities);
    }

  case OperationState::STARTED:

    {
      StringSet locations;
      std::vector<RangeServerConnectionPtr> servers;
      m_context->rsc_manager->get_servers(servers);
      if (servers.empty()) {
        complete_ok();
        break;
      }
      for (auto &rsc : servers)
        locations.insert(rsc->location());
      dispatch_handler.initialize();
      dispatch_handler.DispatchHandlerOperation::start(locations);
    }

    if (!dispatch_handler.wait_for_completion()) {
      std::set<DispatchHandlerOperation::Result> results;
      dispatch_handler.get_results(results);
      for (auto &result : results) {
        if (result.error != Error::OK)
          HT_WARNF("Problem setting state variables at %s - %s",
                   result.location.c_str(), Error::get_text(result.error));
      }
    }

    complete_ok();
    break;

  default:
    HT_FATALF("Unrecognized state %d", state);
  }

  HT_INFOF("Leaving SetState-%lld", (Lld)header.id);
}


void OperationSetState::display_state(std::ostream &os) {
  bool first = true;
  for (auto &spec : m_specs) {
    if (!first)
      os << ",";
    os << SystemVariable::code_to_string(spec.code) << "=" << (spec.value ? "true" : "false");
    first = false;
  }
  os << " ";
}

uint8_t OperationSetState::encoding_version_state() const {
  return 1;
}

size_t OperationSetState::encoded_length_state() const {
  size_t length = m_params.encoded_length() + 12;
  for (auto &spec : m_specs)
    length += spec.encoded_length();
  return length;
}

void OperationSetState::encode_state(uint8_t **bufp) const {
  m_params.encode(bufp);
  Serialization::encode_i64(bufp, m_generation);
  Serialization::encode_i32(bufp, m_specs.size());
  for (auto &spec : m_specs)
    spec.encode(bufp);
}

void OperationSetState::decode_state(uint8_t version, const uint8_t **bufp, size_t *remainp) {
  m_params.decode(bufp, remainp);
  m_generation = Serialization::decode_i64(bufp, remainp);
  size_t count = (size_t)Serialization::decode_i32(bufp, remainp);
  for (size_t i=0; i<count; i++) {
    SystemVariable::Spec spec;
    spec.decode(bufp, remainp);
    m_specs.push_back(spec);
  }
}

void OperationSetState::decode_state_old(uint8_t version, const uint8_t **bufp, size_t *remainp) {
  SystemVariable::Spec spec;
  m_specs.clear();
  if (version == 0)
    Serialization::decode_i32(bufp, remainp); // skip old version
  m_generation = Serialization::decode_i64(bufp, remainp);
  int32_t count = Serialization::decode_i32(bufp, remainp);
  for (int32_t i=0; i<count; i++) {
    spec.code  = Serialization::decode_i32(bufp, remainp);
    spec.value = Serialization::decode_bool(bufp, remainp);
    m_specs.push_back(spec);
  }
}

const String OperationSetState::name() {
  return "OperationSetState";
}

const String OperationSetState::label() {
  return String("SetState ") + SystemVariable::specs_to_string(m_specs);
}