ZLMediaKit/api/source/mk_tcp_private.h
2023-12-09 16:23:51 +08:00

45 lines
1.4 KiB
C++

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