DefaultServerChooser.hpp

Go to the documentation of this file.

//
// Copyright (c) 2010-2024 60East Technologies Inc., All Rights Reserved.
//
// This computer software is owned by 60East Technologies Inc. and is
// protected by U.S. copyright laws and other laws and by international
// treaties.  This computer software is furnished by 60East Technologies
// Inc. pursuant to a written license agreement and may be used, copied,
// transmitted, and stored only in accordance with the terms of such
// license agreement and with the inclusion of the above copyright notice.
// This computer software or any other copies thereof may not be provided
// or otherwise made available to any other person.
//
// U.S. Government Restricted Rights.  This computer software: (a) was
// developed at private expense and is in all respects the proprietary
// information of 60East Technologies Inc.; (b) was not developed with
// government funds; (c) is a trade secret of 60East Technologies Inc.
// for all purposes of the Freedom of Information Act; and (d) is a
// commercial item and thus, pursuant to Section 12.212 of the Federal
// Acquisition Regulations (FAR) and DFAR Supplement Section 227.7202,
// Government's use, duplication or disclosure of the computer software
// is subject to the restrictions set forth by 60East Technologies Inc..
//

#ifndef _DEFAULTSERVERCHOOSER_H_
#define _DEFAULTSERVERCHOOSER_H_

#include <amps/ampsplusplus.hpp>
#include <amps/ServerChooser.hpp>
#include <vector>


namespace AMPS
{

  class DefaultServerChooser : public ServerChooserImpl
  {
  public:
    DefaultServerChooser() : _current(0) {}

    virtual std::string getCurrentURI()
    {
      if (_uris.size() > 0)
      {
        return _uris[_current];
      }
      return std::string();
    }

    void add(const std::string& uri_)
    {
      _uris.push_back(uri_);
    }

    template <class T>
    void addAll(const T& uriContainer_)
    {
      for (typename T::const_iterator i = uriContainer_.begin(); i != uriContainer_.end(); ++i)
      {
        _uris.push_back(*i);
      }
    }

    void remove(const std::string& /*uri_*/)
    {
      //Not permitted in DefaultServerChooser
    }

    virtual Authenticator& getCurrentAuthenticator()
    {
      return DefaultAuthenticator::instance();
    }

    virtual void reportFailure(const AMPSException& exception_,
                               const ConnectionInfo& /* info_ */)
    {
      if (strcmp(exception_.getClassName(), "DisconnectedException") != 0)
      {
        next();
      }
    }

    virtual std::string getError()
    {
      return std::string();
    }

    virtual void reportSuccess(const ConnectionInfo& /* info_ */)
    {
    }

    virtual void next()
    {
      if (_uris.size() == 0)
      {
        return;
      }
      _current = (_current + 1) % _uris.size();
    }

    ~DefaultServerChooser() {}

  private:
    std::vector<std::string> _uris;
    size_t _current;
  };

} //namespace AMPS

#endif //_DEFAULTSERVERCHOOSER_H_