Method Prune

pkg/memory/logic/store_postgres.go:426–478

Prune 清理低价值 Memory

(ctx context.Context, criteria PruneCriteria)

// Prune 清理低价值 Memory
func (s *PostgreSQLStore) Prune(ctx context.Context, criteria PruneCriteria) (int, error) {
	if s.closed {
		return 0, ErrStoreClosed
	}

	args := []any{}
	argIndex := 1

	// 构建 OR 条件
	conditions := []string{}

	if criteria.MinConfidence > 0 {
		conditions = append(conditions, fmt.Sprintf("confidence < $%d", argIndex))
		args = append(args, criteria.MinConfidence)
		argIndex++
	}

	if criteria.SinceLastAccess > 0 {
		conditions = append(conditions, fmt.Sprintf("last_accessed < $%d", argIndex))
		args = append(args, time.Now().Add(-criteria.SinceLastAccess))
		argIndex++
	}

	if criteria.MinAccessCount > 0 && criteria.MaxAge > 0 {
		conditions = append(conditions, fmt.Sprintf("(access_count < $%d AND created_at < $%d)", argIndex, argIndex+1))
		args = append(args, criteria.MinAccessCount, time.Now().Add(-criteria.MaxAge))
		// argIndex += 2 不需要，后续没有使用
	}

	if len(conditions) == 0 {
		return 0, nil
	}

	// 构建查询
	query := fmt.Sprintf(`
		DELETE FROM %s
		WHERE %s
	`, s.tableName, conditions[0])

	var querySb463 strings.Builder
	for i := 1; i < len(conditions); i++ {
		querySb463.WriteString(" OR " + conditions[i])
	}
	query += querySb463.String()

	result, err := s.db.ExecContext(ctx, query, args...)
	if err != nil {
		return 0, NewStoreError("DELETE_ERROR", "failed to prune memories", err)
	}

	rowsAffected, _ := result.RowsAffected()
	return int(rowsAffected), nil
}

// Close 关闭存储
func (s *PostgreSQLStore) Close() error {