libspace/include/sirikata/space/ObjectHostSession.hpp

Go to the documentation of this file.

// Copyright (c) 2011 Sirikata Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can
// be found in the LICENSE file.

#ifndef _SIRIKATA_LIBSPACE_OBJECT_HOST_SESSION_HPP_
#define _SIRIKATA_LIBSPACE_OBJECT_HOST_SESSION_HPP_

#include <sirikata/space/Platform.hpp>
#include <sirikata/core/util/ListenerProvider.hpp>
#include <sirikata/core/ohdp/SST.hpp>
#include <sirikata/space/SpaceContext.hpp>

namespace Sirikata {

// These classes are thin wrappers around ObjectHostConnectionManager and it's
// affiliated classes for tracking sessions

class SIRIKATA_SPACE_EXPORT ObjectHostSession {
  public:
    ObjectHostSession(const OHDP::NodeID& id, OHDPSST::Stream::Ptr strm)
        : mID(id),
        mSSTStream(strm),
        mSeqNo(new SeqNo())
        {
        }

    ~ObjectHostSession() {
        if (mSSTStream) mSSTStream->close(true);
    }

    const OHDP::NodeID& id() const { return mID; }
    const OHDPSST::Stream::Ptr& stream() const { return mSSTStream; }
    const SeqNoPtr& seqNoPtr() const { return mSeqNo; }

  private:
    OHDP::NodeID mID;
    OHDPSST::Stream::Ptr mSSTStream;
    SeqNoPtr mSeqNo;
};
typedef std::tr1::shared_ptr<ObjectHostSession> ObjectHostSessionPtr;

class SIRIKATA_SPACE_EXPORT ObjectHostSessionListener {
  public:
    virtual ~ObjectHostSessionListener() {}

    virtual void onObjectHostSession(const OHDP::NodeID& id, ObjectHostSessionPtr) {}
    virtual void onObjectHostSessionEnded(const OHDP::NodeID& id) {}
};

class ObjectHostSessionManager : public Provider<ObjectHostSessionListener*> {
  public:
    ObjectHostSessionManager(SpaceContext* ctx) {
        ctx->mObjectHostSessionManager = this;
    }

    // Owner interface
    void fireObjectHostSession(const OHDP::NodeID& id, OHDPSST::Stream::Ptr oh_stream) {
        ObjectHostSessionPtr sess(new ObjectHostSession(id, oh_stream));
        mOHSessions[id] = sess;
        notify(&ObjectHostSessionListener::onObjectHostSession, id, sess);

    }
    void fireObjectHostSessionEnded(const OHDP::NodeID& id) {
        notify(&ObjectHostSessionListener::onObjectHostSessionEnded, id);
        ObjectHostSessionMap::const_iterator it = mOHSessions.find(id);
        if (it != mOHSessions.end())
            mOHSessions.erase(it);
    }


    // User interface
    ObjectHostSessionPtr getSession(const OHDP::NodeID& id) {
        ObjectHostSessionMap::const_iterator it = mOHSessions.find(id);
        if (it == mOHSessions.end()) return ObjectHostSessionPtr();
        return it->second;
    }

  private:
    typedef std::tr1::unordered_map<OHDP::NodeID, ObjectHostSessionPtr, OHDP::NodeID::Hasher> ObjectHostSessionMap;
    ObjectHostSessionMap mOHSessions;
};

} // namespace Sirikata

#endif //_SIRIKATA_LIBSPACE_OBJECT_HOST_SESSION_HPP_