mozilla
/
gecko-dev
зеркало из https://github.com/mozilla/gecko-dev.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность
gecko-dev/netwerk/test/TestFileInput2.cpp

483 строки
13 KiB
C++
Исходник Ответственный История

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*-
*
* The contents of this file are subject to the Netscape Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is mozilla.org code.
*
* The Initial Developer of the Original Code is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
*/
#include "nsIServiceManager.h"
#include "nsIInputStream.h"
#include "nsIOutputStream.h"
#include "nsIRunnable.h"
#include "nsIThread.h"
#include "nsISupportsArray.h"
#include "prinrval.h"
#include "nsIFileStream.h"
#include "nsFileSpec.h"
#include <stdio.h>
////////////////////////////////////////////////////////////////////////////////
#include <math.h>
#include "prprf.h"
#include "nsAutoLock.h"
class nsTimeSampler {
public:
nsTimeSampler();
void Reset();
void StartTime();
void EndTime();
void AddTime(PRIntervalTime time);
PRIntervalTime LastInterval() { return mLastInterval; }
char* PrintStats();
protected:
PRIntervalTime mStartTime;
double mSquares;
double mTotalTime;
PRUint32 mCount;
PRIntervalTime mLastInterval;
};
nsTimeSampler::nsTimeSampler()
{
Reset();
}
void
nsTimeSampler::Reset()
{
mStartTime = 0;
mSquares = 0;
mTotalTime = 0;
mCount = 0;
mLastInterval = 0;
}
void
nsTimeSampler::StartTime()
{
mStartTime = PR_IntervalNow();
}
void
nsTimeSampler::EndTime()
{
NS_ASSERTION(mStartTime != 0, "Forgot to call StartTime");
PRIntervalTime endTime = PR_IntervalNow();
mLastInterval = endTime - mStartTime;
AddTime(mLastInterval);
mStartTime = 0;
}
void
nsTimeSampler::AddTime(PRIntervalTime time)
{
nsAutoCMonitor mon(this);
mTotalTime += time;
mSquares += (double)time * (double)time;
mCount++;
}
char*
nsTimeSampler::PrintStats()
{
double mean = mTotalTime / mCount;
double variance = fabs(mSquares / mCount - mean * mean);
double stddev = sqrt(variance);
PRUint32 imean = (PRUint32)mean;
PRUint32 istddev = (PRUint32)stddev;
return PR_smprintf("%d +/- %d ms",
PR_IntervalToMilliseconds(imean),
PR_IntervalToMilliseconds(istddev));
}
////////////////////////////////////////////////////////////////////////////////
nsTimeSampler gTimeSampler;
typedef nsresult (*CreateFun)(nsIRunnable* *result,
const char* inPath,
const char* outPath,
PRUint32 bufferSize);
////////////////////////////////////////////////////////////////////////////////
nsresult
Copy(nsIInputStream* inStr, nsIOutputStream* outStr,
char* buf, PRUint32 bufSize, PRUint32 *copyCount)
{
nsresult rv;
while (PR_TRUE) {
PRUint32 count;
rv = inStr->Read(buf, bufSize, &count);
if (NS_FAILED(rv)) return rv;
if (count == 0) break;
PRUint32 writeCount;
rv = outStr->Write(buf, count, &writeCount);
if (NS_FAILED(rv)) return rv;
NS_ASSERTION(writeCount == count, "didn't write all the data");
*copyCount += writeCount;
}
rv = outStr->Flush();
return rv;
}
////////////////////////////////////////////////////////////////////////////////
class FileSpecWorker : public nsIRunnable {
public:
NS_IMETHOD Run() {
nsresult rv;
PRIntervalTime startTime = PR_IntervalNow();
PRIntervalTime endTime;
nsIInputStream* inStr = nsnull;
nsIOutputStream* outStr = nsnull;
nsISupports* str = nsnull;
PRUint32 copyCount = 0;
nsFileSpec inSpec(mInPath);
nsFileSpec outSpec(mOutPath);
// Open the input stream:
rv = NS_NewTypicalInputFileStream(&str, inSpec);
if (NS_FAILED(rv)) goto done;
rv = str->QueryInterface(NS_GET_IID(nsIInputStream), (void**)&inStr);
NS_RELEASE(str);
if (NS_FAILED(rv)) goto done;
// Open the output stream:
rv = NS_NewTypicalOutputFileStream(&str, outSpec);
if (NS_FAILED(rv)) goto done;
rv = str->QueryInterface(NS_GET_IID(nsIOutputStream), (void**)&outStr);
NS_RELEASE(str);
if (NS_FAILED(rv)) goto done;
// Copy from one to the other
rv = Copy(inStr, outStr, mBuffer, mBufferSize, &copyCount);
if (NS_FAILED(rv)) goto done;
endTime = PR_IntervalNow();
gTimeSampler.AddTime(endTime - startTime);
done:
NS_IF_RELEASE(outStr);
NS_IF_RELEASE(inStr);
NS_IF_RELEASE(str);
return rv;
}
NS_DECL_ISUPPORTS
FileSpecWorker()
: mInPath(nsnull), mOutPath(nsnull), mBuffer(nsnull),
mBufferSize(0)
{
NS_INIT_REFCNT();
}
nsresult Init(const char* inPath, const char* outPath,
PRUint32 bufferSize)
{
mInPath = nsCRT::strdup(inPath);
mOutPath = nsCRT::strdup(outPath);
mBuffer = new char[bufferSize];
mBufferSize = bufferSize;
return (mInPath && mOutPath && mBuffer)
? NS_OK : NS_ERROR_OUT_OF_MEMORY;
}
static nsresult Create(nsIRunnable* *result,
const char* inPath,
const char* outPath,
PRUint32 bufferSize)
{
FileSpecWorker* worker = new FileSpecWorker();
if (worker == nsnull)
return NS_ERROR_OUT_OF_MEMORY;
NS_ADDREF(worker);
nsresult rv = worker->Init(inPath, outPath, bufferSize);
if (NS_FAILED(rv)) {
NS_RELEASE(worker);
return rv;
}
*result = worker;
return NS_OK;
}
virtual ~FileSpecWorker() {
nsCRT::free(mInPath);
nsCRT::free(mOutPath);
delete[] mBuffer;
}
protected:
char* mInPath;
char* mOutPath;
char* mBuffer;
PRUint32 mBufferSize;
};
NS_IMPL_ISUPPORTS(FileSpecWorker, NS_GET_IID(nsIRunnable));
////////////////////////////////////////////////////////////////////////////////
#include "nsIIOService.h"
#include "nsIFileChannel.h"
#include "nsIFileProtocolHandler.h"
static NS_DEFINE_CID(kIOServiceCID, NS_IOSERVICE_CID);
class FileChannelWorker : public nsIRunnable {
public:
NS_IMETHOD Run() {
nsresult rv;
NS_WITH_SERVICE(nsIIOService, ioserv, kIOServiceCID, &rv);
if (NS_FAILED(rv)) return rv;
PRIntervalTime startTime = PR_IntervalNow();
PRIntervalTime endTime;
PRUint32 copyCount = 0;
nsIInputStream* inStr = nsnull;
nsIOutputStream* outStr = nsnull;
nsIFileChannel* inCh = nsnull;
nsIFileChannel* outCh = nsnull;
rv = ioserv->NewChannelFromNativePath(mInPath, &inCh);
if (NS_FAILED(rv)) goto done;
rv = inCh->OpenInputStream(0, -1, &inStr);
if (NS_FAILED(rv)) goto done;
rv = ioserv->NewChannelFromNativePath(mOutPath, &outCh);
if (NS_FAILED(rv)) goto done;
rv = outCh->OpenOutputStream(0, &outStr);
if (NS_FAILED(rv)) goto done;
// Copy from one to the other
rv = Copy(inStr, outStr, mBuffer, mBufferSize, &copyCount);
if (NS_FAILED(rv)) goto done;
endTime = PR_IntervalNow();
gTimeSampler.AddTime(endTime - startTime);
done:
NS_RELEASE(inStr);
NS_RELEASE(inCh);
NS_RELEASE(outStr);
NS_RELEASE(outCh);
return rv;
}
NS_DECL_ISUPPORTS
FileChannelWorker()
: mInPath(nsnull), mOutPath(nsnull), mBuffer(nsnull),
mBufferSize(0)
{
NS_INIT_REFCNT();
}
nsresult Init(const char* inPath, const char* outPath,
PRUint32 bufferSize)
{
mInPath = nsCRT::strdup(inPath);
mOutPath = nsCRT::strdup(outPath);
mBuffer = new char[bufferSize];
mBufferSize = bufferSize;
return (mInPath && mOutPath && mBuffer)
? NS_OK : NS_ERROR_OUT_OF_MEMORY;
}
static nsresult Create(nsIRunnable* *result,
const char* inPath,
const char* outPath,
PRUint32 bufferSize)
{
FileChannelWorker* worker = new FileChannelWorker();
if (worker == nsnull)
return NS_ERROR_OUT_OF_MEMORY;
NS_ADDREF(worker);
nsresult rv = worker->Init(inPath, outPath, bufferSize);
if (NS_FAILED(rv)) {
NS_RELEASE(worker);
return rv;
}
*result = worker;
return NS_OK;
}
virtual ~FileChannelWorker() {
nsCRT::free(mInPath);
nsCRT::free(mOutPath);
delete[] mBuffer;
}
protected:
char* mInPath;
char* mOutPath;
char* mBuffer;
PRUint32 mBufferSize;
};
NS_IMPL_ISUPPORTS(FileChannelWorker, NS_GET_IID(nsIRunnable));
////////////////////////////////////////////////////////////////////////////////
void
Test(CreateFun create, PRUint32 count,
const char* inDir, const char* outDir, PRUint32 bufSize)
{
nsresult rv;
PRUint32 i;
printf("###########\nTest: from %s to %s, bufSize = %d\n",
inDir, outDir, bufSize);
gTimeSampler.Reset();
nsTimeSampler testTime;
testTime.StartTime();
nsISupportsArray* threads;
rv = NS_NewISupportsArray(&threads);
NS_ASSERTION(NS_SUCCEEDED(rv), "NS_NewISupportsArray failed");
nsFileSpec inDirSpec(inDir);
nsDirectoryIterator iter(inDirSpec, PR_TRUE);
for (i = 0; i < count && iter.Exists(); i++, iter++) {
nsIThread* thread;
nsIRunnable* worker;
nsFileSpec& inSpec = iter.Spec();
nsFileSpec outSpec(outDir);
outSpec += inSpec.GetLeafName();
if (outSpec.Exists()) {
outSpec.Delete(PR_FALSE);
}
rv = create(&worker,
inSpec.GetNativePathCString(),
outSpec.GetNativePathCString(),
bufSize);
if (NS_FAILED(rv)) goto done;
rv = NS_NewThread(&thread, worker, 0, PR_JOINABLE_THREAD);
NS_RELEASE(worker);
if (NS_FAILED(rv)) goto done;
PRBool inserted = threads->InsertElementAt(thread, i);
NS_ASSERTION(inserted, "not inserted");
}
PRUint32 j;
for (j = 0; j < i; j++) {
nsIThread* thread = (nsIThread*)threads->ElementAt(j);
if (NS_FAILED(rv)) goto done;
thread->Join();
}
done:
NS_RELEASE(threads);
NS_ASSERTION(rv == NS_OK, "failed");
testTime.EndTime();
char* testStats = testTime.PrintStats();
char* workerStats = gTimeSampler.PrintStats();
printf(" threads = %d\n work time = %s,\n test time = %s\n",
i, workerStats, testStats);
PR_smprintf_free(workerStats);
PR_smprintf_free(testStats);
}
////////////////////////////////////////////////////////////////////////////////
nsresult NS_AutoregisterComponents()
{
nsresult rv = nsComponentManager::AutoRegister(nsIComponentManager::NS_Startup, NULL /* default */);
return rv;
}
int
main(int argc, char* argv[])
{
nsresult rv;
if (argc < 2) {
printf("usage: %s <in-dir> <out-dir>\n", argv[0]);
return -1;
}
char* inDir = argv[1];
char* outDir = argv[2];
rv = NS_AutoregisterComponents();
if (NS_FAILED(rv)) return rv;
CreateFun create = FileChannelWorker::Create;
Test(create, 1, inDir, outDir, 16 * 1024);
#if 1
printf("FileChannelWorker *****************************\n");
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
#endif
create = FileSpecWorker::Create;
printf("FileSpecWorker ********************************\n");
#if 1
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
Test(create, 20, inDir, outDir, 16 * 1024);
#endif
#if 1
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
Test(create, 20, inDir, outDir, 4 * 1024);
#endif
return 0;
}
////////////////////////////////////////////////////////////////////////////////
Ссылка в новой задаче Показать git blame Копировать постоянную ссылку