Scale Stack 0.3 Developer Documentation

scalestack/database_proxy/client.h

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.
 */

#ifndef SCALESTACK_DATABASEPROXY_CLIENT_H
#define SCALESTACK_DATABASEPROXY_CLIENT_H

#include <scalestack/network/stream.h>

namespace scalestack
{
namespace database_proxy
{

class client_service;
class connection;

class client: public network::stream
{
public:

  client(client_service& creator);

  void set_connection(database_proxy::connection* bound_connection,
                      bool need_handshake);

  void yield(void);

  void connected(void);

  size_t read(uint8_t* buffer, size_t size);

  void continue_write(void);

  void flush_write(void);

  void error(void);

  void run(void);

private:

  client(const client&);

  client& operator=(const client&);

  size_t _connection_write(void);

  enum
  {
    SERVER_HANDSHAKE_PACKET,
    SERVER_HANDSHAKE_FLUSH,
    RESULT_PACKET,
    RESULT_FLUSH,
    COLUMN_PACKET,
    COLUMN_FLUSH,
    ROW_PACKET,
    ROW_FLUSH
  } _state, _next_state;
  bool _need_yield;
  bool _shutdown;
  size_t _size;
  size_t _packet_size;
  uint8_t* _buffer;
  database_proxy::connection* _connection;
};

} /* namespace database_proxy */
} /* namespace scalestack */

#endif /* SCALESTACK_DATABASEPROXY_CLIENT_H */

---