Method Set

util/cache/disk_cache.go:114–170

Set 设置缓存

(key string, data []byte, ttl time.Duration)

// Set 设置缓存
func (c *DiskCache) Set(key string, data []byte, ttl time.Duration) error {
	c.mutex.Lock()
	defer c.mutex.Unlock()

	// 如果已存在，先减去旧项的大小
	if meta, exists := c.metadata[key]; exists {
		c.currSize -= int64(meta.Size)
		// 删除旧文件
		filename := c.getFilename(key)
		os.Remove(filepath.Join(c.path, filename))
		os.Remove(filepath.Join(c.path, filename+".meta"))
	}

	// 检查空间
	maxSize := int64(c.maxSizeMB) * 1024 * 1024
	if c.currSize+int64(len(data)) > maxSize {
		// 清理空间
		c.evictLRU(int64(len(data)))
	}

	// 获取文件名
	filename := c.getFilename(key)
	filePath := filepath.Join(c.path, filename)

	// 确保目录存在（防止外部删除缓存目录）
	if err := os.MkdirAll(c.path, 0755); err != nil {
		return fmt.Errorf("创建缓存目录失败: %v", err)
	}

	// 写入文件
	if err := ioutil.WriteFile(filePath, data, 0644); err != nil {
		return err
	}

	// 创建元数据
	now := time.Now()
	meta := &diskCacheMetadata{
		Key:         key,
		Expiry:      now.Add(ttl),
		LastUsed:    now,
		LastModified: now, // 设置最后修改时间
		Size:        len(data),
	}

	// 保存元数据
	if err := c.saveMetadata(key, meta); err != nil {
		// 如果元数据保存失败，删除数据文件
		os.Remove(filePath)
		return err
	}

	// 更新内存中的元数据
	c.metadata[key] = meta
	c.currSize += int64(len(data))

	return nil
}