1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00

Migrate is_never_timeout to utility.

This commit is contained in:
winlin 2019-05-09 08:42:53 +08:00
parent 1b37b90b8f
commit ccba55cdd8
8 changed files with 8 additions and 42 deletions

View file

@ -424,13 +424,6 @@ srs_error_t SimpleSocketStream::writev(const iovec *iov, int iov_size, ssize_t*
return srs_success; return srs_success;
} }
// Interface ISrsProtocolReadWriter
bool SimpleSocketStream::is_never_timeout(srs_utime_t tm)
{
srs_assert(io);
return srs_hijack_io_is_never_timeout(io, tm);
}
srs_error_t SimpleSocketStream::read_fully(void* buf, size_t size, ssize_t* nread) srs_error_t SimpleSocketStream::read_fully(void* buf, size_t size, ssize_t* nread)
{ {
srs_assert(io); srs_assert(io);

View file

@ -65,7 +65,6 @@ public:
virtual srs_error_t writev(const iovec *iov, int iov_size, ssize_t* nwrite); virtual srs_error_t writev(const iovec *iov, int iov_size, ssize_t* nwrite);
// Interface ISrsProtocolReadWriter // Interface ISrsProtocolReadWriter
public: public:
virtual bool is_never_timeout(srs_utime_t tm);
virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread); virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread);
virtual srs_error_t write(void* buf, size_t size, ssize_t* nwrite); virtual srs_error_t write(void* buf, size_t size, ssize_t* nwrite);
}; };

View file

