gecko-dev/netwerk/cache2/CacheFileOutputStream.h

74 lines
2.1 KiB
C++

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef CacheFileOutputStream__h__
#define CacheFileOutputStream__h__
#include "nsIAsyncOutputStream.h"
#include "nsISeekableStream.h"
#include "nsCOMPtr.h"
#include "nsAutoPtr.h"
#include "CacheFileChunk.h"
namespace mozilla {
namespace net {
class CacheFile;
class CacheOutputCloseListener;
class CacheFileOutputStream : public nsIAsyncOutputStream
, public nsISeekableStream
, public CacheFileChunkListener
{
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIOUTPUTSTREAM
NS_DECL_NSIASYNCOUTPUTSTREAM
NS_DECL_NSISEEKABLESTREAM
public:
CacheFileOutputStream(CacheFile *aFile,
CacheOutputCloseListener *aCloseListener,
bool aAlternativeData);
NS_IMETHOD OnChunkRead(nsresult aResult, CacheFileChunk *aChunk) override;
NS_IMETHOD OnChunkWritten(nsresult aResult, CacheFileChunk *aChunk) override;
NS_IMETHOD OnChunkAvailable(nsresult aResult, uint32_t aChunkIdx,
CacheFileChunk *aChunk) override;
NS_IMETHOD OnChunkUpdated(CacheFileChunk *aChunk) override;
void NotifyCloseListener();
bool IsAlternativeData() const { return mAlternativeData; };
// Memory reporting
size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
private:
virtual ~CacheFileOutputStream();
nsresult CloseWithStatusLocked(nsresult aStatus);
void ReleaseChunk();
void EnsureCorrectChunk(bool aReleaseOnly);
void FillHole();
void NotifyListener();
RefPtr<CacheFile> mFile;
RefPtr<CacheFileChunk> mChunk;
RefPtr<CacheOutputCloseListener> mCloseListener;
int64_t mPos;
bool mClosed : 1;
bool const mAlternativeData : 1;
nsresult mStatus;
nsCOMPtr<nsIOutputStreamCallback> mCallback;
uint32_t mCallbackFlags;
nsCOMPtr<nsIEventTarget> mCallbackTarget;
};
} // namespace net
} // namespace mozilla
#endif