35 lines
903 B
Go
35 lines
903 B
Go
package cmd
|
||
|
||
import (
|
||
"github.com/spf13/cobra"
|
||
|
||
"my-spiders/internal/spider"
|
||
)
|
||
|
||
var (
|
||
theImpressionMaxCo int
|
||
theImpressionMaxArticles int
|
||
)
|
||
|
||
var theImpressionCmd = &cobra.Command{
|
||
Use: "theimpression",
|
||
Short: "启动 The Impression 街拍采集器 (street-style)",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
if theImpressionMaxCo <= 0 {
|
||
theImpressionMaxCo = 5
|
||
}
|
||
// 复用全局 -d/--debug 标志
|
||
isDebug := debug
|
||
s := spider.NewTheImpressionSpider(theImpressionMaxCo, IngestClient)
|
||
s.MaxArticles = theImpressionMaxArticles
|
||
s.Debug = isDebug
|
||
s.Run()
|
||
},
|
||
}
|
||
|
||
func init() {
|
||
rootCmd.AddCommand(theImpressionCmd)
|
||
theImpressionCmd.Flags().IntVarP(&theImpressionMaxCo, "max-co", "m", 5, "最大并发数限制")
|
||
theImpressionCmd.Flags().IntVarP(&theImpressionMaxArticles, "max-articles", "n", 0, "最多抓取的街拍任务数（0=不限制，小批量试抓用）")
|
||
}
|