ZLMediaKit/api/source/mk_tcp_private.h

/*
 * Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
 *
 * Use of this source code is governed by MIT-like license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */

#ifndef MK_TCP_PRIVATE_H
#define MK_TCP_PRIVATE_H

#include "mk_tcp.h"
#include "Network/TcpClient.h"
#include "Network/Session.h"

class TcpClientForC : public toolkit::TcpClient {
public:
    using Ptr = std::shared_ptr<TcpClientForC>;
    TcpClientForC(mk_tcp_client_events *events) ;
    ~TcpClientForC() override ;
    void onRecv(const toolkit::Buffer::Ptr &pBuf) override;
    void onError(const toolkit::SockException &ex) override;
    void onManager() override;
    void onConnect(const toolkit::SockException &ex) override;
    void setClient(mk_tcp_client client);
    std::shared_ptr<void> _user_data;
private:
    mk_tcp_client_events _events;
    mk_tcp_client _client;
};

class SessionForC : public toolkit::Session {
public:
    SessionForC(const toolkit::Socket::Ptr &pSock);
    void onRecv(const toolkit::Buffer::Ptr &buffer) override;
    void onError(const toolkit::SockException &err) override;
    void onManager() override;
    std::shared_ptr<void> _user_data;
    uint16_t _local_port;
};

#endif //MK_TCP_PRIVATE_H