Method redactWithTracking

pkg/security/pii_middleware.go:151–180

redactWithTracking 脱敏文本并追踪 PII 匹配。

(ctx context.Context, text string, agentID string)

// redactWithTracking 脱敏文本并追踪 PII 匹配。
func (m *PIIRedactionMiddleware) redactWithTracking(ctx context.Context, text string, agentID string) (string, []PIIMatch, error) {
	if text == "" {
		return text, nil, nil
	}

	// 检测和脱敏
	matches, err := m.redactor.detector.Detect(ctx, text)
	if err != nil {
		return text, nil, err
	}

	if len(matches) == 0 {
		return text, nil, nil
	}

	// 如果启用追踪，保存匹配信息
	if m.enableTracking {
		m.mu.Lock()
		m.tracking[agentID] = append(m.tracking[agentID], matches...)
		m.mu.Unlock()
	}

	// 脱敏
	redacted, err := m.redactor.Redact(ctx, text)
	if err != nil {
		return text, nil, err
	}

	return redacted, matches, nil
}

// OnAgentStop 在 Agent 停止时清除追踪信息。
func (m *PIIRedactionMiddleware) OnAgentStop(ctx context.Context, agentID string) error {