#include "stdafx.h"
#include <ctime>
#include <iostream>
#include <string>
#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/asio.hpp>

using boost::asio::ip::tcp;

std::string make_daytime_string()
{
    using namespace std; // For time_t, time and ctime;
    time_t now = time(0);
    return ctime(&now);
}


//使用use shared_ptr 和 enable_shared_from_this來確保其他程序調用tcp_connection對象之前，它都是存在的
class tcp_connection
    : public boost::enable_shared_from_this<tcp_connection>
{
public:
    typedef boost::shared_ptr<tcp_connection> pointer;

    static pointer create(boost::asio::io_service& io_service)
    {
        return pointer(new tcp_connection(io_service));
    }

    tcp::socket& socket()
    {
        return socket_;
    }

    //在這個函數里， boost::asio::async_write()用來服務于給客戶端發送數據。使用 boost::asio::async_write()而不是 ip::tcp::socket::async_write_some(),來保證整塊數據都被發送出去了
    void start()
    {
        //要發送的數據存儲在message_中，因為我們需要在異步操作執行完之前，保持這個數據有效
        message_ = make_daytime_string();

        boost::asio::async_write(socket_, boost::asio::buffer(message_),
            boost::bind(&tcp_connection::handle_write, shared_from_this(),
                boost::asio::placeholders::error,
                boost::asio::placeholders::bytes_transferred));
        //去除不需要的參數，可以寫成這樣
        /*boost::asio::async_write(socket_, boost::asio::buffer(message_),
            boost::bind(&tcp_connection::handle_write, shared_from_this()));*/
    }

private:
    //此時所有與這個客戶端相關的操作，都由handle_write()負責
    //error和bytes_transferred參數實際上沒有使用，如果想的話，可以直接移除掉
    tcp_connection(boost::asio::io_service& io_service)
        : socket_(io_service)
    {
    }

    void handle_write(const boost::system::error_code& /*error*/,
        size_t /*bytes_transferred*/)
    {
    }
    //去除不需要的參數，可以寫成下面這樣
    /*void handle_write()
    {
    }*/

    tcp::socket socket_;
    std::string message_;
};


//tcp server 類
class tcp_server
{
public:
    //構造函數初始化了一個接收器監聽本地的13端口
    tcp_server(boost::asio::io_service& io_service)
        : acceptor_(io_service, tcp::endpoint(tcp::v4(), 6300))
    {
        start_accept();
    }

private:
    //這個函數建立了一個socket并且初始化了一個異步的操作等待新連接
    void start_accept()
    {
        tcp_connection::pointer new_connection =
            tcp_connection::create(acceptor_.get_io_service());

        acceptor_.async_accept(new_connection->socket(),
            boost::bind(&tcp_server::handle_accept, this, new_connection,
                boost::asio::placeholders::error));
    }

    //這個函數在 start_accept()初始化異步接收操作之后被調用，它服務于客戶端請求，并且調用 start_accept()初始化下一個接收操作
    void handle_accept(tcp_connection::pointer new_connection,
        const boost::system::error_code& error)
    {
        if (!error)
        {
            new_connection->start();
        }

        start_accept();
    }

    tcp::acceptor acceptor_;
};

int main()
{
    try
    {
        //創建了一個server用于接收連接進來的客戶端
        boost::asio::io_service io_service;
        tcp_server server(io_service);
        io_service.run();
    }
    catch (std::exception& e)
    {
        std::cerr << e.what() << std::endl;
    }

    return 0;
}