knowledge-governance/cmd/worker/library_capacity.go

package main

import (
	"context"
	"errors"
	"fmt"
	"math"

	"knowledge-governance/internal/appconfig"
	"knowledge-governance/internal/libraryretention"
	"knowledge-governance/internal/librarysnapshot"
)

// libraryWriteFootprint 是本次写入的容量需求下界：内容字节 + 等量恢复余量。
//
// 为什么要算两份：写入前必须留出"目标被写坏后按预备份恢复"的空间。只按本次内容算，
// 预算刚好够写、不够恢复，故障时就地无法回滚——那比不发布更糟。
func libraryWriteFootprint(contentBytes int64) (int64, error) {
	if contentBytes < 0 {
		return 0, errors.New("内容字节为负：容量口径异常")
	}
	if contentBytes > math.MaxInt64/2 {
		return 0, errors.New("内容字节过大：容量需求溢出，拒绝放行")
	}
	return contentBytes * 2, nil
}

// checkLibraryCapacity 是**执行前**的容量门禁（F11）。
//
// 旧实现只在存储页 GET 里调用 CheckCapacity：那个数字是给人看的，worker 从来不读它，
// 所以"预算为正但明显不够"照样会进入维护、拍备份、写目标。这里把它接到真正要写之前。
//
// 读不到容量/预算时不放行：无法证明装得下就不该动生产。
func checkLibraryCapacity(ctx context.Context, config appconfig.Config, snapshotID string) error {
	if config.LibraryCapacityBudgetBytes <= 0 {
		return errors.New("容量预算未设置（LIBRARY_CAPACITY_BUDGET_BYTES）：无法证明装得下，拒绝进入维护")
	}
	archive, err := librarysnapshot.Open(config.LibraryArchiveRoot, 256*1024*1024)
	if err != nil {
		return fmt.Errorf("归档不可用，无法核算容量：%w", err)
	}
	defer archive.Close()
	usage, usageErr := archive.ContentUsage()
	if usageErr != nil {
		return fmt.Errorf("读不到当前归档内容占用，拒绝放行：%w", usageErr)
	}
	manifest, loadErr := archive.Load(snapshotID)
	if loadErr != nil {
		return fmt.Errorf("读不到待写入快照的清单，无法核算容量：%w", loadErr)
	}
	var incoming int64
	for _, part := range manifest.Parts {
		if part.Bytes < 0 || incoming > math.MaxInt64-part.Bytes {
			return errors.New("快照分片字节异常：拒绝放行")
		}
		incoming += part.Bytes
	}
	required, footprintErr := libraryWriteFootprint(incoming)
	if footprintErr != nil {
		return footprintErr
	}
	capacity := libraryretention.CheckCapacity(usage.ContentBytes, required, config.LibraryCapacityBudgetBytes)
	if !capacity.OK {
		return fmt.Errorf("容量门禁拒绝，未打开维护窗口：%s", capacity.Reason)
	}
	return nil
}