Compare commits

...

3 Commits

Author SHA1 Message Date
root 723c3ffb86 feat(media-supply): resource cache sync and order flow refinements
Frontend CI / Frontend (push) Successful in 3m38s
Backend CI / Backend (push) Failing after 26m59s
Add a resource cache sync entrypoint (RunMediaResourceSyncOnce) driven
by the new scheduler job, and rework the backlink/sync worker around it.
Tune the meijiequan client and slow the default backlink sync interval
from 10m to 30m to reduce upstream pressure.

Trim unused public/transport surface (handler + router + swagger) and
simplify admin-web order management and ops-web media-supply views,
dropping stale shared-types fields.
2026-06-02 14:50:36 +08:00
root 842782b3dd feat(scheduler): random run-window scheduling and deduped manual triggers
Support a random daily run window (random_window_start/end) where the
next auto-run time is derived from a stable per-job hash, avoiding
thundering-herd starts while keeping one run per day. Honors the last
auto-run on initial scheduling so restarts don't double-fire.

Manual triggers can opt into dedupe (dedupe_manual_triggers): an
advisory-locked path collapses concurrent requests onto an existing
pending/running job instead of queueing duplicates. Scheduler error
messages are localized to Chinese.

Register the media_supply_cache_sync job (dry-run aware) and add the
ops migration that seeds it.
2026-06-02 14:50:25 +08:00
root ba2f117265 feat(knowledge): multi-query retrieval with LLM query rewrite
Add a shared structured query builder (knowledge_query.go) and switch
all generation paths to it, so retrieval queries carry task intent
(type, brand, region, keywords, key points) instead of a raw prompt.

ResolveContext now rewrites the query into multiple sub-questions via
the URL-markdown model, embeds and searches each, and merges/dedupes
candidates before rerank; falls back to parsed fallback questions when
rewrite is unavailable or returns too few. Resolve logs gain query
list, count, and rewrite stage/model for observability.

- knowledge_query.go/_test.go: BuildKnowledgeQuery + intent normalization
- knowledge_service.go: per-query embed/search loop, query rewrite, logs
- template_prompt/template_service/prompt_generate/article_imitation/
  kol_generation_worker: adopt the shared query builder
