958245537a
Add plan catalog and subscription schema support, seed baselines, and real importers for core domestic subscriptions plus stable official pricing sources. This commit also hardens the shared fetch layers so the importers can support live collection and database writes instead of relying on manual placeholders alone.
111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
//go:build llm_script
|
||
|
||
package main
|
||
|
||
import (
|
||
"database/sql"
|
||
"flag"
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
|
||
_ "github.com/lib/pq"
|
||
)
|
||
|
||
const catalogSeedVerificationImporterKey = "import_catalog_seed_verification.go"
|
||
|
||
type catalogSeedVerificationConfig struct {
|
||
DryRun bool
|
||
}
|
||
|
||
func main() {
|
||
loadCatalogSeedVerificationEnv()
|
||
|
||
var cfg catalogSeedVerificationConfig
|
||
flag.BoolVar(&cfg.DryRun, "dry-run", false, "仅打印摘要，不写入数据库")
|
||
flag.Parse()
|
||
|
||
db, err := catalogSeedVerificationDB()
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "open db: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
defer db.Close()
|
||
|
||
if err := runCatalogSeedVerificationImport(db, cfg); err != nil {
|
||
fmt.Fprintf(os.Stderr, "import_catalog_seed_verification: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
func loadCatalogSeedVerificationEnv() {
|
||
for _, path := range []string{".env.local", ".env"} {
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
for _, line := range strings.Split(string(data), "\n") {
|
||
line = strings.TrimSpace(line)
|
||
if line == "" || strings.HasPrefix(line, "#") {
|
||
continue
|
||
}
|
||
key, value, ok := strings.Cut(line, "=")
|
||
if !ok {
|
||
continue
|
||
}
|
||
key = strings.TrimSpace(key)
|
||
value = strings.Trim(strings.TrimSpace(value), `"'`)
|
||
if key == "" {
|
||
continue
|
||
}
|
||
if _, exists := os.LookupEnv(key); exists {
|
||
continue
|
||
}
|
||
_ = os.Setenv(key, value)
|
||
}
|
||
}
|
||
}
|
||
|
||
func catalogSeedVerificationDB() (*sql.DB, error) {
|
||
dsn := os.Getenv("DATABASE_URL")
|
||
if dsn == "" {
|
||
dsn = "postgres://long@/llm_intelligence?host=/var/run/postgresql"
|
||
}
|
||
return sql.Open("postgres", dsn)
|
||
}
|
||
|
||
func runCatalogSeedVerificationImport(db *sql.DB, cfg catalogSeedVerificationConfig) error {
|
||
var count int
|
||
if err := db.QueryRow(`
|
||
SELECT COUNT(*)
|
||
FROM plan_catalog_inventory
|
||
WHERE importer_key = $1
|
||
`, catalogSeedVerificationImporterKey).Scan(&count); err != nil {
|
||
return err
|
||
}
|
||
|
||
if cfg.DryRun {
|
||
fmt.Printf("source=catalog-seed-verification entries=%d dry_run=true\n", count)
|
||
return nil
|
||
}
|
||
|
||
_, err := db.Exec(`
|
||
UPDATE plan_catalog_inventory
|
||
SET plan_status = 'confirmed',
|
||
notes = CASE
|
||
WHEN position($2 in COALESCE(notes, '')) > 0 THEN notes
|
||
WHEN COALESCE(notes, '') = '' THEN $2
|
||
ELSE notes || '；' || $2
|
||
END,
|
||
last_checked_at = CURRENT_TIMESTAMP,
|
||
updated_at = CURRENT_TIMESTAMP
|
||
WHERE importer_key = $1
|
||
`, catalogSeedVerificationImporterKey, strings.TrimSpace("当前链路为目录级官方入口核验，结构化公开价格待后续独立 importer 补齐。"))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
fmt.Printf("source=catalog-seed-verification entries=%d dry_run=false\n", count)
|
||
return nil
|
||
}
|