Tazhiyou

template <typename Interface>
void registerSingleton(std::function<std::shared_ptr<Interface>()> factory) {
    ServiceExtractor extractor = makeExtractor<Interface>();
    registerEntry(typeid(Interface), Lifetime::Singleton,
        [factory]() -> std::shared_ptr<void> {
            return std::static_pointer_cast<void>(factory());
        }, extractor);
}

// 实际注册示例（AppController.cpp:436）
ServiceLocator::instance().registerSingleton<UsageStatsService>(
    []() { return std::make_shared<UsageStatsService>(); });

template <typename T>
Result<T> guard(const std::function<T()>& fn,
               ErrorCategory category,
               const QString& context = {}) {
    try {
        return { ResultCode::Ok, {}, fn() };
    } catch (const std::exception& e) {
        report(category, ResultCode::Failed,
                  QString::fromLocal8Bit(e.what()), context);
        return { ResultCode::Failed, QString::fromLocal8Bit(e.what()) };
    } catch (...) {
        report(category, ResultCode::Failed,
                  QStringLiteral("未知异常"), context);
        return { ResultCode::Failed, QStringLiteral("未知异常") };
    }
}

const KnowledgeEntry* loadEntry(qint64 id) {
    ++m_totalAccess;
    auto it = m_index.find(id);
    if (it != m_index.end()) {                  // 命中缓存: 移到头部
        const int idx = it.value();
        if (idx > 0) {
            CacheNode node = m_lruList.takeAt(idx);
            m_lruList.prepend(node);
            m_index.clear();
            for (int i = 0; i < m_lruList.size(); ++i)
                m_index.insert(m_lruList[i].id, i);
        }
        ++m_cacheHits;
        return &m_lruList.first().entry;
    }
    // 未命中: 从磁盘加载, 插入头部, 超预算淘汰尾部
    auto r = m_store->get(id);
    if (!r.ok()) return nullptr;
    m_lruList.prepend({ id, r.data });
    if (m_usedBytes > m_budgetBytes) unloadCold();
    return &m_lruList.first().entry;
}

QStringList tokenize(const QString& text) const {
    QStringList tokens;
    for (const QString& seg : segments) {
        if (seg.contains(/* 中文字符 */)) {
            for (int i = 0; i + 1 < seg.length(); ++i) {   // 中文: 2 字滑窗
                const QChar c1 = seg.at(i), c2 = seg.at(i + 1);
                if (c1.unicode() >= 0x4e00 && c1.unicode() <= 0x9fff)
                    tokens.append(QString(c1) + c2);
            }
        }
    }
    return tokens;
}

void rebuildIdf() {
    const double N = static_cast<double>(qMax(1, m_docCount));
    for (auto it = m_docFreq.begin(); it != m_docFreq.end(); ++it) {
        const double df = static_cast<double>(it.value());
        const double idf = std::log((N + 1.0) / (df + 1.0)) + 1.0;  // 平滑 IDF
        m_idf.insert(it.key(), idf);
    }
}

static constexpr int kLlmTimeoutMs = 60000;   // LLM 调用超时 60s

QString callLlm(const QString& systemPrompt, const QString& userPrompt) {
    QList<ChatMessage> messages; /* System + User 消息构建 */
    QEventLoop loop;
    Result<QString> result;

    ModelManager::instance().chat(messages,
        [&result, &loop](const Result<QString>& r) {
            result = r;
            loop.quit();
        });

    // 超时保护: 60s 后强制退出
    QTimer::singleShot(kLlmTimeoutMs, &loop, &QEventLoop::quit);
    loop.exec();

    if (!result.ok()) {
        TAZ_LOG_WARNING("DocumentAgent: LLM 调用失败: %1", result.message);
        return {};
    }
    return result.data;
}

auto* watcher = new QFutureWatcher<ClassifyResult>(this);
connect(watcher, &QFutureWatcher<ClassifyResult>::finished, this,
    [this, watcher]() {
        auto r = watcher->result();
        if (r.ok()) {
            m_statusLabel->setText(QStringLiteral("已按维度分类: %1 组")
                .arg(r.data.size()));
        }
        watcher->deleteLater();
    });

watcher->setFuture(QtConcurrent::run([root, classifyBy]() {
    FileClassifier fc;
    return fc.classify(root, classifyBy, false);
}));