Zig自定义Set类型时结构体字段被视为常量的编译错误

我正在学习Zig，尝试基于标准库的HashMap实现一个Set类型，代码如下：

fn Set(comptime T: type) type {
        return struct {
                hashMap : hashMapType,
                const hashMapType: type = std.hash_map.AutoHashMap(T, T);
                const Self = @This();
                fn init(allocator: std.mem.Allocator) Self {
                        return Self{
                                .hashMap = hashMapType.init(allocator),
                        };
                }
                fn put(self: Self, element: T) std.mem.Allocator.Error!void {
                        return try self.hashMap.put(element, element);
                }
                fn count(self: Self) hashMapType.Size {
                        return self.hashMap.count();
                }
        };
}

编译调用put函数的代码时，出现了以下错误：

error: expected type '*hash_map.HashMap(set_experiment.rootPair,set_experiment.rootPair,hash_map.AutoContext(set_experiment.rootPair),80)', found '*const hash_map.HashMap(set_experiment.rootPair,set_experiment.rootPair,hash_map.AutoContext(set_experiment.rootPair),80)' 
   return try self.hashMap.put(element, element);
              ~~~~~~~~~~~~^~~~
set_experiment.zig:26:27: note: cast discards const qualifier
/usr/lib/zig/std/hash_map.zig:551:26: note: parameter type declared here
        pub fn put(self: *Self, key: K, value: V) Allocator.Error!void {
                         ^~~~~

我能看出错误是因为内部HashMap被视为常量，但不清楚语法层面符合Zig惯用方式的正确修复方法是什么。

问题原因与修复方案 #

核心问题 #

Zig中，当你把self作为值传递（self: Self）时，函数拿到的是结构体的副本，且默认是不可变的（const）。调用hashMap.put需要一个可变的指针（*HashMap），但此时self.hashMap是const的，无法获取可变指针，因此触发报错。

正确修复方式 #

根据Zig的惯用写法，需要根据函数是否修改结构体内部状态，选择对应的参数类型：

• 修改内部状态的函数（如put）：接收可变指针*Self作为参数，允许修改结构体成员
• 仅读取内部状态的函数（如count）：接收const指针*const Self，明确不会修改结构体，同时避免不必要的拷贝

修改后的代码如下：

fn Set(comptime T: type) type {
        return struct {
                hashMap : hashMapType,
                const hashMapType: type = std.hash_map.AutoHashMap(T, T);
                const Self = @This();
                fn init(allocator: std.mem.Allocator) Self {
                        return Self{
                                .hashMap = hashMapType.init(allocator),
                        };
                }
                // 改为*Self，允许修改内部hashMap
                fn put(self: *Self, element: T) std.mem.Allocator.Error!void {
                        return try self.hashMap.put(element, element);
                }
                // 改为*const Self，明确仅读取操作
                fn count(self: *const Self) hashMapType.Size {
                        return self.hashMap.count();
                }
        };
}

额外优化提示 #

• 记得给Set添加deinit函数，调用内部hashMap.deinit()释放内存，避免泄漏：

  fn deinit(self: *Self) void {
      self.hashMap.deinit();
  }
  

• 后续如果添加remove这类修改操作，同样要使用*Self作为参数；contains这类查询函数，使用*const Self即可。

内容的提问来源于stack exchange，提问作者Philip Adler