heron/instance/src/cpp/boltimpl/tuple-impl.h

/*
 * Copyright 2017 Twitter, Inc.
 *
 * Licensed 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 HERON_INSTANCE_BOLT_TUPLEIMPL_H_
#define HERON_INSTANCE_BOLT_TUPLEIMPL_H_

#include <string>

#include "proto/messages.h"
#include "network/network.h"
#include "basics/basics.h"

#include "tuple/tuple.h"
#include "topology/task-context.h"
#include "serializer/ipluggable-serializer.h"

#include "slave/task-context-impl.h"

namespace heron {
namespace instance {

class TupleImpl : public api::tuple::Tuple {
 public:
  TupleImpl(std::shared_ptr<api::serializer::IPluggableSerializer> serializer,
            std::shared_ptr<TaskContextImpl> taskContext,
            const proto::api::StreamId& stream,
            std::shared_ptr<const proto::system::HeronDataTuple> tup);
  virtual ~TupleImpl();

  // All the interfaces

  /**
   * Returns the number of fields in this tuple.
   */
  virtual int size() const;

  /**
   * Returns the position of the specified field in this tuple.
   */
  virtual int fieldIndex(const std::string& field) const;

  /**
   * Returns true if this tuple contains the specified name of the field.
   */
  virtual bool contains(const std::string& field) const;

  /**
   * Gets the names of the fields in this tuple.
   */
  virtual api::tuple::Fields getFields() const;

  /**
   * Gets the id of the component that created this tuple.
   */
  virtual const std::string& getSourceComponent() const;

  /**
   * Gets the id of the task that created this tuple.
   */
  virtual int getSourceTask() const;

  /**
   * Gets the id of the stream that this tuple was emitted to.
   */
  virtual const std::string& getSourceStreamId() const;

  virtual const std::string& getUnserializedValue(int index) const;

  // For our implementation
  std::shared_ptr<const proto::system::HeronDataTuple> getDataTuple() {
    return tuple_;
  }

  // For our implementation
  int64_t getCreationTimeNs() const {
    return creationTime_;
  }

 private:
  std::shared_ptr<TaskContextImpl> taskContext_;
  proto::api::StreamId streamInfo_;
  std::shared_ptr<const proto::system::HeronDataTuple> tuple_;
  int64_t creationTime_;
};

}  // namespace instance
}  // namespace heron

#endif  // HERON_INSTANCE_BOLT_TUPLEIMPL_H_