[code snippet] 生成随机大文件

[code snippet] 生成随机大文件

一个无聊的测试代码,因为要测试大文件的网络传输,就写了一个随机大文件生成脚本,做个备份。

基本上都是 GPT 生成的,哈哈。

C# 代码

namespace ConsolePlayground;

internal class BigFileGenerater
{

    public async Task Run()
    {
        Console.WriteLine("大文件生成工具,输入要生成的文件大小,单位 GB");
        Console.WriteLine();
        var input = Console.ReadLine();
        if (string.IsNullOrWhiteSpace(input))
        {
            return;
        }
        if (!double.TryParse(input.Trim(), out var size))
        {
            return;
        }

        if (size <= 0 || size > 1000)
        {
            throw new ArgumentOutOfRangeException("size");
        }
        var byteSize = size * 1024 * 1024 * 1024;

        var folder = Environment.CurrentDirectory;
        Console.WriteLine($"将在当前目录 {folder} 生成 {size} GB 的文件");
        Console.WriteLine("是否继续? Y/N");

        input = Console.ReadLine();
        if (string.IsNullOrWhiteSpace(input))
        {
            return;
        }
        if (input.Trim().ToUpper() != "Y")
        {
            return;
        }

        Console.WriteLine("开始生成...");
        await Genetate(folder, byteSize);
        Console.WriteLine("结束生成");
    }

    private async Task Genetate(string folder, double size)
    {
        var file = Path.Combine(folder, $"TEMP_{DateTime.Now:yyyyMMddHHmmss.fff}.tmp");

        using var fileStream = new FileStream(file, FileMode.OpenOrCreate);
        var rng = new Random();
        int step = 1024 * 1024;
        int count = (int)Math.Ceiling(size / step);

        for (int i = 0; i < count; i++)
        {
            byte[] bytesToWrite = GenerateRandomBytes(rng, step);
            await fileStream.WriteAsync(bytesToWrite, 0, step);
            await fileStream.FlushAsync();
        }
    }

    public static byte[] GenerateRandomBytes(Random rng, int length)
    {
        var randomBytes = new byte[length];
        rng.NextBytes(randomBytes);
        return randomBytes;
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值