public void DirectorySizeStressTest()
{
string tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
Directory.CreateDirectory(tempDir);
DirectoryAccounting da = new DirectoryAccounting();
da.SetDirectoryQuota(tempDir, 1024*1024*1);
DateTime start = DateTime.Now;
long lastDirSize = 0;
for (int i = 0; i < 100; i++)
{
string file0 = Path.Combine(tempDir, "file" + i.ToString());
File.WriteAllBytes(file0, new byte[1024 * 1]);
long dirSize = da.GetDirectorySize(tempDir);
Assert.IsTrue(dirSize > lastDirSize);
}
long elapedMs = (DateTime.Now - start).Milliseconds;
Assert.IsTrue(elapedMs < 2000);
Directory.Delete(tempDir, true);
}
public override void Start(ServiceElement options)
{
if (options == null)
{
throw new ArgumentNullException("options");
}
this.baseDir = options.BaseDir;
this.maxStorageSizeMB = options.Uhurufs.MaxStorageSize;
this.useVhd = options.Uhurufs.UseVHD;
this.vhdFixedSize = options.Uhurufs.VHDFixedSize;
this.useFsrm = options.Uhurufs.UseFsrm;
if (this.useFsrm)
{
this.dirAccounting = new DirectoryAccounting();
}
else
{
this.dirAccounting = null;
}
ProvisionedService.Initialize(options.LocalDB);
HashSet<string> sharesCache = new HashSet<string>(WindowsShare.GetShares());
foreach (ProvisionedService instance in ProvisionedService.GetInstances())
{
this.capacity -= this.CapacityUnit();
// This check will make initialization faster.
if (!sharesCache.Contains(instance.Name))
{
// This will setup the instance with new config changes or if the OS is fresh.
// Don't want to fail if an instance is inconsistent or has errors.
try
{
this.InstanceSystemSetup(instance);
foreach (ServiceBinding binding in instance.Bindings)
{
try
{
Bind(instance, binding);
}
catch (Exception ex)
{
Logger.Error("Error binding instance {0} with {1}. Exception: {2}", instance.Name, binding.User, ex.ToString());
}
}
}
catch (Exception ex)
{
Logger.Error("Error setting up instance {0}. Exception: {1}", instance.Name, ex.ToString());
}
}
}
TimerHelper.RecurringCall(
StorageQuotaInterval,
delegate
{
this.EnforceStorageQuota();
});
// initialize qps counter
this.provisionServed = 0;
this.bindingServed = 0;
base.Start(options);
}
public void DirectorySizeTest1()
{
string tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
// This is the structure we are creating
// tempdir
// dir1 (2)
// file1 (2KB)
// dir2 (28 KB)
// dir3 (24 KB)
// dir4 (16KB)
// file4 (16KB)
// file3 (8KB)
// file2 (4KB)
// file0 (1MB)
string dir1 = Path.Combine(tempDir, "dir1");
string dir2 = Path.Combine(tempDir, "dir2");
string dir3 = Path.Combine(dir2, "dir3");
string dir4 = Path.Combine(dir3, "dir4");
Directory.CreateDirectory(dir1);
Directory.CreateDirectory(dir4);
string file0 = Path.Combine(tempDir, "file0");
string file1 = Path.Combine(dir1, "file1");
string file2 = Path.Combine(dir2, "file2");
string file3 = Path.Combine(dir3, "file3");
string file4 = Path.Combine(dir4, "file4");
string file5 = Path.Combine(dir4, "file5");
File.WriteAllBytes(file0, new byte[1024 * 1024]);
File.WriteAllBytes(file1, new byte[2048]);
File.WriteAllBytes(file2, new byte[4096]);
File.WriteAllBytes(file3, new byte[8192]);
File.WriteAllBytes(file4, new byte[16384]);
DirectoryAccounting da = new DirectoryAccounting();
long dirSize = da.GetDirectorySize(tempDir);
long KB = 1024;
Assert.IsTrue(dirSize >= 2*KB + 28*KB + 1024*KB);
da.SetDirectoryQuota(tempDir, 2 * KB + 28 * KB + 1024 * KB + 10 * KB);
bool exceptionRaised = false;
try
{
File.WriteAllBytes(file5, new byte[16384]);
}
catch (IOException ex)
{
if (ex.Message.Contains("There is not enough space on the disk"))
{
exceptionRaised = true;
}
}
Assert.IsTrue(exceptionRaised);
da.RemoveDirectoryQuota(tempDir);
File.WriteAllBytes(file5, new byte[16384]);
Directory.Delete(tempDir, true);
}