@ -54,8 +54,6 @@
* +--+-----------------------------+-+ * +--+-----------------------------+-+
* | IProtocolReadWriter | * | IProtocolReadWriter |
* +----------------------------------+ * +----------------------------------+
* | + is_never_timeout() |
* +----------------------------------+
*/ */
/** /**
@ -121,10 +119,6 @@ class ISrsProtocolReadWriter : virtual public ISrsProtocolReader, virtual public
public: public:
ISrsProtocolReadWriter(); ISrsProtocolReadWriter();
virtual ~ISrsProtocolReadWriter(); virtual ~ISrsProtocolReadWriter();
// For protocol
public:
// Whether the specified tm in srs_utime_t is never timeout.
virtual bool is_never_timeout(srs_utime_t tm) = 0;
}; };
#endif #endif

View file

@ -362,6 +362,11 @@ ssize_t srs_read(srs_netfd_t stfd, void *buf, size_t nbyte, srs_utime_t timeout)
return st_read((st_netfd_t)stfd, buf, nbyte, (st_utime_t)timeout); return st_read((st_netfd_t)stfd, buf, nbyte, (st_utime_t)timeout);
} }
bool srs_is_never_timeout(srs_utime_t tm)
{
return tm == SRS_UTIME_NO_TIMEOUT;
}
SrsStSocket::SrsStSocket() SrsStSocket::SrsStSocket()
{ {
stfd = NULL; stfd = NULL;
@ -379,11 +384,6 @@ srs_error_t SrsStSocket::initialize(srs_netfd_t fd)
return srs_success; return srs_success;
} }
bool SrsStSocket::is_never_timeout(srs_utime_t tm)
{
return tm == SRS_UTIME_NO_TIMEOUT;
}
void SrsStSocket::set_recv_timeout(srs_utime_t tm) void SrsStSocket::set_recv_timeout(srs_utime_t tm)
{ {
rtm = tm; rtm = tm;
@ -593,11 +593,6 @@ void SrsTcpClient::close()
srs_close_stfd(stfd); srs_close_stfd(stfd);
} }
bool SrsTcpClient::is_never_timeout(srs_utime_t tm)
{
return io->is_never_timeout(tm);
}
void SrsTcpClient::set_recv_timeout(srs_utime_t tm) void SrsTcpClient::set_recv_timeout(srs_utime_t tm)
{ {
io->set_recv_timeout(tm); io->set_recv_timeout(tm);

View file

@ -130,7 +130,6 @@ public:
// Initialize the socket with stfd, user must manage it. // Initialize the socket with stfd, user must manage it.
virtual srs_error_t initialize(srs_netfd_t fd); virtual srs_error_t initialize(srs_netfd_t fd);
public: public:
virtual bool is_never_timeout(srs_utime_t tm);
virtual void set_recv_timeout(srs_utime_t tm); virtual void set_recv_timeout(srs_utime_t tm);
virtual srs_utime_t get_recv_timeout(); virtual srs_utime_t get_recv_timeout();
virtual void set_send_timeout(srs_utime_t tm); virtual void set_send_timeout(srs_utime_t tm);
@ -181,7 +180,6 @@ private:
virtual void close(); virtual void close();
// Interface ISrsProtocolReadWriter // Interface ISrsProtocolReadWriter
public: public:
virtual bool is_never_timeout(srs_utime_t tm);
virtual void set_recv_timeout(srs_utime_t tm); virtual void set_recv_timeout(srs_utime_t tm);
virtual srs_utime_t get_recv_timeout(); virtual srs_utime_t get_recv_timeout();
virtual void set_send_timeout(srs_utime_t tm); virtual void set_send_timeout(srs_utime_t tm);

View file

@ -43,11 +43,6 @@ MockEmptyIO::~MockEmptyIO()
{ {
} }
bool MockEmptyIO::is_never_timeout(srs_utime_t /*tm*/)
{
return true;
}
srs_error_t MockEmptyIO::read_fully(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/) srs_error_t MockEmptyIO::read_fully(void* /*buf*/, size_t /*size*/, ssize_t* /*nread*/)
{ {
return srs_success; return srs_success;
@ -112,11 +107,6 @@ MockBufferIO* MockBufferIO::append(string data)
return this; return this;
} }
bool MockBufferIO::is_never_timeout(srs_utime_t tm)
{
return tm == SRS_UTIME_NO_TIMEOUT;
}
srs_error_t MockBufferIO::read_fully(void* buf, size_t size, ssize_t* nread) srs_error_t MockBufferIO::read_fully(void* buf, size_t size, ssize_t* nread)
{ {
if (in_buffer.length() < (int)size) { if (in_buffer.length() < (int)size) {

View file

@ -46,9 +46,6 @@ class MockEmptyIO : public ISrsProtocolReadWriter
public: public:
MockEmptyIO(); MockEmptyIO();
virtual ~MockEmptyIO(); virtual ~MockEmptyIO();
// for protocol
public:
virtual bool is_never_timeout(srs_utime_t tm);
// for handshake. // for handshake.
public: public:
virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread); virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread);
@ -87,9 +84,6 @@ public:
virtual ~MockBufferIO(); virtual ~MockBufferIO();
public: public:
virtual MockBufferIO* append(std::string data); virtual MockBufferIO* append(std::string data);
// for protocol
public:
virtual bool is_never_timeout(srs_utime_t tm);
// for handshake. // for handshake.
public: public:
virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread); virtual srs_error_t read_fully(void* buf, size_t size, ssize_t* nread);

View file

@ -39,6 +39,9 @@ VOID TEST(ServiceTimeTest, TimeUnit)
EXPECT_EQ(1000*1000, SRS_UTIME_SECONDS); EXPECT_EQ(1000*1000, SRS_UTIME_SECONDS);
EXPECT_EQ(60*1000*1000, SRS_UTIME_MINUTES); EXPECT_EQ(60*1000*1000, SRS_UTIME_MINUTES);
EXPECT_EQ(3600*1000*1000LL, SRS_UTIME_HOURS); EXPECT_EQ(3600*1000*1000LL, SRS_UTIME_HOURS);
EXPECT_TRUE(srs_is_never_timeout(SRS_UTIME_NO_TIMEOUT));
EXPECT_FALSE(srs_is_never_timeout(0));
} }
#define MOCK_LISTEN_HOST "127.0.0.1" #define MOCK_LISTEN_HOST "127.0.0.1"