diff --git a/getter/tgchannel.go b/getter/tgchannel.go index 74ab98e..c03f046 100644 --- a/getter/tgchannel.go +++ b/getter/tgchannel.go @@ -22,14 +22,22 @@ type TGChannelGetter struct { func NewTGChannelGetter(options tool.Options) Getter { num, found := options["num"] - if !found || num.(int) <= 0 { - num = 200 + t := 200 + switch num.(type) { + case int: + t = num.(int) + case float64: + t = int(num.(float64)) + } + + if !found || t <= 0 { + t = 200 } url, found := options["channel"] if found { return &TGChannelGetter{ c: colly.NewCollector(), - NumNeeded: num.(int), + NumNeeded: t, Url: "https://t.me/s/" + url.(string), } } diff --git a/getter/web_fanqiangdang.go b/getter/web_fanqiangdang.go index 3554c45..588866d 100644 --- a/getter/web_fanqiangdang.go +++ b/getter/web_fanqiangdang.go @@ -22,14 +22,23 @@ type WebFanqiangdang struct { func NewWebFanqiangdangGetter(options tool.Options) Getter { num, found := options["num"] - if !found || num.(int) <= 0 { - num = 200 + + t := 200 + switch num.(type) { + case int: + t = num.(int) + case float64: + t = int(num.(float64)) + } + + if !found || t <= 0 { + t = 200 } url, found := options["url"] if found { return &WebFanqiangdang{ c: colly.NewCollector(), - NumNeeded: num.(int), + NumNeeded: t, Url: url.(string), } }