[thirdparty/squid.git] / src / DiskIO / DiskThreads / DiskThreadsDiskFile.h

/*
 * Copyright (C) 1996-2021 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */

/* DEBUG: section 79    Disk IO Routines */

#ifndef SQUID_DISKTHREADSDISKFILE_H
#define SQUID_DISKTHREADSDISKFILE_H

#include "cbdata.h"
#include "DiskIO/DiskFile.h"
#include "DiskThreads.h"
#include "typedefs.h" //for DWCB

class DiskThreadsDiskFile : public DiskFile
{
    CBDATA_CLASS(DiskThreadsDiskFile);

public:
    DiskThreadsDiskFile(char const *path);
    ~DiskThreadsDiskFile();
    virtual void open(int flags, mode_t mode, RefCount<IORequestor> callback);
    virtual void create(int flags, mode_t mode, RefCount<IORequestor> callback);
    virtual void read(ReadRequest *);
    virtual void write(WriteRequest *);
    virtual void close();
    virtual bool error() const;
    virtual int getFD() const { return fd;}

    virtual bool canRead() const;
    virtual bool canWrite() const;
    virtual bool ioInProgress() const;

private:
#if ASYNC_READ

    static AIOCB ReadDone;
#else

    static DRCB ReadDone;
#endif
#if ASYNC_WRITE

    static AIOCB WriteDone;
#else

    static DWCB WriteDone;
#endif

    int fd = -1;
    bool errorOccured = false;
    char const *path_ = nullptr;
    size_t inProgressIOs = 0;
    static AIOCB OpenDone;
    void openDone(int fd, const char *buf, int aio_return, int aio_errno);
    RefCount<IORequestor> ioRequestor;
    void doClose();

    void readDone(int fd, const char *buf, int len, int errflag, RefCount<ReadRequest> request);
    void writeDone(int fd, int errflag, size_t len, RefCount<WriteRequest> request);
};

#include "DiskIO/ReadRequest.h"

template <class RT>
class IoResult
{
    CBDATA_CLASS(IoResult);

public:
    IoResult(RefCount<DiskThreadsDiskFile> aFile, RefCount<RT> aRequest) : file(aFile), request(aRequest) {}

    RefCount<DiskThreadsDiskFile> file;
    RefCount<RT> request;
};

template <class RT>
IoResult<RT>
IOResult(RefCount<RT> aRequest, RefCount<DiskThreadsDiskFile> aFile) { return IoResult<RT>(aFile, aRequest);}

#endif /* SQUID_DISKTHREADSDISKFILE_H */
Commit	Line	Data
	1	/*
	2	* Copyright (C) 1996-2021 The Squid Software Foundation and contributors
	3	*
	4	* Squid software is distributed under GPLv2+ license and includes
	5	* contributions from numerous individuals and organizations.
	6	* Please see the COPYING and CONTRIBUTORS files for details.
	7	*/
	8
	9	/* DEBUG: section 79 Disk IO Routines */
	10
	11	#ifndef SQUID_DISKTHREADSDISKFILE_H
	12	#define SQUID_DISKTHREADSDISKFILE_H
	13
	14	#include "cbdata.h"
	15	#include "DiskIO/DiskFile.h"
	16	#include "DiskThreads.h"
	17	#include "typedefs.h" //for DWCB
	18
	19	class DiskThreadsDiskFile : public DiskFile
	20	{
	21	CBDATA_CLASS(DiskThreadsDiskFile);
	22
	23	public:
	24	DiskThreadsDiskFile(char const *path);
	25	~DiskThreadsDiskFile();
	26	virtual void open(int flags, mode_t mode, RefCount<IORequestor> callback);
	27	virtual void create(int flags, mode_t mode, RefCount<IORequestor> callback);
	28	virtual void read(ReadRequest *);
	29	virtual void write(WriteRequest *);
	30	virtual void close();
	31	virtual bool error() const;
	32	virtual int getFD() const { return fd;}
	33
	34	virtual bool canRead() const;
	35	virtual bool canWrite() const;
	36	virtual bool ioInProgress() const;
	37
	38	private:
	39	#if ASYNC_READ
	40
	41	static AIOCB ReadDone;
	42	#else
	43
	44	static DRCB ReadDone;
	45	#endif
	46	#if ASYNC_WRITE
	47
	48	static AIOCB WriteDone;
	49	#else
	50
	51	static DWCB WriteDone;
	52	#endif
	53
	54	int fd = -1;
	55	bool errorOccured = false;
	56	char const *path_ = nullptr;
	57	size_t inProgressIOs = 0;
	58	static AIOCB OpenDone;
	59	void openDone(int fd, const char *buf, int aio_return, int aio_errno);
	60	RefCount<IORequestor> ioRequestor;
	61	void doClose();
	62
	63	void readDone(int fd, const char *buf, int len, int errflag, RefCount<ReadRequest> request);
	64	void writeDone(int fd, int errflag, size_t len, RefCount<WriteRequest> request);
	65	};
	66
	67	#include "DiskIO/ReadRequest.h"
	68
	69	template <class RT>
	70	class IoResult
	71	{
	72	CBDATA_CLASS(IoResult);
	73
	74	public:
	75	IoResult(RefCount<DiskThreadsDiskFile> aFile, RefCount<RT> aRequest) : file(aFile), request(aRequest) {}
	76
	77	RefCount<DiskThreadsDiskFile> file;
	78	RefCount<RT> request;
	79	};
	80
	81	template <class RT>
	82	IoResult<RT>
	83	IOResult(RefCount<RT> aRequest, RefCount<DiskThreadsDiskFile> aFile) { return IoResult<RT>(aFile, aRequest);}
	84
	85	#endif /* SQUID_DISKTHREADSDISKFILE_H */
	86