|
@@ -13,40 +13,41 @@ import (
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
- wait = &sync.WaitGroup{}
|
|
|
- pool = make(chan bool, Config.SyncPool)
|
|
|
+ wait = &sync.WaitGroup{}
|
|
|
+ pool = make(chan bool, Config.SyncPool)
|
|
|
+ allMap = map[string]Entity{
|
|
|
+ "raw_user": Raw_user,
|
|
|
+ "raw_product": Raw_product,
|
|
|
+ "raw_page": Raw_page,
|
|
|
+ "raw_element": Raw_element,
|
|
|
+ "raw_event": Raw_event,
|
|
|
+ "action_login": Action_login,
|
|
|
+ "action_page": Action_page,
|
|
|
+ "action_event": Action_event,
|
|
|
+ "action_order": Action_order,
|
|
|
+ "action_order_spec": Action_order_spec,
|
|
|
+ "action_product_record": Action_product_record,
|
|
|
+ "user_prize": User_prize,
|
|
|
+ "integral_flow": Integral_flow,
|
|
|
+ "point_type": Point_type,
|
|
|
+ }
|
|
|
)
|
|
|
|
|
|
func main() {
|
|
|
m := flag.Int("m", 0, "模式 0:定时任务 1:非定时任务")
|
|
|
t := flag.String("t", "", "表名")
|
|
|
flag.Parse()
|
|
|
- all := []Entity{Raw_user, Raw_product, Raw_page, Raw_element, Raw_event, Action_login, Action_page, Action_event, Action_order, Action_order_spec, Action_product_record}
|
|
|
+ all := []Entity{}
|
|
|
+ for _, v := range allMap {
|
|
|
+ all = append(all, v)
|
|
|
+ }
|
|
|
if *m == 1 {
|
|
|
- if *t == "raw_user" {
|
|
|
- run(Raw_user)
|
|
|
- } else if *t == "raw_product" {
|
|
|
- run(Raw_product)
|
|
|
- } else if *t == "raw_page" {
|
|
|
- run(Raw_page)
|
|
|
- } else if *t == "raw_element" {
|
|
|
- run(Raw_element)
|
|
|
- } else if *t == "raw_event" {
|
|
|
- run(Raw_event)
|
|
|
- } else if *t == "action_login" {
|
|
|
- run(Action_login)
|
|
|
- } else if *t == "action_page" {
|
|
|
- run(Action_page)
|
|
|
- } else if *t == "action_event" {
|
|
|
- run(Action_event)
|
|
|
- } else if *t == "action_order" {
|
|
|
- run(Action_order)
|
|
|
- } else if *t == "action_order_spec" {
|
|
|
- run(Action_order_spec)
|
|
|
- } else if *t == "action_product_record" {
|
|
|
- run(Action_product_record)
|
|
|
- } else if *t == "" {
|
|
|
+ if *t == "" {
|
|
|
run(all...)
|
|
|
+ } else if allMap[*t] != nil {
|
|
|
+ run(allMap[*t])
|
|
|
+ } else {
|
|
|
+ log.Fatalln(*t, "表的同步数据任务不存在")
|
|
|
}
|
|
|
} else {
|
|
|
SimpleCrontab(false, Config.RunTime, func() {
|