LordMZTE/dotfiles
1
0
Fork 0
mirror of https://mzte.de/git/LordMZTE/dotfiles.git synced 2024-06-09 06:50:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

randomwallpaper: use arena allocator for filenames

Browse source
This commit is contained in:
LordMZTE 2022-05-24 17:44:16 +02:00
parent cf25334721
commit 314c75262f
Signed by: LordMZTE
GPG key ID: B64802DC33A64FF6
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
scripts/randomwallpaper/src/walker.zig
View file

@ -2,18 +2,20 @@ const std = @import("std");
pub const Walker = struct {
files: std.ArrayList([]u8),
filename_arena: std.heap.ArenaAllocator,
buf: [64]u8 = undefined,
pub const open_opts = std.fs.Dir.OpenDirOptions{ .iterate = true };
pub fn init(alloc: std.mem.Allocator) Walker {
return Walker{ .files = std.ArrayList([]u8).init(alloc) };
return Walker{
.files = std.ArrayList([]u8).init(alloc),
.filename_arena = std.heap.ArenaAllocator.init(alloc),
};
}
pub fn deinit(self: *Walker) void {
for (self.files.items) |file| {
self.files.allocator.free(file);
}
self.filename_arena.deinit();
self.files.deinit();
}
@ -22,7 +24,7 @@ pub const Walker = struct {
while (try iter.next()) |e| {
switch (e.kind) {
.File => {
const path = try dir.realpathAlloc(self.files.allocator, e.name);
const path = try dir.realpathAlloc(self.filename_arena.allocator(), e.name);
try self.files.append(path);
},
.Directory => {
@ -35,7 +37,8 @@ pub const Walker = struct {
var subdir = dir.openDir(p, open_opts) catch |err| {
switch (err) {
std.fs.Dir.OpenError.NotDir => {
try self.files.append(p);
const fpath = try self.filename_arena.allocator().dupe(u8, p);
try self.files.append(fpath);
continue;
},
else => {