src/DiskIO/DiskThreads/DiskThreadsDiskFile.h

   1 /*
   2  * Copyright (C) 1996-2020 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