datatools/clueSync/main.go

package main

import (
	"flag"
	"fmt"
	"log"

	"app.yhyue.com/moapp/jybase/common"
	elastic "app.yhyue.com/moapp/jybase/esv1"
	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"app.yhyue.com/moapp/jybase/redis"
	"github.com/robfig/cron"
)

var (
	cfg    = new(Config)
	Mysql  *mysql.Mysql
	TiDb   *mysql.Mysql
	Base   *mysql.Mysql
	Es     *elastic.Elastic
	Mgo    *mongodb.MongodbSim
	MgoLog *mongodb.MongodbSim
	// configFile = flag.String("c", "./config.yaml", "配置文件")
	mode = flag.Int("m", 1, "")
)

func main() {
	flag.Parse()
	common.ReadConfig(&cfg)
	// _ = config.LoadConfigWithYaml(*configFile, cfg)
	log.Println("cfg ", cfg)
	TiDb = &mysql.Mysql{
		Address:      cfg.TiDb.Host + ":" + fmt.Sprint(cfg.TiDb.Port),
		UserName:     cfg.TiDb.User,
		PassWord:     cfg.TiDb.Password,
		DBName:       cfg.TiDb.Database,
		MaxOpenConns: cfg.TiDb.PollSize,
		MaxIdleConns: cfg.TiDb.MaxIdle,
	}
	TiDb.Init()
	Mysql = &mysql.Mysql{
		Address:      cfg.Mysql.Host + ":" + fmt.Sprint(cfg.Mysql.Port),
		UserName:     cfg.Mysql.User,
		PassWord:     cfg.Mysql.Password,
		DBName:       cfg.Mysql.Database,
		MaxOpenConns: cfg.Mysql.PollSize,
		MaxIdleConns: cfg.Mysql.MaxIdle,
	}
	Mysql.Init()
	Base = &mysql.Mysql{
		Address:      cfg.BaseService.Host + ":" + fmt.Sprint(cfg.BaseService.Port),
		UserName:     cfg.BaseService.User,
		PassWord:     cfg.BaseService.Password,
		DBName:       cfg.BaseService.Database,
		MaxOpenConns: cfg.BaseService.PollSize,
		MaxIdleConns: cfg.BaseService.MaxIdle,
	}
	Base.Init()
	Es = &elastic.Elastic{
		S_esurl: cfg.Es.Address,
		I_size:  cfg.Es.DbSize,
	}
	Es.InitElasticSize()
	redis.InitRedis(cfg.RedisServer)
	Mgo = mongodb.NewMgo(cfg.Mgo.Address, cfg.Mgo.DbName, cfg.Mgo.DbSize)
	MgoLog = mongodb.NewMgoWithUser(cfg.MgoLog.Address, cfg.MgoLog.DbName, cfg.MgoLog.User, cfg.MgoLog.Password, cfg.MgoLog.DbSize)
	InitArea()
	if *mode == 1 {
		// 未支付订单 30分钟一次
		orders()
		a := cron.New()
		a.AddFunc(cfg.CornExp1, func() {
			orders()
		})
		a.Start()
		// 新注册用户 5分钟一次
		// users()
		b := cron.New()
		b.AddFunc(cfg.CornExp2, func() {
			users()
			saleLeads()
		})
		b.Start()
		// 留资 5分钟一次
		// c := cron.New()
		// c.AddFunc(cfg.CornExp3, func() {
		// 	saleLeads()
		// })
		// c.Start()
		// 用户标签一天一次
		// tagAllSync()
		d := cron.New()
		d.AddFunc(cfg.CornExp4, func() {
			tagAllSync()
		})
		d.Start()
		// 用户标签增量 5分钟一次
		tagAddSync()
		e := cron.New()
		e.AddFunc(cfg.CornExp5, func() {
			tagAddSync()
		})
		e.Start()
		//订阅增量 5分钟一次
		subscribeAddSync()
		f := cron.New()
		f.AddFunc(cfg.CornExp6, func() {
			subscribeAddSync()
		})
		f.Start()
		//自动进入任务车 1天一次
		// autoTask()
		// autoTasks()
		g := cron.New()
		g.AddFunc(cfg.CornExp7, func() {
			autoTask()
			autoTasks()
		})
		g.Start()
		select {}
	} else if *mode == 2 {
		users()
	} else if *mode == 3 {
		saleLeads()
	} else if *mode == 4 {
		saleLeads()
		users()
	} else if *mode == 5 {
		tagAllSync()
	} else if *mode == 6 {
		tagAddSync()
		e := cron.New()
		e.AddFunc(cfg.CornExp5, func() {
			tagAddSync()
		})
		e.Start()
		select {}
	} else if *mode == 7 {
		subscribeAddSync()
		e := cron.New()
		e.AddFunc(cfg.CornExp6, func() {
			subscribeAddSync()
		})
		e.Start()
		select {}
	} else if *mode == 8 {
		subscribeAllSync()
	} else if *mode == 9 {
		autoTask()
	} else if *mode == 10 {
		getAreaCode("6447753fb4f3c077da9d5336")
	} else if *mode == 11 {
		trail()
	}
}