- generation_observability: richer task error logging
2026-06-02 14:50:12 +08:00
32 changed files with 1606 additions and 318 deletions
-7
View File
@@ -128,7 +128,6 @@ import type {
ScheduleTaskListResponse,
ScheduleTaskRequest,
ScheduleTaskStatusRequest,
SyncMediaSupplyBacklinksResponse,
TemplateAnalyzeTaskRequest,
TemplateAnalyzeTaskResultResponse,
TemplateAssistTaskCreateResponse,
@@ -1257,12 +1256,6 @@ export const mediaSupplyApi = {
getOrder(id: number) {
return apiClient.get<MediaSupplyOrderDetail>(`/api/tenant/media-supply/orders/${id}`)
},
syncBacklinks() {
return apiClient.post<SyncMediaSupplyBacklinksResponse, Record<string, never>>(
'/api/tenant/media-supply/orders/sync-backlinks',
{},
)
},
wallet() {
return apiClient.get<MediaSupplyWalletStatus>('/api/tenant/media-supply/wallet')
},
@@ -1,9 +1,9 @@
<script setup lang="ts">
import { CopyOutlined, ReloadOutlined } from '@ant-design/icons-vue'
import type { MediaSupplyOrderDetail } from '@geo/shared-types'
import { useMutation, useQuery } from '@tanstack/vue-query'
import { useQuery } from '@tanstack/vue-query'
import { message, type TableColumnsType } from 'ant-design-vue'
import { computed, ref } from 'vue'
import { computed, onBeforeUnmount, ref, watch } from 'vue'
import { mediaSupplyApi } from '@/lib/api'
import { formatDateTime } from '@/lib/display'
@@ -19,11 +19,14 @@ function copyToClipboard(text?: string | null): void {
message.warning('链接为空,无法复制')
return
}
void navigator.clipboard.writeText(url).then(() => {
message.success('已复制到剪贴板')
}).catch(() => {
message.error('复制失败,请手动复制')
})
void navigator.clipboard
.writeText(url)
.then(() => {
message.success('复制到剪贴板')
})
.catch(() => {
message.error('复制失败,请手动复制')
})
}
const ordersQuery = useQuery({
@@ -36,35 +39,6 @@ const ordersQuery = useQuery({
}),
})
const syncBacklinksMutation = useMutation({
mutationFn: () => mediaSupplyApi.syncBacklinks(),
onSuccess: async (result) => {
await ordersQuery.refetch()
if (result.refunded_orders > 0) {
message.warning(`已同步 ${result.refunded_orders} 个退稿并退款`)
return
}
if (result.links_updated > 0) {
message.success(`已同步 ${result.links_updated} 条回链`)
return
}
if (result.order_codes_updated > 0) {
message.success(`已同步 ${result.order_codes_updated} 个外部单号`)
return
}
message.info(
result.published_fetched > 0
? '已检查已发表记录,暂未发现新的回链'
: result.problem_fetched > 0
? '已检查退稿记录,暂未发现需要退款的订单'
: result.pending_fetched > 0
? '已检查发表中记录,暂未发现新的外部单号'
: '暂未读取到投稿记录',
)
},
onError: () => message.error('同步回链失败,请稍后重试'),
})
const orders = computed(() => ordersQuery.data.value?.items ?? [])
const total = computed(() => ordersQuery.data.value?.total ?? 0)
@@ -90,10 +64,6 @@ function refresh(): void {
void ordersQuery.refetch()
}
function syncBacklinks(): void {
void syncBacklinksMutation.mutateAsync()
}
function handleTableChange(nextPage: number, nextPageSize: number): void {
page.value = nextPage
pageSize.value = nextPageSize
@@ -119,7 +89,7 @@ function orderStatusMeta(order: MediaSupplyOrderDetail): { label: string; color:
? { label: '已发表', color: 'success' }
: { label: '发表中', color: 'processing' }
case 'failed':
return order.wallet_refunded_at
return order.wallet_refunded_at && order.supplier_status === 'rejected'
? { label: '已退稿', color: 'default' }
: { label: '投稿失败', color: 'error' }
default:
@@ -149,6 +119,52 @@ function pendingBacklinkCount(order: MediaSupplyOrderDetail): number {
return (order.items ?? []).filter((item) => !item.external_article_url).length
}
const shouldPollOrders = computed(() =>
orders.value.some(
(order) =>
order.status === 'queued' ||
order.status === 'submitting' ||
(order.status === 'submitted' && pendingBacklinkCount(order) > 0),
),
)
let pollingTimer: ReturnType<typeof setInterval> | undefined
function stopOrderPolling(): void {
if (pollingTimer) {
clearInterval(pollingTimer)
pollingTimer = undefined
}
}
function startOrderPolling(): void {
if (pollingTimer) {
return
}
pollingTimer = setInterval(() => {
if (typeof document !== 'undefined' && document.visibilityState === 'hidden') {
return
}
if (!ordersQuery.isFetching.value) {
void ordersQuery.refetch()
}
}, 10_000)
}
watch(
shouldPollOrders,
(enabled) => {
if (enabled) {
startOrderPolling()
} else {
stopOrderPolling()
}
},
{ immediate: true },
)
onBeforeUnmount(stopOrderPolling)
function mediaNames(order: MediaSupplyOrderDetail): string[] {
const seen = new Set<string>()
const names: string[] = []
@@ -233,15 +249,6 @@ function formatOrderError(message?: string | null): string {
<ReloadOutlined :class="{ 'refresh-spinning': ordersQuery.isFetching.value }" />
刷新
</button>
<button
type="button"
class="toolbar-btn-sync"
:disabled="syncBacklinksMutation.isPending.value"
@click="syncBacklinks"
>
<ReloadOutlined v-if="syncBacklinksMutation.isPending.value" class="refresh-spinning" />
同步状态/回链
</button>
</div>
</section>
@@ -320,7 +327,7 @@ function formatOrderError(message?: string | null): string {
<span v-else>{{ backlinkSummary(record) }}</span>
</template>
<template v-else-if="column.key === 'external_order_code'">
{{ record.external_order_code || record.external_order_id || '--' }}
{{ record.external_order_code || '--' }}
</template>
<template v-else-if="column.key === 'created_at'">
{{ formatDateTime(record.created_at) }}
+6 -2
View File
@@ -87,8 +87,12 @@ export const opsMediaSupplyApi = {
)
},
queueSync(modelId = 1) {
return http.post<{ job_id: number }, { model_id: number }>('/media-supply/sync-jobs', {
model_id: modelId,
return http.post<
{ id: number; job_key: string; status: string },
{ reason: string; config_override: { model_id: number } }
>('/scheduler/jobs/media_supply_cache_sync/run', {
reason: 'ops 手动同步媒体资源缓存',
config_override: { model_id: modelId },
})
},
listWallets(params: OpsMediaSupplyWalletsParams) {
@@ -461,7 +461,7 @@ async function queueSync(): Promise<void> {
syncLoading.value = true
try {
await opsMediaSupplyApi.queueSync(MODEL_ID_AUTHORITY_NEWS)
message.success('同步任务已提交')
message.success('调度任务已提交')
} catch (error) {
showOpsError(error)
} finally {
-12
View File
@@ -652,7 +652,6 @@ export interface MediaSupplyOrderDetail {
title: string
remark?: string | null
order_brand?: string | null
external_order_id?: string | null
external_order_code?: string | null
supplier_status?: string | null
cost_total_cents: number
@@ -684,17 +683,6 @@ export interface ListMediaSupplyOrdersResponse {
page_size: number
}
export interface SyncMediaSupplyBacklinksResponse {
published_fetched: number
pending_fetched: number
problem_fetched: number
orders_checked: number
orders_updated: number
order_codes_updated: number
refunded_orders: number
links_updated: number
}
export interface MediaSupplyWalletStatus {
tenant_id: number
user_id: number
+13
View File
@@ -181,6 +181,19 @@ func main() {
return schedulerRunRetentionWorker.RunOnce(runCtx, jobCtx)
},
},
{
Key: "media_supply_cache_sync",
Run: func(runCtx context.Context, jobCtx internalscheduler.JobRunContext) (map[string]any, error) {
modelID := internalscheduler.AsInt(jobCtx.Config, "model_id", 1)
if jobCtx.DryRun {
return map[string]any{
"model_id": modelID,
"synced": 0,
}, nil
}
return app.MediaSupplyService.RunMediaResourceSyncOnce(runCtx, modelID)
},
},
})
startSchedulerWorker(ctx, &workerWG, "control_plane", app.Logger.Sugar(), controlPlane.Run)
+38 -8
View File
@@ -85,7 +85,7 @@ func (s *SchedulerService) ListJobs(ctx context.Context, in SchedulerJobListInpu
Offset: (page - 1) * size,
})
if err != nil {
return nil, response.ErrInternal(53001, "scheduler_jobs_query_failed", "failed to list scheduler jobs")
return nil, response.ErrInternal(53001, "scheduler_jobs_query_failed", "调度任务列表读取失败")
}
return &SchedulerJobListResult{Items: items, Total: total, Page: page, Size: size}, nil
}
@@ -140,22 +140,37 @@ func (s *SchedulerService) SetEnabled(ctx context.Context, actor *Actor, key str
func (s *SchedulerService) Trigger(ctx context.Context, actor *Actor, key string, triggerType string, in SchedulerTriggerInput) (*domain.SchedulerTrigger, error) {
key = normalizeSchedulerJobKey(key)
if _, err := s.repo.GetJobForRuntime(ctx, key); err != nil {
job, err := s.repo.GetJobForRuntime(ctx, key)
if err != nil {
return nil, mapSchedulerJobError(err)
}
triggerType = strings.TrimSpace(triggerType)
if triggerType != domain.SchedulerTriggerManual && triggerType != domain.SchedulerTriggerDryRun {
return nil, response.ErrBadRequest(40083, "invalid_trigger_type", "trigger_type 不合法")
}
trigger, err := s.repo.CreateTrigger(ctx, repository.SchedulerTriggerCreate{
create := repository.SchedulerTriggerCreate{
JobKey: key,
TriggerType: triggerType,
RequestedBy: actorID(actor),
Reason: strings.TrimSpace(in.Reason),
ConfigOverride: sanitizeSchedulerConfig(in.ConfigOverride),
})
}
if triggerType == domain.SchedulerTriggerManual && boolFromSchedulerConfig(job.Config, "dedupe_manual_triggers") {
trigger, created, triggerErr := s.repo.CreateDedupedManualTrigger(ctx, create)
if triggerErr != nil {
return nil, response.ErrInternal(53004, "scheduler_trigger_create_failed", "创建调度触发失败")
}
if created {
s.auditScheduler(ctx, actor, ActionSchedulerJobRun, key, map[string]any{
"trigger_id": trigger.ID,
"reason": strings.TrimSpace(in.Reason),
})
}
return trigger, nil
}
trigger, err := s.repo.CreateTrigger(ctx, create)
if err != nil {
return nil, response.ErrInternal(53004, "scheduler_trigger_create_failed", "failed to create scheduler trigger")
return nil, response.ErrInternal(53004, "scheduler_trigger_create_failed", "创建调度触发失败")
}
action := ActionSchedulerJobRun
if triggerType == domain.SchedulerTriggerDryRun {
@@ -180,15 +195,30 @@ func (s *SchedulerService) ListRuns(ctx context.Context, key, status string, pag
Offset: (page - 1) * size,
})
if err != nil {
return nil, response.ErrInternal(53005, "scheduler_runs_query_failed", "failed to list scheduler runs")
return nil, response.ErrInternal(53005, "scheduler_runs_query_failed", "调度运行历史读取失败")
}
return &SchedulerRunListResult{Items: items, Total: total, Page: page, Size: size}, nil
}
func boolFromSchedulerConfig(config map[string]any, key string) bool {
value, ok := config[key]
if !ok || value == nil {
return false
}
switch typed := value.(type) {
case bool:
return typed
case string:
return strings.EqualFold(strings.TrimSpace(typed), "true")
default:
return false
}
}
func (s *SchedulerService) ListInstances(ctx context.Context) ([]domain.SchedulerInstance, error) {
items, err := s.repo.ListInstances(ctx)
if err != nil {
return nil, response.ErrInternal(53006, "scheduler_instances_query_failed", "failed to list scheduler instances")
return nil, response.ErrInternal(53006, "scheduler_instances_query_failed", "调度实例读取失败")
}
return items, nil
}
@@ -255,7 +285,7 @@ func mapSchedulerJobError(err error) error {
if errors.Is(err, repository.ErrSchedulerJobNotFound) {
return response.ErrNotFound(40480, "scheduler_job_not_found", "调度任务不存在")
}
return response.ErrInternal(53002, "scheduler_job_query_failed", "failed to load scheduler job")
return response.ErrInternal(53002, "scheduler_job_query_failed", "调度任务读取失败")
}
func actorID(actor *Actor) *int64 {
+109
View File
@@ -270,6 +270,115 @@ func (r *SchedulerRepository) CreateTrigger(ctx context.Context, in SchedulerTri
`, in.JobKey, in.TriggerType, in.RequestedBy, reason, configJSON))
}
func (r *SchedulerRepository) CreateDedupedManualTrigger(ctx context.Context, in SchedulerTriggerCreate) (*domain.SchedulerTrigger, bool, error) {
if in.TriggerType != domain.SchedulerTriggerManual {
trigger, err := r.CreateTrigger(ctx, in)
return trigger, true, err
}
tx, err := r.pool.Begin(ctx)
if err != nil {
return nil, false, err
}
defer tx.Rollback(ctx)
if _, err := tx.Exec(ctx, `SELECT pg_advisory_xact_lock(hashtext($1), hashtext($2))`, "scheduler_trigger", in.JobKey); err != nil {
return nil, false, err
}
if item, ok, err := findActiveTrigger(ctx, tx, in.JobKey, in.TriggerType); err != nil {
return nil, false, err
} else if ok {
return item, false, tx.Commit(ctx)
}
if _, ok, err := findRunningRun(ctx, tx, in.JobKey); err != nil {
return nil, false, err
} else if ok {
reason := in.Reason
if reason == "" {
reason = "已有调度任务正在执行,本次请求已合并"
}
item := &domain.SchedulerTrigger{
ID: 0,
JobKey: in.JobKey,
TriggerType: in.TriggerType,
Status: domain.SchedulerRunSkipped,
RequestedBy: in.RequestedBy,
Reason: nullableString(reason),
ConfigOverride: decodeSchedulerJSONMap(nil),
RequestedAt: time.Now().UTC(),
}
if in.ConfigOverride != nil {
item.ConfigOverride = in.ConfigOverride
}
return item, false, tx.Commit(ctx)
}
configJSON, err := marshalSchedulerJSONMap(in.ConfigOverride)
if err != nil {
return nil, false, err
}
trigger, err := scanSchedulerTrigger(tx.QueryRow(ctx, `
INSERT INTO ops.scheduler_job_triggers (job_key, trigger_type, requested_by, reason, config_override)
VALUES ($1, $2, $3, $4, COALESCE($5::jsonb, '{}'::jsonb))
RETURNING id, job_key, trigger_type, status, requested_by, reason, config_override,
scheduler_instance_id, run_id, requested_at, claimed_at, finished_at, error_message
`, in.JobKey, in.TriggerType, in.RequestedBy, nullableString(in.Reason), configJSON))
if err != nil {
return nil, false, err
}
return trigger, true, tx.Commit(ctx)
}
func (r *SchedulerRepository) FindActiveTrigger(ctx context.Context, jobKey, triggerType string) (*domain.SchedulerTrigger, bool, error) {
return findActiveTrigger(ctx, r.pool, jobKey, triggerType)
}
func findActiveTrigger(ctx context.Context, q interface {
QueryRow(context.Context, string, ...any) pgx.Row
}, jobKey, triggerType string) (*domain.SchedulerTrigger, bool, error) {
item, err := scanSchedulerTrigger(q.QueryRow(ctx, `
SELECT id, job_key, trigger_type, status, requested_by, reason, config_override,
scheduler_instance_id, run_id, requested_at, claimed_at, finished_at, error_message
FROM ops.scheduler_job_triggers
WHERE job_key = $1
AND trigger_type = $2
AND status IN ('pending', 'claimed')
ORDER BY requested_at ASC, id ASC
LIMIT 1
`, jobKey, triggerType))
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
return nil, false, nil
}
return nil, false, err
}
return item, true, nil
}
func (r *SchedulerRepository) FindRunningRun(ctx context.Context, jobKey string) (*domain.SchedulerRun, bool, error) {
return findRunningRun(ctx, r.pool, jobKey)
}
func findRunningRun(ctx context.Context, q interface {
QueryRow(context.Context, string, ...any) pgx.Row
}, jobKey string) (*domain.SchedulerRun, bool, error) {
item, err := scanSchedulerRun(q.QueryRow(ctx, `
SELECT id, job_key, trigger_id, scheduler_instance_id, trigger_type, status,
config_version, config_snapshot, stats, error_message, started_at,
finished_at, duration_ms
FROM ops.scheduler_job_runs
WHERE job_key = $1 AND status = 'running'
ORDER BY started_at DESC, id DESC
LIMIT 1
`, jobKey))
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
return nil, false, nil
}
return nil, false, err
}
return item, true, nil
}
func (r *SchedulerRepository) ClaimPendingTrigger(ctx context.Context, jobKey, instanceID string) (*SchedulerTriggerClaim, bool, error) {
var claim SchedulerTriggerClaim
var configRaw []byte
+108
View File
@@ -222,6 +222,27 @@ func (p *ControlPlane) nextAutoAt(ctx context.Context, key string, now time.Time
base := now.In(loc)
return schedule.Next(base)
}
if window, ok := parseRandomRunWindowLocal(job.Config); ok {
loc, locErr := time.LoadLocation(job.Timezone)
if locErr != nil {
loc = time.Local
}
var lastStartedAt *time.Time
if initial {
startedAt, ok, lastErr := p.repo.LastAutoRunStartedAt(ctx, key)
if lastErr != nil {
if p.logger != nil {
p.logger.Warn("scheduler last auto run lookup failed",
zap.String("job_key", key),
zap.Error(lastErr),
)
}
} else if ok {
lastStartedAt = &startedAt
}
}
return nextRandomWindowAutoAt(key, now, loc, window, initial, lastStartedAt)
}
if runAt := parseRunAtLocal(job.Config); runAt != "" {
loc, locErr := time.LoadLocation(job.Timezone)
if locErr != nil {
@@ -629,6 +650,93 @@ func parseRunAtLocal(config map[string]any) string {
return fmt.Sprint(value)
}
type randomRunWindowLocal struct {
StartHour int
StartMinute int
EndHour int
EndMinute int
}
func parseRandomRunWindowLocal(config map[string]any) (randomRunWindowLocal, bool) {
start := firstConfigString(config, "random_window_start", "random_window_start_local", "random_run_window_start")
end := firstConfigString(config, "random_window_end", "random_window_end_local", "random_run_window_end")
if start == "" || end == "" {
return randomRunWindowLocal{}, false
}
startHour, startMinute, startOK := parseHourMinute(start)
endHour, endMinute, endOK := parseHourMinute(end)
startTotal := startHour*60 + startMinute
endTotal := endHour*60 + endMinute
if !startOK || !endOK || endTotal <= startTotal {
return randomRunWindowLocal{}, false
}
return randomRunWindowLocal{
StartHour: startHour,
StartMinute: startMinute,
EndHour: endHour,
EndMinute: endMinute,
}, true
}
func nextRandomWindowAutoAt(key string, now time.Time, loc *time.Location, window randomRunWindowLocal, initial bool, lastStartedAt *time.Time) time.Time {
if loc == nil {
loc = time.Local
}
localNow := now.In(loc)
todayStart := time.Date(localNow.Year(), localNow.Month(), localNow.Day(), window.StartHour, window.StartMinute, 0, 0, loc)
todayEnd := time.Date(localNow.Year(), localNow.Month(), localNow.Day(), window.EndHour, window.EndMinute, 0, 0, loc)
if !todayEnd.After(todayStart) {
return now.Add(schedulerConfigRetryInterval)
}
todayCandidate := randomWindowCandidate(key, todayStart, todayEnd)
if !initial {
tomorrowStart := todayStart.AddDate(0, 0, 1)
return randomWindowCandidate(key, tomorrowStart, tomorrowStart.Add(todayEnd.Sub(todayStart)))
}
if lastStartedAt != nil && sameLocalDate((*lastStartedAt).In(loc), localNow) {
tomorrowStart := todayStart.AddDate(0, 0, 1)
return randomWindowCandidate(key, tomorrowStart, tomorrowStart.Add(todayEnd.Sub(todayStart)))
}
if localNow.Before(todayCandidate) {
return todayCandidate
}
if localNow.Before(todayEnd) {
return now
}
tomorrowStart := todayStart.AddDate(0, 0, 1)
return randomWindowCandidate(key, tomorrowStart, tomorrowStart.Add(todayEnd.Sub(todayStart)))
}
func randomWindowCandidate(key string, start, end time.Time) time.Time {
seconds := int64(end.Sub(start).Seconds())
if seconds <= 0 {
return start
}
hash := fnv.New64a()
_, _ = hash.Write([]byte("scheduler-random-window:" + key + ":" + start.Format("2006-01-02") + ":" + start.Format("15:04") + ":" + end.Format("15:04")))
offset := int64(hash.Sum64() % uint64(seconds))
return start.Add(time.Duration(offset) * time.Second)
}
func sameLocalDate(a, b time.Time) bool {
ay, am, ad := a.Date()
by, bm, bd := b.Date()
return ay == by && am == bm && ad == bd
}
func firstConfigString(config map[string]any, keys ...string) string {
for _, key := range keys {
value, ok := config[key]
if !ok || value == nil {
continue
}
if text := strings.TrimSpace(fmt.Sprint(value)); text != "" {
return text
}
}
return ""
}
func parseHourMinute(value string) (int, int, bool) {
parts := strings.Split(strings.TrimSpace(value), ":")
if len(parts) != 2 {
@@ -46,6 +46,46 @@ func TestNextIntervalAutoAtSchedulesAfterCurrentCompletion(t *testing.T) {
}
}
func TestNextRandomWindowAutoAtIsStableInsideWindow(t *testing.T) {
loc := time.FixedZone("CST", 8*60*60)
window, ok := parseRandomRunWindowLocal(map[string]any{
"random_window_start": "01:00",
"random_window_end": "04:00",
})
if !ok {
t.Fatal("expected random window config to parse")
}
now := time.Date(2026, 6, 2, 0, 30, 0, 0, loc)
got := nextRandomWindowAutoAt("media_supply_cache_sync", now, loc, window, true, nil)
gotAgain := nextRandomWindowAutoAt("media_supply_cache_sync", now.Add(10*time.Minute), loc, window, true, nil)
start := time.Date(2026, 6, 2, 1, 0, 0, 0, loc)
end := time.Date(2026, 6, 2, 4, 0, 0, 0, loc)
if got.Before(start) || !got.Before(end) {
t.Fatalf("nextRandomWindowAutoAt() = %s, want inside [%s, %s)", got, start, end)
}
if !got.Equal(gotAgain) {
t.Fatalf("expected stable candidate, got %s and %s", got, gotAgain)
}
}
func TestNextRandomWindowAutoAtSkipsTodayAfterAutoRun(t *testing.T) {
loc := time.FixedZone("CST", 8*60*60)
window, _ := parseRandomRunWindowLocal(map[string]any{
"random_window_start": "01:00",
"random_window_end": "04:00",
})
now := time.Date(2026, 6, 2, 2, 0, 0, 0, loc)
last := time.Date(2026, 6, 2, 1, 30, 0, 0, loc)
got := nextRandomWindowAutoAt("media_supply_cache_sync", now, loc, window, true, &last)
wantStart := time.Date(2026, 6, 3, 1, 0, 0, 0, loc)
wantEnd := time.Date(2026, 6, 3, 4, 0, 0, 0, loc)
if got.Before(wantStart) || !got.Before(wantEnd) {
t.Fatalf("nextRandomWindowAutoAt() = %s, want inside tomorrow window [%s, %s)", got, wantStart, wantEnd)
}
}
func TestIsEmptySuccessRunRequiresConfiguredZeroMetrics(t *testing.T) {
stats := map[string]any{
"hydrated_count": 0,
+1 -1
View File
@@ -844,7 +844,7 @@ func NormalizeMediaSupplyConfig(cfg *MediaSupplyConfig) {
cfg.Worker.SyncMaxAttempts = 2
}
if cfg.Worker.BacklinkSyncInterval <= 0 {
cfg.Worker.BacklinkSyncInterval = 10 * time.Minute
cfg.Worker.BacklinkSyncInterval = 30 * time.Minute
}
if cfg.Worker.BacklinkBatchSize <= 0 {
cfg.Worker.BacklinkBatchSize = 20
@@ -214,7 +214,6 @@ var routeDocs = map[string]routeDoc{
"POST /api/tenant/media-supply/orders": {"创建媒介投放订单", "为选定媒介资源创建投放订单。"},
"GET /api/tenant/media-supply/orders": {"媒介订单列表", "分页查询当前 Workspace 的媒介投放订单。"},
"GET /api/tenant/media-supply/orders/:id": {"媒介订单详情", "读取指定媒介投放订单详情。"},
"POST /api/tenant/media-supply/orders/sync-backlinks": {"同步媒介订单链接", "从供应商侧同步媒介订单的发布链接。"},
"GET /api/tenant/media-supply/wallet": {"媒介钱包状态", "读取当前 Workspace 的媒介供应钱包余额与状态。"},
"GET /api/tenant/media-supply/wallet/ledgers": {"媒介钱包流水", "分页查询媒介供应钱包收支流水。"},
"POST /api/tenant/media-supply/wallet/adjustments": {"调整媒介钱包", "创建媒介钱包余额调整记录。"},
@@ -469,7 +469,7 @@ func (s *ArticleImitationService) failGeneration(ctx context.Context, job articl
now := time.Now()
logCtx := articleImitationJobLogContext(job, stage, "", "failed", GenerationTaskResultFailure)
RecordGenerationTaskEvent(GenerationTaskEventFailure, logCtx)
LogGenerationTaskWarn(s.logger, "imitation generation failed", logCtx, zap.Error(genErr))
LogGenerationTaskWarn(s.logger, "imitation generation failed", logCtx, GenerationTaskErrorFields(genErr)...)
title := strings.TrimSpace(job.InitialTitle)
if title == "" {
title = articleImitationFallbackTitle
@@ -659,22 +659,18 @@ func buildImitationKnowledgeQuery(params map[string]interface{}) string {
return ""
}
parts := make([]string, 0, 8)
appendValue := func(value string) {
value = strings.TrimSpace(value)
if value == "" {
return
}
parts = append(parts, value)
intent := knowledgeQueryIntent{
TaskType: "仿写文章",
Title: extractString(params, "source_title"),
Topic: firstNonEmptyText(extractString(params, "source_title"), extractString(params, "extra_requirements")),
BrandName: extractString(params, "brand_name"),
Region: extractString(params, "region"),
PrimaryKeyword: firstNonEmptyText(extractString(params, "source_title"), knowledgeQueryKeywordText(extractStringList(params["keywords"], 16))),
Keywords: mergeKnowledgeQueryKeywords(params["keywords"]),
KeyPoints: extractString(params, "preserve_points"),
Extra: strings.Join([]string{extractString(params, "avoid_points"), extractString(params, "extra_requirements")}, "\n"),
}
appendValue(extractString(params, "source_title"))
appendValue(extractString(params, "brand_name"))
appendValue(extractString(params, "region"))
appendValue(strings.Join(extractStringList(params["keywords"], 16), "\n"))
appendValue(extractString(params, "extra_requirements"))
return strings.Join(parts, "\n")
return buildKnowledgeQueryInputText(intent)
}
func appendPromptLine(b *strings.Builder, label, value string) {
@@ -1,11 +1,14 @@
package app
import (
"errors"
"strings"
"sync/atomic"
"time"
"go.uber.org/zap"
"github.com/geo-platform/tenant-api/internal/shared/response"
)
const GenerationTaskStatusRunning = "running"
@@ -103,6 +106,30 @@ func LogGenerationTaskError(logger *zap.Logger, message string, ctx GenerationTa
logger.Error(message, append(GenerationTaskLogFields(ctx), fields...)...)
}
func GenerationTaskErrorFields(err error) []zap.Field {
if err == nil {
return []zap.Field{}
}
fields := []zap.Field{zap.Error(err)}
var appErr *response.AppError
if errors.As(err, &appErr) {
if appErr.Code > 0 {
fields = append(fields, zap.Int("error_code", appErr.Code))
}
if appErr.HTTPStatus > 0 {
fields = append(fields, zap.Int("http_status", appErr.HTTPStatus))
}
if message := strings.TrimSpace(appErr.Message); message != "" {
fields = append(fields, zap.String("error_message", message))
}
if detail := strings.TrimSpace(appErr.Detail); detail != "" {
fields = append(fields, zap.String("error_detail", detail))
}
}
return fields
}
type GenerationTaskMetricsSnapshot struct {
TransitionTotal int64 `json:"transition_total"`
FailureTotal int64 `json:"failure_total"`
@@ -4,6 +4,7 @@ import (
"strings"
"testing"
"github.com/geo-platform/tenant-api/internal/shared/response"
"github.com/stretchr/testify/assert"
)
@@ -91,6 +92,34 @@ func TestGenerationTaskMetricsRecordsEventsAndPrometheus(t *testing.T) {
assert.True(t, strings.HasSuffix(lastEventLine, " 1"), lastEventLine)
}
func TestFormatGenerationFailureKeepsAppErrorBriefForUI(t *testing.T) {
err := response.ErrServiceUnavailable(50353, "embedding_failed", "request siliconflow: timeout")
got := formatGenerationFailure("knowledge_resolve", err)
assert.Equal(t, "知识库检索失败 [knowledge_resolve]: embedding_failed", got)
}
func TestGenerationTaskErrorFieldsIncludesAppErrorDetailForLogs(t *testing.T) {
err := response.ErrServiceUnavailable(50353, "embedding_failed", "request siliconflow: timeout")
fields := GenerationTaskErrorFields(err)
values := map[string]any{}
for _, field := range fields {
switch {
case field.String != "":
values[field.Key] = field.String
case field.Integer != 0:
values[field.Key] = int(field.Integer)
}
}
assert.Equal(t, "embedding_failed", values["error_message"])
assert.Equal(t, "request siliconflow: timeout", values["error_detail"])
assert.Equal(t, 50353, values["error_code"])
assert.Equal(t, 503, values["http_status"])
}
func findPrometheusMetricLine(output, metricName string) string {
for _, line := range strings.Split(output, "\n") {
if strings.HasPrefix(line, metricName+"{") || strings.HasPrefix(line, metricName+" ") {
@@ -0,0 +1,293 @@
package app
import (
"encoding/json"
"fmt"
"strings"
)
const maxKnowledgeQueryQuestions = 3
type knowledgeQueryIntent struct {
TaskType string
TemplateName string
Title string
Topic string
BrandName string
Region string
PrimaryKeyword string
Keywords []string
KeyPoints string
Extra string
}
type KnowledgeQueryInput struct {
TaskType string
TemplateName string
Title string
Topic string
BrandName string
Region string
PrimaryKeyword string
Keywords []string
KeyPoints string
Extra string
}
func BuildKnowledgeQuery(input KnowledgeQueryInput) string {
intent := knowledgeQueryIntent{
TaskType: input.TaskType,
TemplateName: input.TemplateName,
Title: input.Title,
Topic: input.Topic,
BrandName: input.BrandName,
Region: input.Region,
PrimaryKeyword: input.PrimaryKeyword,
Keywords: input.Keywords,
KeyPoints: input.KeyPoints,
Extra: input.Extra,
}
return buildKnowledgeQueryInputText(intent)
}
func buildKnowledgeQueryQuestions(intent knowledgeQueryIntent) string {
questions := buildKnowledgeQueryQuestionList(intent)
return strings.Join(questions, "\n")
}
func buildKnowledgeQueryInputText(intent knowledgeQueryIntent) string {
intent = normalizeKnowledgeQueryIntent(intent)
lines := make([]string, 0, 12)
appendLine := func(label, value string) {
value = strings.TrimSpace(value)
if value != "" {
lines = append(lines, fmt.Sprintf("%s%s", label, value))
}
}
appendLine("文章类型", intent.TaskType)
appendLine("模板名称", intent.TemplateName)
appendLine("标题", intent.Title)
appendLine("主题", intent.Topic)
appendLine("品牌", intent.BrandName)
appendLine("地域", intent.Region)
appendLine("主关键词", intent.PrimaryKeyword)
appendLine("关键词", knowledgeQueryKeywordText(intent.Keywords))
appendLine("重点要求", intent.KeyPoints)
appendLine("其他要求", intent.Extra)
fallback := buildKnowledgeQueryQuestionList(intent)
if len(fallback) > 0 {
lines = append(lines, "兜底检索问题:")
for _, question := range fallback {
lines = append(lines, "- "+question)
}
}
return strings.Join(lines, "\n")
}
func buildKnowledgeQueryRewritePrompt(rawQuery string) string {
rawQuery = strings.TrimSpace(rawQuery)
if rawQuery == "" {
return ""
}
return strings.TrimSpace(fmt.Sprintf(`你是知识库检索 Query Rewrite 助手。请根据用户的文章生成输入,改写出 3 个最适合去企业/产品知识库中检索的自然语言问题。
目标:
1. 让检索问题覆盖“文章类型/写作任务”真正需要的知识,而不是只重复品牌名或地域词。
2. 优先围绕用户输入的标题、主题、品牌、地域、关键词、重点要求生成问题。
3. 问题应适合在知识库中召回:业务范围、产品服务、优势案例、地址联系方式、资质事实、用户痛点、对比维度、避坑点、适用场景。
4. 如果输入中出现“兜底检索问题”,只能作为参考;你需要根据上方结构化字段重新生成更贴近本次文章的 3 个问题。
规则:
- 必须输出 3 个问题。
- 每个问题 18-60 个中文字符,尽量具体。
- 不要生成泛泛的问题,如“这个品牌怎么样”。
- 不要编造输入中没有的品牌、地域、产品或事实。
- 只输出 JSON,不要 Markdown,不要解释。
用户输入:
%s`, rawQuery))
}
func parseKnowledgeQueryRewriteOutput(content string) []string {
content = strings.TrimSpace(content)
if content == "" {
return []string{}
}
content = strings.TrimPrefix(content, "```json")
content = strings.TrimPrefix(content, "```")
content = strings.TrimSuffix(content, "```")
content = strings.TrimSpace(content)
var payload struct {
Questions []string `json:"questions"`
}
if err := json.Unmarshal([]byte(content), &payload); err == nil && len(payload.Questions) > 0 {
return normalizeKnowledgeQueryQuestions(payload.Questions)
}
var list []string
if err := json.Unmarshal([]byte(content), &list); err == nil && len(list) > 0 {
return normalizeKnowledgeQueryQuestions(list)
}
return []string{}
}
func normalizeKnowledgeQueryQuestions(values []string) []string {
questions := make([]string, 0, maxKnowledgeQueryQuestions)
for _, value := range values {
appendKnowledgeQuestion(&questions, value)
if len(questions) >= maxKnowledgeQueryQuestions {
break
}
}
return questions
}
func buildKnowledgeQueryQuestionList(intent knowledgeQueryIntent) []string {
intent = normalizeKnowledgeQueryIntent(intent)
questions := make([]string, 0, maxKnowledgeQueryQuestions)
subject := firstNonEmptyText(intent.Topic, intent.Title, intent.PrimaryKeyword, intent.BrandName, "当前主题")
brandScope := firstNonEmptyText(joinKnowledgeQueryParts("", intent.BrandName, intent.Region), intent.BrandName, subject)
keywordScope := knowledgeQueryKeywordText(intent.Keywords)
switch {
case strings.Contains(intent.TaskType, "推荐榜") || strings.Contains(strings.ToLower(intent.TaskType), "top"):
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 中有哪些与 %s 相关的品牌、产品、服务能力和选型依据?", subject, brandScope))
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 写作时需要引用哪些关于 %s 的准确事实、优势、案例、地址或联系方式?", intent.TaskType, firstNonEmptyText(intent.BrandName, subject)))
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 相关关键词 %s 对应的用户关注点、避坑点和对比维度是什么?", subject, firstNonEmptyText(keywordScope, intent.PrimaryKeyword, subject)))
case strings.Contains(intent.TaskType, "评测") || strings.Contains(strings.ToLower(intent.TaskType), "review"):
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 的产品特性、参数、卖点、适用人群和使用场景是什么?", subject))
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 有哪些真实体验、优缺点、案例、价格或售后信息可用于评测?", firstNonEmptyText(intent.BrandName, subject)))
appendKnowledgeQuestion(&questions, fmt.Sprintf("围绕 %s 和 %s,用户最关心哪些购买决策问题?", subject, firstNonEmptyText(keywordScope, intent.PrimaryKeyword, subject)))
case strings.Contains(intent.TaskType, "仿写"):
appendKnowledgeQuestion(&questions, fmt.Sprintf("仿写 %s 时,%s 有哪些必须保留或补充的品牌事实和业务信息?", subject, firstNonEmptyText(intent.BrandName, subject)))
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 相关关键词 %s 对应的产品、服务、案例和用户需求是什么?", subject, firstNonEmptyText(keywordScope, intent.PrimaryKeyword, subject)))
appendKnowledgeQuestion(&questions, fmt.Sprintf("生成 %s 时需要避免遗漏哪些地址、联系方式、优势、资质或落地案例?", firstNonEmptyText(intent.TaskType, "仿写文章")))
default:
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 这篇%s需要引用哪些核心事实、产品服务、案例和用户关注点?", subject, firstNonEmptyText(intent.TaskType, "文章")))
appendKnowledgeQuestion(&questions, fmt.Sprintf("%s 与 %s 相关的准确资料、品牌信息、地址、联系方式和业务范围是什么?", firstNonEmptyText(intent.BrandName, subject), firstNonEmptyText(keywordScope, intent.PrimaryKeyword, subject)))
appendKnowledgeQuestion(&questions, fmt.Sprintf("围绕 %s,用户最可能关心哪些问题、痛点、对比维度和解决方案?", firstNonEmptyText(keywordScope, subject)))
}
appendKnowledgeQuestion(&questions, intent.KeyPoints)
appendKnowledgeQuestion(&questions, intent.Extra)
return questions
}
func normalizeKnowledgeQueryIntent(intent knowledgeQueryIntent) knowledgeQueryIntent {
intent.TaskType = strings.TrimSpace(intent.TaskType)
intent.TemplateName = strings.TrimSpace(intent.TemplateName)
intent.Title = strings.TrimSpace(intent.Title)
intent.Topic = strings.TrimSpace(intent.Topic)
intent.BrandName = strings.TrimSpace(intent.BrandName)
intent.Region = strings.TrimSpace(intent.Region)
intent.PrimaryKeyword = strings.TrimSpace(intent.PrimaryKeyword)
intent.KeyPoints = strings.TrimSpace(intent.KeyPoints)
intent.Extra = strings.TrimSpace(intent.Extra)
intent.Keywords = normalizeKnowledgeQueryKeywords(intent.Keywords, 6)
if intent.TaskType == "" {
intent.TaskType = strings.TrimSpace(intent.TemplateName)
}
return intent
}
func appendKnowledgeQuestion(questions *[]string, question string) {
if questions == nil || len(*questions) >= maxKnowledgeQueryQuestions {
return
}
question = normalizeKnowledgeQueryQuestion(question)
if question == "" {
return
}
for _, existing := range *questions {
if normalizeKnowledgeQueryQuestion(existing) == question {
return
}
}
*questions = append(*questions, question)
}
func normalizeKnowledgeQueryQuestion(question string) string {
question = strings.TrimSpace(question)
if question == "" {
return ""
}
question = strings.Join(strings.Fields(question), " ")
for _, bad := range []string{" ", " ", " ", " ?"} {
question = strings.ReplaceAll(question, bad, strings.TrimSpace(bad))
}
question = strings.Trim(question, " ,;")
if question == "" {
return ""
}
if !strings.HasSuffix(question, "") && !strings.HasSuffix(question, "?") {
question += ""
}
return question
}
func mergeKnowledgeQueryKeywords(values ...interface{}) []string {
keywords := make([]string, 0)
for _, value := range values {
keywords = append(keywords, extractStringList(value, 16)...)
if text := strings.TrimSpace(formatPromptValue(value)); text != "" && len(extractStringList(value, 1)) == 0 {
keywords = append(keywords, text)
}
}
return normalizeKnowledgeQueryKeywords(keywords, 6)
}
func normalizeKnowledgeQueryKeywords(keywords []string, limit int) []string {
result := make([]string, 0, len(keywords))
seen := make(map[string]struct{}, len(keywords))
for _, keyword := range keywords {
keyword = strings.TrimSpace(keyword)
if keyword == "" {
continue
}
key := strings.ToLower(strings.Join(strings.Fields(keyword), " "))
if _, ok := seen[key]; ok {
continue
}
seen[key] = struct{}{}
result = append(result, keyword)
if limit > 0 && len(result) >= limit {
break
}
}
return result
}
func knowledgeQueryKeywordText(keywords []string) string {
keywords = normalizeKnowledgeQueryKeywords(keywords, 6)
return strings.Join(keywords, "、")
}
func firstNonEmptyPromptString(params map[string]interface{}, keys ...string) string {
for _, key := range keys {
if text := strings.TrimSpace(stringValue(params[key])); text != "" {
return text
}
}
return ""
}
func firstNonEmptyText(values ...string) string {
for _, value := range values {
if text := strings.TrimSpace(value); text != "" {
return text
}
}
return ""
}
func joinKnowledgeQueryParts(sep string, values ...string) string {
parts := make([]string, 0, len(values))
for _, value := range values {
if text := strings.TrimSpace(value); text != "" {
parts = append(parts, text)
}
}
return strings.Join(parts, sep)
}
@@ -0,0 +1,141 @@
package app
import (
"context"
"errors"
"strings"
"testing"
"github.com/geo-platform/tenant-api/internal/shared/llm"
)
func TestBuildGenerationKnowledgeQueryIncludesContextAndFallbackQuestions(t *testing.T) {
query := buildGenerationKnowledgeQuery("top_x_article", "Top X 推荐文章", map[string]interface{}{
"topic": "合肥全屋定制",
"brand_name": "安徽海翔家居",
"region": "合肥",
"primary_keyword": "合肥定制家具",
"supplemental_questions": []string{"合肥家具品牌怎么选", "全屋定制避坑"},
"key_points": "突出门店与服务能力",
})
for _, expected := range []string{
"文章类型:推荐榜文章",
"模板名称:Top X 推荐文章",
"主题:合肥全屋定制",
"品牌:安徽海翔家居",
"地域:合肥",
"主关键词:合肥定制家具",
"兜底检索问题:",
} {
if !strings.Contains(query, expected) {
t.Fatalf("buildGenerationKnowledgeQuery() = %q, want %q", query, expected)
}
}
questions := normalizeKnowledgeResolveQueries(query)
if len(questions) != maxKnowledgeQueryQuestions {
t.Fatalf("normalizeKnowledgeResolveQueries() len = %d, want %d: %#v", len(questions), maxKnowledgeQueryQuestions, questions)
}
for _, question := range questions {
if !strings.HasSuffix(question, "") {
t.Fatalf("question = %q, want Chinese question mark", question)
}
}
}
func TestParseKnowledgeQueryRewriteOutputAcceptsObjectAndArray(t *testing.T) {
objectQuestions := parseKnowledgeQueryRewriteOutput(`{"questions":["安徽海翔家居在合肥有哪些门店和联系方式?","安徽海翔家居主营哪些全屋定制产品和服务?","合肥全屋定制用户选择时关注哪些案例和售后?"]}`)
if len(objectQuestions) != maxKnowledgeQueryQuestions {
t.Fatalf("object questions len = %d, want %d: %#v", len(objectQuestions), maxKnowledgeQueryQuestions, objectQuestions)
}
if objectQuestions[0] != "安徽海翔家居在合肥有哪些门店和联系方式?" {
t.Fatalf("first object question = %q", objectQuestions[0])
}
arrayQuestions := parseKnowledgeQueryRewriteOutput(`["安徽海翔家居有什么品牌优势?","合肥全屋定制有哪些案例可引用?","全屋定制选型需要哪些避坑信息?"]`)
if len(arrayQuestions) != maxKnowledgeQueryQuestions {
t.Fatalf("array questions len = %d, want %d: %#v", len(arrayQuestions), maxKnowledgeQueryQuestions, arrayQuestions)
}
}
func TestNormalizeKnowledgeResolveQueriesPrefersFallbackQuestionBlock(t *testing.T) {
query := strings.Join([]string{
"文章类型:推荐榜文章",
"主题:合肥全屋定制",
"品牌:安徽海翔家居",
"重点要求:",
"请突出本地服务",
"不要写成硬广",
"兜底检索问题:",
"- 安徽海翔家居在合肥有哪些门店和联系方式?",
"- 安徽海翔家居主营哪些全屋定制产品和服务?",
"- 合肥全屋定制用户选择时关注哪些案例和售后?",
}, "\n")
got := normalizeKnowledgeResolveQueries(query)
want := []string{
"安徽海翔家居在合肥有哪些门店和联系方式?",
"安徽海翔家居主营哪些全屋定制产品和服务?",
"合肥全屋定制用户选择时关注哪些案例和售后?",
}
if len(got) != len(want) {
t.Fatalf("normalizeKnowledgeResolveQueries() = %#v, want %#v", got, want)
}
for index := range want {
if got[index] != want[index] {
t.Fatalf("normalizeKnowledgeResolveQueries()[%d] = %q, want %q", index, got[index], want[index])
}
}
}
func TestRewriteKnowledgeResolveQueriesUsesKnowledgeURLModelAndJSONObject(t *testing.T) {
client := &fakeKnowledgeQueryLLM{
content: `{"questions":["安徽海翔家居在合肥有哪些门店和联系方式?","安徽海翔家居主营哪些全屋定制产品和服务?","合肥全屋定制用户选择时关注哪些案例和售后?"]}`,
}
svc := &KnowledgeService{
llmClient: client,
cfg: knowledgeRuntimeConfig{URLMarkdownModel: "knowledge-url-fast"},
}
questions, err := svc.rewriteKnowledgeResolveQueries(context.Background(), "品牌:安徽海翔家居\n地域:合肥")
if err != nil {
t.Fatalf("rewriteKnowledgeResolveQueries() error = %v", err)
}
if len(questions) != maxKnowledgeQueryQuestions {
t.Fatalf("questions len = %d, want %d: %#v", len(questions), maxKnowledgeQueryQuestions, questions)
}
if client.request.Model != "knowledge-url-fast" {
t.Fatalf("Generate model = %q, want knowledge-url-fast", client.request.Model)
}
if client.request.ResponseFormat == nil || client.request.ResponseFormat.Type != llm.ResponseFormatTypeJSONObject {
t.Fatalf("ResponseFormat = %#v, want json_object", client.request.ResponseFormat)
}
if !strings.Contains(client.request.Prompt, "品牌:安徽海翔家居") || !strings.Contains(client.request.Prompt, "必须输出 3 个问题") {
t.Fatalf("Prompt = %q, want raw user input and three-question instruction", client.request.Prompt)
}
}
type fakeKnowledgeQueryLLM struct {
request llm.GenerateRequest
content string
err error
}
func (f *fakeKnowledgeQueryLLM) Validate() error {
if f.err != nil {
return f.err
}
return nil
}
func (f *fakeKnowledgeQueryLLM) Generate(_ context.Context, req llm.GenerateRequest, _ func(string)) (*llm.GenerateResult, error) {
f.request = req
if f.err != nil {
return nil, f.err
}
if strings.TrimSpace(f.content) == "" {
return nil, errors.New("empty fake content")
}
return &llm.GenerateResult{Content: f.content, Model: req.Model}, nil
}
+150 -18
View File
@@ -836,9 +836,34 @@ func (s *KnowledgeService) ResolveContext(
if len(requestedGroupIDs) == 0 {
return nil, nil
}
if strings.TrimSpace(query) == "" {
queries := normalizeKnowledgeResolveQueries(query)
if len(queries) == 0 {
return nil, nil
}
queryRewriteStage := "fallback"
queryRewriteModel := strings.TrimSpace(s.runtimeConfig().URLMarkdownModel)
if rewritten, err := s.rewriteKnowledgeResolveQueries(ctx, query); err != nil {
if s.logger != nil {
s.logger.Warn("knowledge query rewrite failed",
zap.String("query", knowledgeLogPreview(query, 600)),
zap.String("query_rewrite_model", queryRewriteModel),
zap.Strings("fallback_queries", queries),
zap.Error(err),
)
}
} else if len(rewritten) >= maxKnowledgeQueryQuestions {
queries = rewritten
queryRewriteStage = "model"
} else if len(rewritten) > 0 && s.logger != nil {
s.logger.Warn("knowledge query rewrite returned insufficient questions",
zap.String("query", knowledgeLogPreview(query, 600)),
zap.String("query_rewrite_model", queryRewriteModel),
zap.Strings("model_queries", rewritten),
zap.Int("model_query_count", len(rewritten)),
zap.Strings("fallback_queries", queries),
)
}
rerankQuery := strings.Join(queries, "\n")
if err := s.provider.Validate(); err != nil {
return nil, response.ErrServiceUnavailable(50351, "retrieval_unavailable", err.Error())
}
@@ -861,32 +886,39 @@ func (s *KnowledgeService) ResolveContext(
PreciseFacts: s.collectKnowledgeContextPreciseFacts(ctx, tenantID, searchGroupIDs, nil),
}
vectors, err := s.provider.Embed(ctx, []string{query})
vectors, err := s.provider.Embed(ctx, queries)
if err != nil {
return nil, response.ErrServiceUnavailable(50353, "embedding_failed", err.Error())
}
if len(vectors) == 0 || len(vectors[0]) == 0 {
return nil, response.ErrServiceUnavailable(50353, "embedding_failed", "embedding vector is empty")
if len(vectors) != len(queries) {
return nil, response.ErrServiceUnavailable(50353, "embedding_failed", "embedding result count mismatch")
}
runtimeCfg := s.runtimeConfig()
searchPoints, err := s.store.Search(ctx, retrieval.SearchRequest{
Vector: vectors[0],
TenantID: tenantID,
GroupIDs: searchGroupIDs,
Limit: runtimeCfg.RecallLimit,
})
if err != nil {
return nil, response.ErrServiceUnavailable(50354, "qdrant_search_failed", err.Error())
searchPoints := make([]retrieval.SearchPoint, 0, len(queries)*runtimeCfg.RecallLimit)
for index, vector := range vectors {
if len(vector) == 0 {
return nil, response.ErrServiceUnavailable(50353, "embedding_failed", fmt.Sprintf("embedding vector is empty at query index %d", index))
}
points, err := s.store.Search(ctx, retrieval.SearchRequest{
Vector: vector,
TenantID: tenantID,
GroupIDs: searchGroupIDs,
Limit: runtimeCfg.RecallLimit,
})
if err != nil {
return nil, response.ErrServiceUnavailable(50354, "qdrant_search_failed", err.Error())
}
searchPoints = append(searchPoints, points...)
}
if len(searchPoints) == 0 {
s.logKnowledgeResolve(baseContext, tenantID, query, "no_search_points", nil, nil)
s.logKnowledgeResolve(baseContext, tenantID, rerankQuery, "no_search_points", queryRewriteStage, queryRewriteModel, nil, nil)
return baseContext, nil
}
candidates := dedupeKnowledgeSearchCandidates(knowledgeSnippetsFromSearchPoints(searchPoints))
if len(candidates) == 0 {
s.logKnowledgeResolve(baseContext, tenantID, query, "no_candidate_text", nil, nil)
s.logKnowledgeResolve(baseContext, tenantID, rerankQuery, "no_candidate_text", queryRewriteStage, queryRewriteModel, nil, nil)
return baseContext, nil
}
@@ -895,7 +927,7 @@ func (s *KnowledgeService) ResolveContext(
return nil, err
}
if len(candidates) == 0 {
s.logKnowledgeResolve(baseContext, tenantID, query, "no_active_candidates", nil, nil)
s.logKnowledgeResolve(baseContext, tenantID, rerankQuery, "no_active_candidates", queryRewriteStage, queryRewriteModel, nil, nil)
return baseContext, nil
}
@@ -904,13 +936,13 @@ func (s *KnowledgeService) ResolveContext(
documents = append(documents, candidate.RerankText())
}
reranked, err := s.provider.Rerank(ctx, query, documents, minInt(runtimeCfg.RerankTopN, len(documents)))
reranked, err := s.provider.Rerank(ctx, rerankQuery, documents, minInt(runtimeCfg.RerankTopN, len(documents)))
if err != nil {
selected := selectKnowledgeCandidateSnippets(candidates, runtimeCfg.RerankTopN)
selected = s.enrichKnowledgePromptSnippets(ctx, tenantID, selected)
baseContext.Snippets = selected
baseContext.PreciseFacts = s.collectKnowledgeContextPreciseFacts(ctx, tenantID, searchGroupIDs, selected)
s.logKnowledgeResolve(baseContext, tenantID, query, "rerank_failed_fallback", candidates, selected)
s.logKnowledgeResolve(baseContext, tenantID, rerankQuery, "rerank_failed_fallback", queryRewriteStage, queryRewriteModel, candidates, selected)
return baseContext, nil
}
@@ -918,21 +950,48 @@ func (s *KnowledgeService) ResolveContext(
selected = s.enrichKnowledgePromptSnippets(ctx, tenantID, selected)
baseContext.Snippets = selected
baseContext.PreciseFacts = s.collectKnowledgeContextPreciseFacts(ctx, tenantID, searchGroupIDs, selected)
s.logKnowledgeResolve(baseContext, tenantID, query, "resolved", candidates, selected)
s.logKnowledgeResolve(baseContext, tenantID, rerankQuery, "resolved", queryRewriteStage, queryRewriteModel, candidates, selected)
return baseContext, nil
}
func (s *KnowledgeService) rewriteKnowledgeResolveQueries(ctx context.Context, query string) ([]string, error) {
if s == nil || s.llmClient == nil {
return []string{}, nil
}
prompt := buildKnowledgeQueryRewritePrompt(query)
if prompt == "" {
return []string{}, nil
}
runtimeCfg := s.runtimeConfig()
result, err := s.llmClient.Generate(ctx, llm.GenerateRequest{
Model: runtimeCfg.URLMarkdownModel,
Prompt: prompt,
Timeout: 15 * time.Second,
MaxOutputTokens: 600,
ResponseFormat: &llm.ResponseFormat{
Type: llm.ResponseFormatTypeJSONObject,
},
}, nil)
if err != nil {
return nil, err
}
return parseKnowledgeQueryRewriteOutput(result.Content), nil
}
func (s *KnowledgeService) logKnowledgeResolve(
knowledgeCtx *KnowledgeContext,
tenantID int64,
query string,
stage string,
queryRewriteStage string,
queryRewriteModel string,
candidates []KnowledgeSnippet,
selected []KnowledgeSnippet,
) {
if s == nil || s.logger == nil || knowledgeCtx == nil {
return
}
queries := normalizeKnowledgeResolveQueries(query)
s.logger.Info("knowledge resolve result",
zap.String("stage", strings.TrimSpace(stage)),
@@ -940,6 +999,10 @@ func (s *KnowledgeService) logKnowledgeResolve(
zap.Int64s("group_ids", knowledgeCtx.GroupIDs),
zap.Strings("group_names", knowledgeCtx.GroupNames),
zap.String("query", knowledgeLogPreview(query, 600)),
zap.Strings("queries", queries),
zap.Int("query_count", len(queries)),
zap.String("query_rewrite_stage", strings.TrimSpace(queryRewriteStage)),
zap.String("query_rewrite_model", strings.TrimSpace(queryRewriteModel)),
zap.Int("candidate_count", len(candidates)),
zap.Int("selected_count", len(selected)),
zap.Int("precise_fact_count", len(knowledgeCtx.PreciseFacts)),
@@ -947,6 +1010,75 @@ func (s *KnowledgeService) logKnowledgeResolve(
)
}
func normalizeKnowledgeResolveQueries(query string) []string {
lines := strings.Split(strings.TrimSpace(query), "\n")
queries := make([]string, 0, minInt(maxKnowledgeQueryQuestions, len(lines)))
seen := make(map[string]struct{}, len(lines))
for index, line := range lines {
if strings.TrimSuffix(strings.TrimSpace(line), "") != "兜底检索问题" {
continue
}
for _, fallbackLine := range lines[index+1:] {
text := strings.TrimSpace(fallbackLine)
if text == "" {
continue
}
if !strings.HasPrefix(text, "-") {
break
}
appendNormalizedKnowledgeResolveQuery(&queries, seen, strings.TrimSpace(strings.TrimPrefix(text, "-")))
if len(queries) >= maxKnowledgeQueryQuestions {
return queries
}
}
if len(queries) > 0 {
return queries
}
}
inFallbackQuestions := false
for _, line := range lines {
text := strings.TrimSpace(line)
if text == "" {
continue
}
if strings.TrimSuffix(text, "") == "兜底检索问题" {
inFallbackQuestions = true
continue
}
if strings.Contains(text, "") && !strings.HasPrefix(text, "-") && !inFallbackQuestions {
continue
}
if inFallbackQuestions && !strings.HasPrefix(text, "-") {
continue
}
text = strings.TrimPrefix(text, "-")
text = strings.TrimSpace(text)
appendNormalizedKnowledgeResolveQuery(&queries, seen, text)
if len(queries) >= maxKnowledgeQueryQuestions {
break
}
}
return queries
}
func appendNormalizedKnowledgeResolveQuery(queries *[]string, seen map[string]struct{}, text string) {
if queries == nil || len(*queries) >= maxKnowledgeQueryQuestions {
return
}
text = strings.TrimSpace(text)
if text == "" {
return
}
key := strings.ToLower(strings.Join(strings.Fields(text), " "))
if _, exists := seen[key]; exists {
return
}
seen[key] = struct{}{}
*queries = append(*queries, text)
}
// for debug
func summarizeKnowledgeSnippetsForLog(snippets []KnowledgeSnippet) []map[string]any {
if len(snippets) == 0 {
@@ -45,7 +45,7 @@ func mediaSupplyPublicErrorMessage(err error) string {
case errors.Is(err, errMeijiequanOrderRejected):
return "媒体供应商拒绝投稿,请检查文章内容或联系管理员处理"
case errors.Is(err, errMediaSupplySelectedResourcesMissing):
return "媒体资源价格刷新失败,请稍后重试或重新选择媒体"
return "媒体资源暂不可用或已下架,请重新选择媒体"
case errors.Is(err, errMediaSupplyLockedSellBelowCost):
return "媒体成本价已变化,当前锁定价格低于成本,请重新提交"
case errors.Is(err, errMediaSupplyAccountUserIDMissing):
@@ -63,7 +63,7 @@ func mediaSupplyPublicGenericErrorMessage(message string) string {
lower := strings.ToLower(message)
switch {
case strings.Contains(lower, "failed to refresh supplier price"):
return "媒体资源价格刷新失败,请稍后重试或重新选择媒体"
return "媒体资源暂不可用或已下架,请重新选择媒体"
case strings.Contains(lower, "supplier cost increased above locked sell price"):
return "媒体成本价已变化,当前锁定价格低于成本,请重新提交"
case strings.Contains(lower, "username") || strings.Contains(lower, "password") || strings.Contains(lower, "credentials"):
@@ -22,6 +22,7 @@ import (
type MediaSupplyService struct {
pool *pgxpool.Pool
redis *goredis.Client
logger *zap.Logger
cfg config.Provider
client *MeijiequanClient
@@ -35,6 +36,7 @@ func NewMediaSupplyService(pool *pgxpool.Pool, redis *goredis.Client, logger *za
client := NewMeijiequanClient(redis, current.MediaSupply.Meijiequan)
return &MediaSupplyService{
pool: pool,
redis: redis,
logger: logger,
cfg: cfg,
client: client,
@@ -403,19 +405,6 @@ func (s *MediaSupplyService) SessionStatus(ctx context.Context) (*MeijiequanSess
return status, nil
}
func (s *MediaSupplyService) SyncBacklinks(ctx context.Context) (*SyncMediaSupplyBacklinksResponse, error) {
if !isTenantAdmin(auth.MustActor(ctx)) {
return nil, response.ErrForbidden(40360, "media_supply_admin_required", "需要租户管理员权限")
}
_ = s.mediaSupplyConfig()
worker := NewMediaSupplyWorker(s, s.pool, s.logger)
result, err := worker.syncPublishedBacklinksNow(ctx)
if err != nil {
return nil, mediaSupplyClientError(err)
}
return result, nil
}
func (s *MediaSupplyService) SetResourcePrice(ctx context.Context, resourceID int64, req SetSupplierMediaPriceRequest) error {
if actor, ok := auth.ActorFromCtx(ctx); ok && !isTenantAdmin(actor) {
return response.ErrForbidden(40360, "media_supply_admin_required", "需要租户管理员权限")
@@ -982,18 +971,34 @@ func (s *MediaSupplyService) GetOrder(ctx context.Context, orderID int64) (*Medi
return &item, nil
}
type mediaResourceSyncStats struct {
Synced int
RemovedPriceOverride int
}
func (s *MediaSupplyService) UpsertSyncedResources(ctx context.Context, resources []UpsertSupplierMediaResourceInput) (int, error) {
stats, err := s.upsertSyncedResourcesWithStats(ctx, resources)
return stats.Synced, err
}
func (s *MediaSupplyService) upsertSyncedResourcesWithStats(ctx context.Context, resources []UpsertSupplierMediaResourceInput) (mediaResourceSyncStats, error) {
if len(resources) == 0 {
return 0, nil
return mediaResourceSyncStats{}, nil
}
count := 0
tx, err := s.pool.Begin(ctx)
if err != nil {
return mediaResourceSyncStats{}, err
}
defer tx.Rollback(ctx)
stats := mediaResourceSyncStats{}
for _, resource := range resources {
raw := compactJSON(resource.Raw)
costPrices, _ := json.Marshal(resource.CostPrices)
if len(costPrices) == 0 {
costPrices = []byte(`{}`)
}
if _, err := s.pool.Exec(ctx, `
var resourceID int64
if err := tx.QueryRow(ctx, `
INSERT INTO supplier_media_resources (
supplier, supplier_resource_id, model_id, name, status, cost_price_cents,
cost_prices_json, sale_price_label, resource_url, baidu_weight, resource_remark,
@@ -1022,18 +1027,111 @@ func (s *MediaSupplyService) UpsertSyncedResources(ctx context.Context, resource
raw_json = EXCLUDED.raw_json,
updated_at = NOW(),
deleted_at = NULL
RETURNING id
`, resource.Supplier, resource.SupplierResourceID, resource.ModelID, resource.Name, resource.Status,
resource.CostPriceCents, costPrices, nullableTrimmedString(resource.SalePriceLabel),
nullableTrimmedString(resource.ResourceURL), resource.BaiduWeight, nullableTrimmedString(resource.ResourceRemark),
nullableTrimmedString(resource.ChannelType), nullableTrimmedString(resource.Region),
nullableTrimmedString(resource.InclusionEffect), nullableTrimmedString(resource.LinkType),
nullableTrimmedString(resource.PublishRate), nullableTrimmedString(resource.DeliverySpeed),
resource.SupplierUpdatedAt, raw); err != nil {
return count, err
resource.SupplierUpdatedAt, raw).Scan(&resourceID); err != nil {
return stats, err
}
count++
removed, err := s.deletePriceOverridesBelowCost(ctx, tx, resourceID)
if err != nil {
return stats, err
}
stats.RemovedPriceOverride += removed
stats.Synced++
}
return count, nil
if err := tx.Commit(ctx); err != nil {
return stats, err
}
return stats, nil
}
func (s *MediaSupplyService) deletePriceOverridesBelowCost(ctx context.Context, q interface {
Exec(context.Context, string, ...any) (pgconn.CommandTag, error)
}, resourceID int64) (int, error) {
if s == nil || q == nil || resourceID <= 0 {
return 0, nil
}
tag, err := q.Exec(ctx, `
DELETE FROM supplier_media_price_overrides o
USING supplier_media_resources r
WHERE o.resource_id = r.id
AND r.id = $1
AND r.supplier = $2
AND r.deleted_at IS NULL
AND o.enabled = TRUE
AND o.sell_price_cents < COALESCE(
CASE
WHEN o.price_type = 'price' THEN COALESCE(NULLIF(r.cost_prices_json->>'price', '')::bigint, NULLIF(r.cost_prices_json->>'sale_price', '')::bigint)
ELSE NULLIF(r.cost_prices_json->>o.price_type, '')::bigint
END,
r.cost_price_cents
)
`, resourceID, mediaSupplySupplierMeijiequan)
return int(tag.RowsAffected()), err
}
func (s *MediaSupplyService) RunMediaResourceSyncOnce(ctx context.Context, modelID int) (map[string]any, error) {
if s == nil || s.pool == nil || s.client == nil {
return nil, response.ErrServiceUnavailable(50360, "media_supply_store_unavailable", "媒体投稿服务暂不可用")
}
cfg := s.mediaSupplyConfig()
if !cfg.Enabled {
return map[string]any{"synced": 0, "model_id": modelID, "disabled": true}, nil
}
if modelID <= 0 {
modelID = 1
}
if status, statusErr := s.client.SessionStatus(ctx); statusErr != nil || status == nil || !status.Available {
if loginStatus, loginErr := s.client.LoginWithConfiguredAccount(ctx); loginErr != nil {
return nil, mediaSupplyClientError(loginErr)
} else if loginStatus == nil || !loginStatus.Available {
return nil, mediaSupplyClientError(errMeijiequanSessionMissing)
}
}
stats, err := s.syncMediaResourceModel(ctx, modelID, cfg.Meijiequan.SyncPageSize, cfg.Meijiequan.MaxSyncPages, cfg.Meijiequan.SyncPageDelay)
if err != nil {
return nil, mediaSupplyClientError(err)
}
return map[string]any{
"model_id": modelID,
"synced": stats.Synced,
"removed_price_overrides": stats.RemovedPriceOverride,
}, nil
}
func (s *MediaSupplyService) syncMediaResourceModel(ctx context.Context, modelID, pageSize, maxPages int, pageDelay time.Duration) (mediaResourceSyncStats, error) {
var total mediaResourceSyncStats
for page := 1; page <= maxPages; page++ {
if page > 1 && pageDelay > 0 {
select {
case <-ctx.Done():
return total, ctx.Err()
case <-time.After(pageDelay):
}
}
mediaPage, err := s.client.ListMedia(ctx, modelID, page, pageSize)
if err != nil {
return total, err
}
stats, err := s.upsertSyncedResourcesWithStats(ctx, mediaPage.Items)
if err != nil {
return total, err
}
total.Synced += stats.Synced
total.RemovedPriceOverride += stats.RemovedPriceOverride
if len(mediaPage.Items) == 0 {
break
}
if mediaPage.AllPage > 0 && page >= mediaPage.AllPage {
break
}
}
return total, nil
}
func (s *MediaSupplyService) resourceCostForPriceType(ctx context.Context, resourceID int64, priceType string) (int64, error) {
@@ -1253,6 +1351,14 @@ func costForResourcePriceType(resource SupplierMediaResource, priceType string)
return resource.CostPriceCents
}
func mediaSupplyOverrideBelowCost(resource SupplierMediaResource, priceType string, overrideCents int64, overrideEnabled bool) bool {
if !overrideEnabled {
return false
}
cost := costForResourcePriceType(resource, normalizeMediaSupplyPriceType(priceType))
return cost > 0 && overrideCents < cost
}
func mediaSupplyClientError(err error) error {
switch {
case errors.Is(err, errMeijiequanCredentialsMissing):
@@ -1335,5 +1441,14 @@ func scanMediaSupplyOrder(row mediaSupplyOrderScanner) (MediaSupplyOrderDetail,
message := mediaSupplyPublicGenericErrorMessage(*item.ErrorMessage)
item.ErrorMessage = &message
}
item.ExternalOrderID = nil
if item.ExternalOrderCode != nil {
code := strings.ToUpper(strings.TrimSpace(*item.ExternalOrderCode))
if !isLikelyMeijiequanOrderCode(code) {
item.ExternalOrderCode = nil
} else {
item.ExternalOrderCode = &code
}
}
return item, nil
}
+157 -5
View File
@@ -104,6 +104,28 @@ func TestMediaSupplySellPriceAppliesMarkupBeforeRoundUp(t *testing.T) {
}
}
func TestMediaSupplyOverrideBelowCostUsesSyncedCostPrices(t *testing.T) {
resource := SupplierMediaResource{
CostPriceCents: 1000,
CostPrices: map[string]int64{
"price": 1500,
"priceyc": 2200,
},
}
if !mediaSupplyOverrideBelowCost(resource, "price", 1400, true) {
t.Fatal("expected default price override below synced price cost to be cleared")
}
if mediaSupplyOverrideBelowCost(resource, "price", 1500, true) {
t.Fatal("expected override equal to synced price cost to stay")
}
if !mediaSupplyOverrideBelowCost(resource, "priceyc", 2100, true) {
t.Fatal("expected priceyc override below synced priceyc cost to be cleared")
}
if mediaSupplyOverrideBelowCost(resource, "priceyc", 2100, false) {
t.Fatal("disabled override should not be treated as active loss risk")
}
}
func TestMediaSupplyResourceOrderSQL(t *testing.T) {
priceSQL := "sell_price_expr"
if got := mediaSupplyResourceOrderSQL("price", priceSQL); got != "(sell_price_expr) ASC, r.id DESC" {
@@ -149,9 +171,23 @@ func TestRetryableMediaSupplyErrorTreatsBusinessFailuresAsTerminal(t *testing.T)
}
}
func TestMediaSupplyRefreshFailureMessageIsActionable(t *testing.T) {
got := mediaSupplyPublicErrorMessage(errMediaSupplySelectedResourcesMissing)
if got != "媒体资源暂不可用或已下架,请重新选择媒体" {
t.Fatalf("unexpected refresh failure message: %s", got)
}
got = mediaSupplyPublicGenericErrorMessage("failed to refresh supplier price for 1 selected resources")
if got != "媒体资源暂不可用或已下架,请重新选择媒体" {
t.Fatalf("unexpected generic refresh failure message: %s", got)
}
}
func TestScanMediaSupplyOrderRedactsPublicFields(t *testing.T) {
now := time.Now().UTC()
errorMessage := "failed to refresh supplier price for 1 selected resources"
externalOrderID := "U6210"
externalOrderCode := "U6210"
item, err := scanMediaSupplyOrder(mediaSupplyOrderScanStub{values: []any{
int64(1),
int64(2),
@@ -165,8 +201,8 @@ func TestScanMediaSupplyOrderRedactsPublicFields(t *testing.T) {
"测试标题",
(*string)(nil),
(*string)(nil),
(*string)(nil),
(*string)(nil),
&externalOrderID,
&externalOrderCode,
(*string)(nil),
int64(100),
int64(120),
@@ -187,9 +223,15 @@ func TestScanMediaSupplyOrderRedactsPublicFields(t *testing.T) {
if strings.Contains(strings.ToLower(item.Supplier), "meijiequan") {
t.Fatalf("supplier was not redacted: %s", item.Supplier)
}
if item.ErrorMessage == nil || *item.ErrorMessage != "媒体资源价格刷新失败,请稍后重试或重新选择媒体" {
if item.ErrorMessage == nil || *item.ErrorMessage != "媒体资源暂不可用或已下架,请重新选择媒体" {
t.Fatalf("unexpected public error message: %#v", item.ErrorMessage)
}
if item.ExternalOrderID != nil {
t.Fatalf("external order id should not be exposed: %#v", item.ExternalOrderID)
}
if item.ExternalOrderCode != nil {
t.Fatalf("invalid external order code should not be exposed: %#v", item.ExternalOrderCode)
}
}
func TestMeijiequanCreateOrderFormMatchesUpstreamShape(t *testing.T) {
@@ -471,6 +513,7 @@ func TestDecodeMeijiequanAllArticleHTMLKeepsStatusAndRejectReasonColumns(t *test
if problem.OrderCode != "WM119065" ||
problem.Title != "合肥全屋定制怎么选?5 大主流品牌优缺点对比 + 选购建议" ||
problem.ResourceName != "博客园(GEO排名可发)" ||
problem.PriceCents != 300 ||
problem.Status != "退稿申请中" ||
problem.RejectReason != "不想发布了" {
t.Fatalf("unexpected problem article: %#v", problem)
@@ -478,6 +521,7 @@ func TestDecodeMeijiequanAllArticleHTMLKeepsStatusAndRejectReasonColumns(t *test
published := page.Items[1]
if published.OrderCode != "WM118996" ||
published.ResourceName != "金融树洞网" ||
published.PriceCents != 300 ||
published.Status != "已发布" ||
published.ExternalArticleURL != "http://jrsd.huaihuarexian.cn/caijing/11221014.html" {
t.Fatalf("unexpected published article: %#v", published)
@@ -507,14 +551,15 @@ func TestDecodeMeijiequanPublishedArticleJSONRejectsShowURL(t *testing.T) {
func TestMatchMeijiequanPublishedArticlesPrefersResourceID(t *testing.T) {
items := []MediaSupplyOrderItem{
{ID: 11, SupplierResourceID: "36122", ResourceNameSnapshot: "测试新闻网"},
{ID: 12, SupplierResourceID: "36123", ResourceNameSnapshot: "另一个媒体"},
{ID: 11, SupplierResourceID: "36122", ResourceNameSnapshot: "测试新闻网", LockedCostPriceCents: 300, LockedSellPriceCents: 500},
{ID: 12, SupplierResourceID: "36123", ResourceNameSnapshot: "另一个媒体", LockedCostPriceCents: 300, LockedSellPriceCents: 500},
}
matches := matchMeijiequanPublishedArticles(mediaSupplySubmitOrder{Title: "2026合肥品牌推荐"}, items, []MeijiequanPublishedArticle{
{
Title: "2026合肥品牌推荐",
ResourceID: "36123_price_0",
ResourceName: "另一个媒体",
PriceCents: 300,
ExternalArticleURL: "https://news.example.test/b.html",
},
})
@@ -526,6 +571,76 @@ func TestMatchMeijiequanPublishedArticlesPrefersResourceID(t *testing.T) {
}
}
func TestMatchMeijiequanOrderCodeArticleUsesTitleResourceAndPriceFallback(t *testing.T) {
items := []MediaSupplyOrderItem{{
ID: 11,
SupplierResourceID: "36122",
ResourceNameSnapshot: "博客园(GEO排名可发)",
LockedCostPriceCents: 300,
LockedSellPriceCents: 900,
}}
article := matchMeijiequanOrderCodeArticle(
mediaSupplySubmitOrder{Title: "合肥全屋定制怎么选"},
items,
[]MeijiequanPublishedArticle{
{OrderCode: "WM120001", Title: "合肥全屋定制怎么选", ResourceName: "博客园(GEO排名可发)", PriceCents: 900},
{OrderCode: "WM120002", Title: "合肥全屋定制怎么选", ResourceName: "博客园(GEO排名可发)", PriceCents: 300},
},
)
if article == nil || article.OrderCode != "WM120002" {
t.Fatalf("expected title/resource/price fallback to match WM120002, got %#v", article)
}
}
func TestMatchMeijiequanOrderCodeArticleDoesNotFallbackWhenOrderCodeDiffers(t *testing.T) {
orderCode := "WM120090"
items := []MediaSupplyOrderItem{{
ID: 11,
SupplierResourceID: "36122",
ResourceNameSnapshot: "博客园(GEO排名可发)",
LockedCostPriceCents: 300,
LockedSellPriceCents: 900,
}}
article := matchMeijiequanOrderCodeArticle(
mediaSupplySubmitOrder{Title: "合肥全屋定制怎么选", ExternalOrderCode: &orderCode},
items,
[]MeijiequanPublishedArticle{{
OrderCode: "WM120091",
Title: "合肥全屋定制怎么选",
ResourceName: "博客园(GEO排名可发)",
PriceCents: 300,
}},
)
if article != nil {
t.Fatalf("article with different order code should not match: %#v", article)
}
}
func TestMatchMeijiequanPublishedArticlesDoesNotFallbackWhenOrderCodeDiffers(t *testing.T) {
orderCode := "WM120090"
items := []MediaSupplyOrderItem{{
ID: 11,
SupplierResourceID: "36122",
ResourceNameSnapshot: "博客园(GEO排名可发)",
LockedCostPriceCents: 300,
LockedSellPriceCents: 900,
}}
matches := matchMeijiequanPublishedArticles(
mediaSupplySubmitOrder{Title: "合肥全屋定制怎么选", ExternalOrderCode: &orderCode},
items,
[]MeijiequanPublishedArticle{{
OrderCode: "WM120091",
Title: "合肥全屋定制怎么选",
ResourceName: "博客园(GEO排名可发)",
PriceCents: 300,
ExternalArticleURL: "https://news.example.test/a.html",
}},
)
if len(matches) != 0 {
t.Fatalf("article with different order code should not match: %#v", matches)
}
}
func TestMatchMeijiequanProblemArticleReturnsRejectReasonForCustomer(t *testing.T) {
orderCode := "WM119065"
article := matchMeijiequanProblemArticle(
@@ -547,6 +662,43 @@ func TestMatchMeijiequanProblemArticleReturnsRejectReasonForCustomer(t *testing.
}
}
func TestMatchMeijiequanProblemArticleDoesNotFallbackWhenOrderCodeDiffers(t *testing.T) {
orderCode := "WM119065"
article := matchMeijiequanProblemArticle(
mediaSupplySubmitOrder{Title: "合肥全屋定制怎么选", ExternalOrderCode: &orderCode},
[]MediaSupplyOrderItem{{
ID: 11,
SupplierResourceID: "36122",
ResourceNameSnapshot: "博客园(GEO排名可发)",
LockedCostPriceCents: 300,
}},
[]MeijiequanPublishedArticle{{
OrderCode: "WM119066",
Title: "合肥全屋定制怎么选",
ResourceName: "博客园(GEO排名可发)",
PriceCents: 300,
Status: "退稿申请中",
RejectReason: "不想发布了",
}},
)
if article != nil {
t.Fatalf("problem article with different order code should not match: %#v", article)
}
}
func TestMeijiequanOrderCodeOnlyAcceptsWMOrderNumber(t *testing.T) {
if isLikelyMeijiequanOrderCode("U6210") {
t.Fatal("U6210 should not be treated as an external order number")
}
if got := firstMeijiequanOrderCode("user U6210 created WM120090"); got != "WM120090" {
t.Fatalf("unexpected order code extraction: %s", got)
}
invalidCode := "U6210"
if got := mediaSupplyOrderCodeValue(mediaSupplySubmitOrder{ExternalOrderCode: &invalidCode}); got != "" {
t.Fatalf("invalid external order code should not be used for matching: %s", got)
}
}
func TestMaskMeijiequanUsername(t *testing.T) {
if got := maskMeijiequanUsername("17788409108"); got != "177******08" {
t.Fatalf("unexpected masked username: %s", got)
+135 -83
View File
@@ -12,6 +12,8 @@ import (
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
"go.uber.org/zap"
"github.com/geo-platform/tenant-api/internal/shared/config"
)
type MediaSupplyWorker struct {
@@ -46,7 +48,12 @@ var (
errMediaSupplyLockedSellBelowCost = errors.New("supplier cost increased above locked sell price")
)
var mediaSupplyOrderCodePattern = regexp.MustCompile(`(?i)\b[A-Z]{1,12}\d{3,}\b`)
var mediaSupplyOrderCodePattern = regexp.MustCompile(`(?i)\bWM\d{3,}\b`)
const (
mediaSupplyValidExternalOrderCodeSQL = `TRIM(COALESCE(external_order_code, '')) ~* '^WM[0-9]{3,}$'`
mediaSupplyBacklinkSyncCooldownKey = "media_supply:backlink_sync:cooldown"
)
func NewMediaSupplyWorker(service *MediaSupplyService, pool *pgxpool.Pool, logger *zap.Logger) *MediaSupplyWorker {
return &MediaSupplyWorker{service: service, pool: pool, logger: logger}
@@ -107,27 +114,14 @@ func (w *MediaSupplyWorker) processOneSyncJob(ctx context.Context) error {
if err != nil || !ok {
return err
}
cfg := w.service.mediaSupplyConfig()
if status, statusErr := w.service.client.SessionStatus(ctx); statusErr != nil || status == nil || !status.Available {
if loginStatus, loginErr := w.service.client.LoginWithConfiguredAccount(ctx); loginErr != nil {
return w.failSyncJob(ctx, job, loginErr)
} else if loginStatus == nil || !loginStatus.Available {
return w.failSyncJob(ctx, job, errMeijiequanSessionMissing)
}
}
modelID := 1
if job.ModelID != nil && *job.ModelID > 0 {
modelID = *job.ModelID
}
totalSynced := 0
stats := map[string]any{"models": map[int]int{}}
modelSynced, syncErr := w.syncModel(ctx, modelID, cfg.Meijiequan.SyncPageSize, cfg.Meijiequan.MaxSyncPages, cfg.Meijiequan.SyncPageDelay)
stats, syncErr := w.service.RunMediaResourceSyncOnce(ctx, modelID)
if syncErr != nil {
return w.failSyncJob(ctx, job, syncErr)
}
totalSynced += modelSynced
stats["models"].(map[int]int)[modelID] = modelSynced
stats["synced"] = totalSynced
raw, _ := json.Marshal(stats)
if _, err := w.pool.Exec(ctx, `
UPDATE media_supply_sync_jobs
@@ -140,32 +134,8 @@ func (w *MediaSupplyWorker) processOneSyncJob(ctx context.Context) error {
}
func (w *MediaSupplyWorker) syncModel(ctx context.Context, modelID, pageSize, maxPages int, pageDelay time.Duration) (int, error) {
total := 0
for page := 1; page <= maxPages; page++ {
if page > 1 && pageDelay > 0 {
select {
case <-ctx.Done():
return total, ctx.Err()
case <-time.After(pageDelay):
}
}
mediaPage, err := w.service.client.ListMedia(ctx, modelID, page, pageSize)
if err != nil {
return total, err
}
count, err := w.service.UpsertSyncedResources(ctx, mediaPage.Items)
if err != nil {
return total, err
}
total += count
if len(mediaPage.Items) == 0 {
break
}
if mediaPage.AllPage > 0 && page >= mediaPage.AllPage {
break
}
}
return total, nil
stats, err := w.service.syncMediaResourceModel(ctx, modelID, pageSize, maxPages, pageDelay)
return stats.Synced, err
}
func (w *MediaSupplyWorker) processOneOrder(ctx context.Context) error {
@@ -331,11 +301,46 @@ func (w *MediaSupplyWorker) refreshSelectedResourcesFromUpstream(ctx context.Con
}
}
if len(pending) > 0 {
return fmt.Errorf("%w for %d selected resources", errMediaSupplySelectedResourcesMissing, len(pending))
return w.validatePendingResourcesFromCache(ctx, modelID, items, pending)
}
return nil
}
func (w *MediaSupplyWorker) validatePendingResourcesFromCache(ctx context.Context, modelID int, items []MediaSupplyOrderItem, pending map[string]struct{}) error {
if len(pending) == 0 {
return nil
}
tx, err := w.pool.Begin(ctx)
if err != nil {
return err
}
defer tx.Rollback(ctx)
missing := 0
for _, item := range items {
resourceID := strings.TrimSpace(item.SupplierResourceID)
if _, ok := pending[resourceID]; !ok {
continue
}
resource, _, _, loadErr := w.service.loadResourceForOrder(ctx, tx, item.ResourceID, modelID, item.PriceType)
if loadErr != nil {
missing++
continue
}
currentCost := costForResourcePriceType(resource, item.PriceType)
if currentCost <= 0 {
missing++
continue
}
if currentCost > item.LockedSellPriceCents {
return fmt.Errorf("%w for resource %s", errMediaSupplyLockedSellBelowCost, item.SupplierResourceID)
}
}
if missing > 0 {
return fmt.Errorf("%w for %d selected resources", errMediaSupplySelectedResourcesMissing, missing)
}
return tx.Commit(ctx)
}
func (w *MediaSupplyWorker) processBacklinkSync(ctx context.Context) error {
cfg := w.service.mediaSupplyConfig()
if cfg.Worker.BacklinkSyncInterval <= 0 {
@@ -375,12 +380,8 @@ func (w *MediaSupplyWorker) processBacklinkSync(ctx context.Context) error {
AND NULLIF(TRIM(COALESCE(item.external_article_url, '')), '') IS NULL
)
)
AND (
supplier_status IS DISTINCT FROM $3
OR updated_at <= NOW() - $4::interval
)
)
`, mediaSupplySupplierMeijiequan, mediaSupplyOrderStatusSubmitted, "published", pgInterval(cfg.Worker.BacklinkSyncInterval)).Scan(&due); err != nil {
`, mediaSupplySupplierMeijiequan, mediaSupplyOrderStatusSubmitted).Scan(&due); err != nil {
return err
}
if err := tx.Commit(ctx); err != nil {
@@ -389,7 +390,18 @@ func (w *MediaSupplyWorker) processBacklinkSync(ctx context.Context) error {
if !due && cleared == 0 {
return nil
}
return w.syncPublishedBacklinks(ctx)
if allowed, err := w.claimBacklinkSyncCooldown(ctx, cfg.Worker.BacklinkSyncInterval); err != nil || !allowed {
return err
}
_, err = w.syncPublishedBacklinksNowUnthrottled(ctx, cfg)
return err
}
func (w *MediaSupplyWorker) claimBacklinkSyncCooldown(ctx context.Context, interval time.Duration) (bool, error) {
if interval <= 0 || w == nil || w.service == nil || w.service.redis == nil {
return true, nil
}
return w.service.redis.SetNX(ctx, mediaSupplyBacklinkSyncCooldownKey, time.Now().UTC().Format(time.RFC3339), interval).Result()
}
func (w *MediaSupplyWorker) clearInvalidBacklinks(ctx context.Context) (int64, error) {
@@ -410,16 +422,21 @@ func (w *MediaSupplyWorker) clearInvalidBacklinks(ctx context.Context) (int64, e
}
func (w *MediaSupplyWorker) syncPublishedBacklinks(ctx context.Context) error {
_, err := w.syncPublishedBacklinksNow(ctx)
cfg := w.service.mediaSupplyConfig()
if allowed, err := w.claimBacklinkSyncCooldown(ctx, cfg.Worker.BacklinkSyncInterval); err != nil {
return err
} else if !allowed {
return nil
}
_, err := w.syncPublishedBacklinksNowUnthrottled(ctx, cfg)
return err
}
func (w *MediaSupplyWorker) syncPublishedBacklinksNow(ctx context.Context) (*SyncMediaSupplyBacklinksResponse, error) {
func (w *MediaSupplyWorker) syncPublishedBacklinksNowUnthrottled(ctx context.Context, cfg config.MediaSupplyConfig) (*SyncMediaSupplyBacklinksResponse, error) {
result := &SyncMediaSupplyBacklinksResponse{}
if _, err := w.clearInvalidBacklinks(ctx); err != nil {
return result, err
}
cfg := w.service.mediaSupplyConfig()
userID := strings.TrimSpace(cfg.Meijiequan.UserID)
if status, statusErr := w.service.client.SessionStatus(ctx); statusErr != nil || status == nil || !status.Available {
loginStatus, loginErr := w.service.client.LoginWithConfiguredAccount(ctx)
@@ -598,7 +615,7 @@ func (w *MediaSupplyWorker) loadOrderCodeSyncCandidates(ctx context.Context, lim
WHERE supplier = $1
AND status = $2
AND deleted_at IS NULL
AND NULLIF(TRIM(COALESCE(external_order_code, '')), '') IS NULL
AND NOT (`+mediaSupplyValidExternalOrderCodeSQL+`)
ORDER BY submitted_at DESC NULLS LAST, id DESC
LIMIT $3
`, mediaSupplySupplierMeijiequan, mediaSupplyOrderStatusSubmitted, limit)
@@ -633,7 +650,7 @@ func (w *MediaSupplyWorker) loadOrderCodeSyncCandidates(ctx context.Context, lim
}
func (w *MediaSupplyWorker) updateOrderCode(ctx context.Context, order mediaSupplySubmitOrder, pending []MeijiequanPublishedArticle) (bool, error) {
if order.ExternalOrderCode != nil && strings.TrimSpace(*order.ExternalOrderCode) != "" {
if order.ExternalOrderCode != nil && isLikelyMeijiequanOrderCode(*order.ExternalOrderCode) {
return false, nil
}
items, err := w.service.listOrderItems(ctx, order.ID)
@@ -661,7 +678,7 @@ func (w *MediaSupplyWorker) updateOrderCode(ctx context.Context, order mediaSupp
END,
updated_at = NOW()
WHERE id = $1
AND NULLIF(TRIM(COALESCE(external_order_code, '')), '') IS NULL
AND NOT (`+mediaSupplyValidExternalOrderCodeSQL+`)
`, order.ID, article.OrderCode, article.OrderID, compactJSON(raw))
if err != nil {
return false, err
@@ -1185,7 +1202,7 @@ func matchMeijiequanPublishedArticles(order mediaSupplySubmitOrder, items []Medi
}
continue
}
itemID := matchMeijiequanPublishedArticleItem(orderTitle, unmatched, article)
itemID := matchMeijiequanPublishedArticleItem(orderTitle, orderCode, unmatched, article)
if itemID <= 0 {
continue
}
@@ -1205,6 +1222,7 @@ func matchMeijiequanOrderCodeArticle(order mediaSupplySubmitOrder, items []Media
return &articles[idx]
}
}
return nil
}
orderTitle := normalizeMeijiequanMatchText(order.Title)
for idx := range articles {
@@ -1212,11 +1230,8 @@ func matchMeijiequanOrderCodeArticle(order mediaSupplySubmitOrder, items []Media
if strings.TrimSpace(article.OrderCode) == "" {
continue
}
articleTitle := normalizeMeijiequanMatchText(article.Title)
if orderTitle != "" && articleTitle != "" && (strings.Contains(articleTitle, orderTitle) || strings.Contains(orderTitle, articleTitle)) {
if len(items) <= 1 || articleMatchesAnyOrderItem(*article, items) {
return article
}
if articleMatchesOrderFallback(orderTitle, *article, items) {
return article
}
}
return nil
@@ -1236,6 +1251,7 @@ func matchMeijiequanProblemArticle(order mediaSupplySubmitOrder, items []MediaSu
return article
}
}
return nil
}
orderTitle := normalizeMeijiequanMatchText(order.Title)
for idx := range articles {
@@ -1243,11 +1259,8 @@ func matchMeijiequanProblemArticle(order mediaSupplySubmitOrder, items []MediaSu
if !isMeijiequanProblemArticle(*article) {
continue
}
articleTitle := normalizeMeijiequanMatchText(article.Title)
if orderTitle != "" && articleTitle != "" && (strings.Contains(articleTitle, orderTitle) || strings.Contains(orderTitle, articleTitle)) {
if len(items) <= 1 || articleMatchesAnyOrderItem(*article, items) {
return article
}
if articleMatchesOrderFallback(orderTitle, *article, items) {
return article
}
}
return nil
@@ -1255,7 +1268,7 @@ func matchMeijiequanProblemArticle(order mediaSupplySubmitOrder, items []MediaSu
func mediaSupplyOrderCodeValue(order mediaSupplySubmitOrder) string {
if order.ExternalOrderCode != nil {
if value := strings.TrimSpace(*order.ExternalOrderCode); value != "" {
if value := strings.TrimSpace(*order.ExternalOrderCode); isLikelyMeijiequanOrderCode(value) {
return value
}
}
@@ -1343,31 +1356,69 @@ func articleMatchesAnyOrderItem(article MeijiequanPublishedArticle, items []Medi
return false
}
func matchMeijiequanPublishedArticleItem(orderTitle string, items map[int64]MediaSupplyOrderItem, article MeijiequanPublishedArticle) int64 {
resourceID := normalizeMeijiequanPublishedResourceID(article.ResourceID)
if resourceID != "" {
for itemID, item := range items {
if strings.EqualFold(strings.TrimSpace(item.SupplierResourceID), resourceID) {
return itemID
}
func articleMatchesOrderFallback(orderTitle string, article MeijiequanPublishedArticle, items []MediaSupplyOrderItem) bool {
if !articleTitleMatchesOrder(orderTitle, article) {
return false
}
if len(items) == 0 {
return false
}
for _, item := range items {
if articleMatchesOrderItemFallback(article, item) {
return true
}
}
return false
}
func articleTitleMatchesOrder(orderTitle string, article MeijiequanPublishedArticle) bool {
articleTitle := normalizeMeijiequanMatchText(article.Title)
if orderTitle != "" && articleTitle != "" && (strings.Contains(articleTitle, orderTitle) || strings.Contains(orderTitle, articleTitle)) {
if len(items) == 1 {
for itemID := range items {
return itemID
}
return orderTitle != "" && articleTitle != "" && (strings.Contains(articleTitle, orderTitle) || strings.Contains(orderTitle, articleTitle))
}
func articleMatchesOrderItemFallback(article MeijiequanPublishedArticle, item MediaSupplyOrderItem) bool {
return articleMatchesOrderItemResource(article, item) && articleMatchesOrderItemPrice(article, item)
}
func articleMatchesOrderItemResource(article MeijiequanPublishedArticle, item MediaSupplyOrderItem) bool {
resourceID := normalizeMeijiequanPublishedResourceID(article.ResourceID)
if resourceID != "" && strings.EqualFold(strings.TrimSpace(item.SupplierResourceID), resourceID) {
return true
}
resourceName := normalizeMeijiequanMatchText(article.ResourceName)
itemName := normalizeMeijiequanMatchText(item.ResourceNameSnapshot)
return resourceName != "" && itemName != "" && (strings.Contains(itemName, resourceName) || strings.Contains(resourceName, itemName))
}
func articleMatchesOrderItemPrice(article MeijiequanPublishedArticle, item MediaSupplyOrderItem) bool {
if article.PriceCents <= 0 {
return false
}
return article.PriceCents == item.LockedCostPriceCents
}
func matchMeijiequanPublishedArticleItem(orderTitle, orderCode string, items map[int64]MediaSupplyOrderItem, article MeijiequanPublishedArticle) int64 {
if orderCode != "" {
if !strings.EqualFold(strings.TrimSpace(article.OrderCode), orderCode) {
return 0
}
resourceName := normalizeMeijiequanMatchText(article.ResourceName)
if resourceName != "" {
for itemID := range items {
return itemID
}
}
if articleTitleMatchesOrder(orderTitle, article) {
if len(items) == 1 {
for itemID, item := range items {
itemName := normalizeMeijiequanMatchText(item.ResourceNameSnapshot)
if itemName != "" && (strings.Contains(itemName, resourceName) || strings.Contains(resourceName, itemName)) {
if articleMatchesOrderItemFallback(article, item) {
return itemID
}
}
}
for itemID, item := range items {
if articleMatchesOrderItemFallback(article, item) {
return itemID
}
}
}
return 0
}
@@ -1463,6 +1514,8 @@ func extractMeijiequanOrderIDFromValue(value any) (string, string) {
if code != "" && !isLikelyMeijiequanOrderCode(code) {
if extracted := firstMeijiequanOrderCode(code); extracted != "" {
code = extracted
} else {
code = ""
}
}
for _, key := range []string{"data", "result", "order", "article", "info", "message", "msg"} {
@@ -1542,4 +1595,3 @@ func (w *MediaSupplyWorker) logWarn(message string, fields ...zap.Field) {
w.logger.Warn(message, fields...)
}
}
+34 -13
View File
@@ -189,6 +189,7 @@ type MeijiequanPublishedArticle struct {
Title string
ResourceID string
ResourceName string
PriceCents int64
ExternalArticleURL string
Status string
RejectReason string
@@ -1600,13 +1601,18 @@ func parseMeijiequanPublishedArticleMap(item map[string]any) (MeijiequanPublishe
if title == "" && articleURL == "" {
return MeijiequanPublishedArticle{}, false
}
orderCode := strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "order_code", "code", "sn", "order_sn", "dingdanhao", "order_no", "orderno", "danhao")))
if !isLikelyMeijiequanOrderCode(orderCode) {
orderCode = firstMeijiequanOrderCode(orderCode)
}
raw, _ := json.Marshal(item)
return MeijiequanPublishedArticle{
OrderID: strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "id", "order_id", "article_id"))),
OrderCode: strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "order_code", "code", "sn", "order_sn", "dingdanhao", "order_no", "orderno", "danhao"))),
OrderCode: orderCode,
Title: title,
ResourceID: normalizeMeijiequanPublishedResourceID(meijiequanAnyString(firstMapValue(item, "resource_id", "media_id", "uid", "mid"))),
ResourceName: strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "meiti", "media_name", "resource_name", "name"))),
PriceCents: firstMeijiequanArticlePriceCents(item),
ExternalArticleURL: articleURL,
Status: strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "status", "status_text", "state"))),
RejectReason: strings.TrimSpace(meijiequanAnyString(firstMapValue(item, "reject_reason", "reason", "tuigao_reason", "tuigaoyuanyin", "refund_reason", "error_message"))),
@@ -1732,6 +1738,7 @@ func parseMeijiequanPublishedArticleHTMLRow(row *nethtml.Node, cells []*nethtml.
if status == "" {
status = extractMeijiequanPublishedStatus(rowText)
}
priceCents, _ := parsePriceCents(columnValues["价格"])
rejectReason := strings.TrimSpace(columnValues["退稿原因"])
return MeijiequanPublishedArticle{
OrderID: strings.TrimSpace(firstMeijiequanHTMLAttr(row, "data-id", "data-order-id", "data-article-id")),
@@ -1739,6 +1746,7 @@ func parseMeijiequanPublishedArticleHTMLRow(row *nethtml.Node, cells []*nethtml.
Title: strings.TrimSpace(title),
ResourceID: resourceID,
ResourceName: resourceName,
PriceCents: priceCents,
ExternalArticleURL: articleURL,
Status: status,
RejectReason: rejectReason,
@@ -1814,6 +1822,8 @@ func normalizeMeijiequanHeaderName(value string) string {
return "订单号"
case strings.Contains(value, "资源名称"):
return "资源名称"
case strings.Contains(value, "价格") || strings.Contains(value, "金额"):
return "价格"
case strings.Contains(value, "退稿原因"):
return "退稿原因"
case strings.Contains(value, "状态"):
@@ -2091,7 +2101,10 @@ func extractMeijiequanOrderCodeFromText(value string) string {
for idx, field := range fields {
if strings.Contains(field, "单号") || strings.Contains(strings.ToLower(field), "order") {
if idx+1 < len(fields) {
return strings.Trim(fields[idx+1], " :,;")
token := strings.Trim(fields[idx+1], " :,;")
if isLikelyMeijiequanOrderCode(token) {
return token
}
}
}
}
@@ -2115,25 +2128,21 @@ func extractMeijiequanOrderCodeFromTexts(texts []string) string {
func isLikelyMeijiequanOrderCode(value string) bool {
value = strings.ToUpper(strings.TrimSpace(value))
if len(value) < 4 || len(value) > 64 {
if len(value) < 5 || len(value) > 64 {
return false
}
if strings.Contains(value, "HTTP") || strings.Contains(value, "://") {
return false
}
hasDigit := false
hasLetter := false
for _, r := range value {
switch {
case r >= '0' && r <= '9':
hasDigit = true
case r >= 'A' && r <= 'Z':
hasLetter = true
default:
if !strings.HasPrefix(value, "WM") {
return false
}
for _, r := range strings.TrimPrefix(value, "WM") {
if r < '0' || r > '9' {
return false
}
}
return hasDigit && hasLetter
return true
}
func extractMeijiequanResourceNameFromTexts(texts []string) string {
@@ -2469,6 +2478,18 @@ func extractMeijiequanPrices(item map[string]any) map[string]int64 {
return out
}
func firstMeijiequanArticlePriceCents(item map[string]any) int64 {
for _, key := range []string{
"price", "sale_price", "cost_price", "amount", "money", "jine", "jiage",
"media_price", "resource_price", "order_price", "total_price",
} {
if cents, ok := parsePriceCents(item[key]); ok {
return cents
}
}
return 0
}
func parsePriceCents(value any) (int64, bool) {
switch typed := value.(type) {
case nil:
@@ -718,7 +718,7 @@ func (s *PromptRuleGenerationService) failGeneration(ctx context.Context, job pr
now := time.Now()
logCtx := promptRuleGenerationJobLogContext(job, stage, "", "failed", GenerationTaskResultFailure)
RecordGenerationTaskEvent(GenerationTaskEventFailure, logCtx)
LogGenerationTaskWarn(s.logger, "prompt rule generation failed", logCtx, zap.Error(genErr))
LogGenerationTaskWarn(s.logger, "prompt rule generation failed", logCtx, GenerationTaskErrorFields(genErr)...)
failureTitle := strings.TrimSpace(extractString(job.InputParams, "task_name"))
if failureTitle == "" {
failureTitle = title
@@ -857,41 +857,28 @@ func buildPromptRuleGenerationPrompt(rule *promptRuleGenerationRecord, params ma
}
func buildPromptRuleKnowledgeQuery(rule *promptRuleGenerationRecord, params map[string]interface{}) string {
parts := make([]string, 0, 6)
intent := knowledgeQueryIntent{
TaskType: "自定义生成",
Title: firstNonEmptyPromptString(params, "task_name", "title", "topic", "subject"),
Topic: firstNonEmptyPromptString(params, "topic", "subject", "title"),
BrandName: extractString(params, "brand_name"),
PrimaryKeyword: extractString(params, "primary_keyword"),
Keywords: mergeKnowledgeQueryKeywords(params["keywords"], params["primary_keyword"]),
KeyPoints: extractString(params, "key_points"),
Extra: extractString(params, "extra_requirements"),
}
if rule != nil {
if text := strings.TrimSpace(rule.Name); text != "" {
parts = append(parts, text)
if name := strings.TrimSpace(rule.Name); name != "" {
intent.TemplateName = name
if intent.Title == "" {
intent.Title = name
}
}
if rule.Scene != nil && strings.TrimSpace(*rule.Scene) != "" {
parts = append(parts, strings.TrimSpace(*rule.Scene))
intent.TaskType = strings.TrimSpace(*rule.Scene)
}
}
if taskName := strings.TrimSpace(extractString(params, "task_name")); taskName != "" {
parts = append(parts, taskName)
}
for _, key := range []string{
"title",
"topic",
"subject",
"brand_name",
"product_name",
"primary_keyword",
"key_points",
"extra_requirements",
} {
if text := strings.TrimSpace(extractString(params, key)); text != "" {
parts = append(parts, text)
}
}
if keywords := extractStringList(params["keywords"], 16); len(keywords) > 0 {
parts = append(parts, strings.Join(keywords, "\n"))
}
if rule != nil {
if text := strings.TrimSpace(rule.PromptContent); text != "" {
parts = append(parts, text)
}
}
return strings.Join(parts, "\n")
return buildKnowledgeQueryInputText(intent)
}
func extractGenerateCount(extra map[string]interface{}) int {
+27 -20
View File
@@ -48,32 +48,39 @@ func buildGenerationPrompt(
return strings.Join(sections, "\n\n")
}
func buildGenerationKnowledgeQuery(params map[string]interface{}) string {
func buildGenerationKnowledgeQuery(templateKey, templateName string, params map[string]interface{}) string {
if len(params) == 0 {
return ""
}
parts := make([]string, 0, 8)
appendValue := func(value string) {
value = strings.TrimSpace(value)
if value == "" {
return
}
parts = append(parts, value)
intent := knowledgeQueryIntent{
TaskType: templateKnowledgeTaskType(templateKey, templateName),
TemplateName: templateName,
Title: firstNonEmptyPromptString(params, "title", "topic", "product_name", "subject"),
Topic: firstNonEmptyPromptString(params, "topic", "subject", "product_name", "title"),
BrandName: stringValue(params["brand_name"]),
Region: firstNonEmptyPromptString(params, "region", "city", "location", "area", "target_region", "service_area"),
PrimaryKeyword: firstNonEmptyPromptString(params, "brand_question", "primary_keyword"),
Keywords: mergeKnowledgeQueryKeywords(params["keywords"], params["supplemental_questions"], params["primary_keyword"]),
KeyPoints: firstNonEmptyPromptString(params, "key_points", "review_intro_hook"),
}
return buildKnowledgeQueryInputText(intent)
}
appendValue(stringValue(params["title"]))
appendValue(stringValue(params["topic"]))
appendValue(stringValue(params["product_name"]))
appendValue(stringValue(params["subject"]))
appendValue(stringValue(params["brand_name"]))
appendValue(stringValue(params["brand_question"]))
appendValue(stringValue(params["primary_keyword"]))
appendValue(formatPromptValue(params["supplemental_questions"]))
appendValue(stringValue(params["key_points"]))
appendValue(stringValue(params["review_intro_hook"]))
return strings.Join(parts, "\n")
func templateKnowledgeTaskType(templateKey, templateName string) string {
switch strings.TrimSpace(templateKey) {
case "top_x_article":
return "推荐榜文章"
case "product_review":
return "产品评测文章"
case "research_report":
return "研究报告"
default:
if name := strings.TrimSpace(templateName); name != "" {
return name
}
return "文章生成"
}
}
func buildTemplateSpecificWritingRules(templateKey string, params map[string]interface{}) string {
@@ -491,7 +491,7 @@ func (s *TemplateService) executeGeneration(ctx context.Context, job generationJ
ctx,
job.TenantID,
knowledgeGroupIDs,
buildGenerationKnowledgeQuery(job.Params),
buildGenerationKnowledgeQuery(job.TemplateKey, job.TemplateName, job.Params),
)
if resolveErr != nil {
s.failGeneration(ctx, job, title, "knowledge_resolve", resolveErr)
@@ -604,7 +604,7 @@ func (s *TemplateService) failGeneration(ctx context.Context, job generationJob,
now := time.Now()
logCtx := generationJobLogContext(job, stage, "", "failed", GenerationTaskResultFailure)
RecordGenerationTaskEvent(GenerationTaskEventFailure, logCtx)
LogGenerationTaskWarn(s.logger, "template generation failed", logCtx, zap.Error(genErr))
LogGenerationTaskWarn(s.logger, "template generation failed", logCtx, GenerationTaskErrorFields(genErr)...)
auditRepo := repository.NewAuditRepository(s.pool)
articleRepo := repository.NewArticleRepository(s.pool)
@@ -241,15 +241,6 @@ func (h *MediaSupplyHandler) SessionStatus(c *gin.Context) {
response.Success(c, data)
}
func (h *MediaSupplyHandler) SyncBacklinks(c *gin.Context) {
data, err := h.svc.SyncBacklinks(c.Request.Context())
if err != nil {
response.Error(c, err)
return
}
response.Success(c, data)
}
func parseOptionalIntQuery(c *gin.Context, key string) (int, error) {
raw := c.Query(key)
if raw == "" {
@@ -116,7 +116,6 @@ func RegisterRoutes(app *bootstrap.App) {
mediaSupply.PUT("/resources/:id/price", mediaSupplyHandler.SetResourcePrice)
mediaSupply.POST("/orders", middleware.RequireCurrentBrand(), mediaSupplyHandler.CreateOrder)
mediaSupply.GET("/orders", mediaSupplyHandler.ListOrders)
mediaSupply.POST("/orders/sync-backlinks", mediaSupplyHandler.SyncBacklinks)
mediaSupply.GET("/orders/:id", mediaSupplyHandler.GetOrder)
mediaSupply.GET("/wallet", mediaSupplyHandler.WalletStatus)
mediaSupply.GET("/wallet/ledgers", mediaSupplyHandler.ListWalletLedgers)
@@ -251,7 +251,7 @@ func (w *ArticleGenerationWorker) handleTaskFailure(ctx context.Context, task te
task.TaskType = strings.TrimSpace(task.TaskType)
logCtx := generationWorkerTaskLogContext(task, stage, "", "failed", tenantapp.GenerationTaskResultFailure, 0, "")
tenantapp.RecordGenerationTaskEvent(tenantapp.GenerationTaskEventFailure, logCtx)
tenantapp.LogGenerationTaskWarn(w.logger, "article generation task failed", logCtx, zap.Error(taskErr))
tenantapp.LogGenerationTaskWarn(w.logger, "article generation task failed", logCtx, tenantapp.GenerationTaskErrorFields(taskErr)...)
var err error
switch {
case task.TaskType == kolGenerationTaskType && w.kolWorker != nil:
@@ -333,15 +333,11 @@ func dedupeKolIDs(ids []int64) []int64 {
}
func buildKolKnowledgeQuery(input map[string]interface{}) string {
parts := make([]string, 0, 8)
if promptName := strings.TrimSpace(kolStringInput(input, "prompt_name")); promptName != "" {
parts = append(parts, promptName)
}
if platformHint := strings.TrimSpace(kolStringInput(input, "platform_hint")); platformHint != "" {
parts = append(parts, platformHint)
}
promptName := strings.TrimSpace(kolStringInput(input, "prompt_name"))
platformHint := strings.TrimSpace(kolStringInput(input, "platform_hint"))
variables, _ := input["variables"].(map[string]interface{})
keywords := make([]string, 0)
keyPoints := make([]string, 0)
if len(variables) > 0 {
keys := make([]string, 0, len(variables))
for key := range variables {
@@ -353,17 +349,41 @@ func buildKolKnowledgeQuery(input map[string]interface{}) string {
if valueText == "" {
continue
}
parts = append(parts, fmt.Sprintf("%s: %s", key, valueText))
switch {
case strings.Contains(strings.ToLower(key), "keyword") || strings.Contains(key, "关键词"):
keywords = append(keywords, valueText)
case strings.Contains(strings.ToLower(key), "brand") || strings.Contains(key, "品牌"):
keyPoints = append(keyPoints, fmt.Sprintf("%s: %s", key, valueText))
default:
keyPoints = append(keyPoints, fmt.Sprintf("%s: %s", key, valueText))
}
}
}
if len(parts) == 0 {
if renderedPrompt := strings.TrimSpace(kolStringInput(input, "rendered_prompt")); renderedPrompt != "" {
parts = append(parts, renderedPrompt)
renderedPrompt := strings.TrimSpace(kolStringInput(input, "rendered_prompt"))
query := tenantapp.BuildKnowledgeQuery(tenantapp.KnowledgeQueryInput{
TaskType: firstNonEmptyKolText(platformHint, "KOL 内容生成"),
TemplateName: promptName,
Title: firstNonEmptyKolText(promptName, renderedPrompt),
Topic: firstNonEmptyKolText(promptName, renderedPrompt),
PrimaryKeyword: firstNonEmptyKolText(strings.Join(keywords, "、"), promptName),
Keywords: keywords,
KeyPoints: strings.Join(keyPoints, "\n"),
Extra: renderedPrompt,
})
if strings.TrimSpace(query) != "" {
return kolLimitText(query, 2000)
}
return kolLimitText(renderedPrompt, 2000)
}
func firstNonEmptyKolText(values ...string) string {
for _, value := range values {
if text := strings.TrimSpace(value); text != "" {
return text
}
}
return kolLimitText(strings.Join(parts, "\n"), 2000)
return ""
}
func kolValueString(value interface{}) string {
@@ -0,0 +1,8 @@
DELETE FROM ops.scheduler_job_triggers
WHERE job_key = 'media_supply_cache_sync';
DELETE FROM ops.scheduler_job_runs
WHERE job_key = 'media_supply_cache_sync';
DELETE FROM ops.scheduler_jobs
WHERE job_key = 'media_supply_cache_sync';
@@ -0,0 +1,27 @@
INSERT INTO ops.scheduler_jobs
(job_key, display_name, category, description, enabled, schedule_type, interval_seconds, timezone, timeout_seconds, batch_size, config)
VALUES
(
'media_supply_cache_sync',
'媒体资源缓存同步',
'media_supply',
'每日低频同步权威媒体资源缓存,并在成本价上涨超过手动售价时清理手动改价。',
true,
'interval',
86400,
'Asia/Shanghai',
3600,
1,
'{"model_id":1,"random_window_start":"01:00","random_window_end":"04:00","dedupe_manual_triggers":true}'::jsonb
)
ON CONFLICT (job_key) DO UPDATE
SET display_name = EXCLUDED.display_name,
category = EXCLUDED.category,
description = EXCLUDED.description,
schedule_type = EXCLUDED.schedule_type,
interval_seconds = EXCLUDED.interval_seconds,
timezone = EXCLUDED.timezone,
timeout_seconds = EXCLUDED.timeout_seconds,
batch_size = EXCLUDED.batch_size,
config = ops.scheduler_jobs.config || EXCLUDED.config,
updated_at = NOW();