Scale Stack 0.3 Developer Documentation

scalestack/proxy/client.cc

Go to the documentation of this file.

/*
 * Scale Stack
 *
 * Copyright 2010 Eric Day
 *
 * 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.
 */

#include "config.h"

#include <scalestack/kernel/module.h>
#include <scalestack/kernel/option.h>
#include <scalestack/proxy/client.h>
#include <scalestack/proxy/server.h>
#include <scalestack/proxy/server_service.h>

namespace scalestack
{
namespace proxy
{

/*
 * Public methods.
 */

client::client(kernel::module& module, server_service& server_service):
  network::stream(module),
  _binding(),
  _shutdown(),
  _shutdown_write(),
  _size(),
  _buffer(),
  _server(),
  _server_service(server_service)
{
}

void client::shutdown(void)
{
  _shutdown = true;
  if (_server == NULL)
  {
    if (_binding)
    {
      if (_server_service.remove_client(this))
        stop();
      else
        run_now();
    }
    else
      stop();
  }
  else
    run_now();
}

void client::timer_expired(void)
{
  shutdown();
}

void client::run(void)
{
  if (_server != NULL)
    _server->run_now();
}

void client::connected(void)
{
  _binding = true;
  _server_service.add_client(this);
  set_timer(_module.get_option("server_timeout").get_size_value() * 1000);
}

void client::set_server(server* server)
{
  set_timer(0);
  _server = server;
  if (_shutdown)
    run_now();
  else
    continue_write();
}

size_t client::read(uint8_t* buffer, size_t size)
{
  _size = size;
  _buffer = buffer;
  return _server_write();
}

void client::read_eof(void)
{
  if (!_shutdown)
  {
    if (_size == 0)
      _server->shutdown_write();
    else
      _shutdown_write = true;
  }
}

void client::flush_write(void)
{
  if (!_shutdown)
    _server->continue_write();
}

void client::continue_write(void)
{
  consume(_server_write());
}

/*
 * Private methods.
 */

size_t client::_server_write(void)
{
  if (_server == NULL || _shutdown)
    return 0;

  size_t written = _server->write(_buffer, _size, true);
  _size -= written;
  _buffer += written;

  if (_shutdown_write && _size == 0)
    _server->shutdown_write();

  return written;
}

} /* namespace proxy */
} /* namespace scalestack */

---