You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
214 lines
4.6 KiB
214 lines
4.6 KiB
package main
|
|
|
|
import (
|
|
"LAPP_ETL/common"
|
|
dal "LAPP_ETL/dao/etl"
|
|
"LAPP_ETL/global"
|
|
"LAPP_ETL/grmi"
|
|
"LAPP_ETL/infra/config"
|
|
"LAPP_ETL/infra/db"
|
|
"LAPP_ETL/infra/driver"
|
|
"LAPP_ETL/infra/logger"
|
|
meta "LAPP_ETL/meta/etl"
|
|
"LAPP_ETL/scheduler"
|
|
"LAPP_ETL/utils"
|
|
"LAPP_ETL/web/routes"
|
|
"fmt"
|
|
"github.com/kardianos/service"
|
|
"github.com/kataras/iris/v12"
|
|
"log"
|
|
_ "net/http/pprof"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
)
|
|
|
|
//基于windows系统服务器，安装成服务
|
|
|
|
type program struct{}
|
|
|
|
func (p *program) Start(s service.Service) error {
|
|
// Start should not block. Do the actual work async.
|
|
go p.run()
|
|
return nil
|
|
}
|
|
|
|
func (p *program) run() {
|
|
appMain()
|
|
}
|
|
|
|
func (p *program) Stop(s service.Service) error {
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
svcConfig := &service.Config{
|
|
Name: "ETL",
|
|
DisplayName: "ETL",
|
|
Description: "this is ETL",
|
|
}
|
|
|
|
prg := &program{}
|
|
s, err := service.New(prg, svcConfig)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if len(os.Args) > 1 {
|
|
verb := os.Args[1]
|
|
switch verb {
|
|
case "install":
|
|
s.Install()
|
|
log.Println("服务安装成功")
|
|
return
|
|
case "remove":
|
|
s.Uninstall()
|
|
log.Println("服务卸载成功")
|
|
return
|
|
case "start":
|
|
s.Start()
|
|
log.Println("服务启动")
|
|
return
|
|
case "stop":
|
|
s.Stop()
|
|
log.Println("服务暂停")
|
|
return
|
|
}
|
|
|
|
}
|
|
err = s.Run()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
func appMain() {
|
|
//监控
|
|
//go func() {
|
|
// http.ListenAndServe("0.0.0.0:8899", nil)
|
|
//}()
|
|
// init app global config
|
|
err := config.InitConfig()
|
|
if err != nil {
|
|
log.Fatal("init app config failed, error:", err)
|
|
}
|
|
// create need directory
|
|
processDir, err := utils.GetCurrentPath("")
|
|
if err != nil {
|
|
log.Fatal("get process dir failed, error:" , err)
|
|
}
|
|
global.ShellPath = path.Join(processDir, "shell")
|
|
err = InitMustDirectory(processDir)
|
|
if err != nil {
|
|
log.Fatal("create need directory failed, error:", err)
|
|
}
|
|
//init logger config
|
|
if err = logger.InitConfig(); err != nil {
|
|
log.Fatal("init logger config failed, error:", err.Error())
|
|
}
|
|
// init global location time zone
|
|
global.TimezoneLocation, err = time.LoadLocation("Asia/Shanghai")
|
|
if err != nil {
|
|
log.Fatal("init global time zone location failed, error:", err)
|
|
}
|
|
|
|
// init app db engine
|
|
if err = db.InitDB(); err != nil {
|
|
log.Fatal("init app db failed, error:", err)
|
|
}
|
|
// init app mongo engine
|
|
if err = db.InitMongo(); err != nil {
|
|
log.Fatal("init app mongo failed, error:", err)
|
|
}
|
|
// init etcd client
|
|
if err = db.InitEtcdClient(); err != nil {
|
|
log.Fatal("init etcd client failed, error:", err)
|
|
}
|
|
|
|
|
|
|
|
// init db engine
|
|
err = InitDBEngine()
|
|
if err != nil {
|
|
log.Fatal("init db engine failed, error:", err)
|
|
}
|
|
|
|
// start scheduler
|
|
err = scheduler.Start()
|
|
if err != nil {
|
|
log.Fatal("start scheduler failed error:", err)
|
|
}
|
|
// register app service to etcd
|
|
if config.AppConfig.UseETCD {
|
|
common.CheckServiceDao = dal.NewETCDServiceDAO(db.AppEtcdClient, config.AppConfig.App.LocalAddr)
|
|
defer common.CheckServiceDao.RevokeLease()
|
|
err = RegisterAppService()
|
|
if err != nil {
|
|
log.Fatal("register service to etcd failed, error:", err)
|
|
}
|
|
}
|
|
// new iris application
|
|
app := iris.New()
|
|
routes.Hub(app)
|
|
app.HandleDir("/public", "./web/public")
|
|
//middleware.PreSettring(app)
|
|
|
|
// init iris config
|
|
irisConfig := iris.DefaultConfiguration()
|
|
irisConfig.DisableBodyConsumptionOnUnmarshal = false
|
|
irisConfig.DisablePathCorrection = false
|
|
irisConfig.EnablePathEscape = false
|
|
irisConfig.FireMethodNotAllowed = false
|
|
irisConfig.TimeFormat = "2006-01-02 15:04:05"
|
|
irisConfig.Charset = "UTF-8"
|
|
|
|
// run iris server
|
|
err = app.Run(iris.Addr(fmt.Sprintf(":%d", config.AppConfig.App.Port)), iris.WithConfiguration(irisConfig))
|
|
if err != nil {
|
|
log.Fatal("run iris server failed, error:", err)
|
|
}
|
|
select {
|
|
|
|
}
|
|
}
|
|
|
|
func InitDBEngine() error {
|
|
session := db.AppEngine.NewSession()
|
|
defer session.Close()
|
|
dao := dal.NewDBDAO(session)
|
|
dbLi, err := dao.Select(nil, []grmi.Field{meta.DB_ID})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, dbConfig := range dbLi {
|
|
err = driver.InitDriver(dbConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func InitMustDirectory(processDir string) error {
|
|
var err error
|
|
shellDir := path.Join(processDir,"shell")
|
|
if !utils.IsExist(shellDir) {
|
|
err = os.MkdirAll(shellDir, os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
uploadDir := path.Join(processDir, "file/uploadExcel")
|
|
if !utils.IsExist(uploadDir) {
|
|
err = os.MkdirAll(uploadDir, os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RegisterAppService() error {
|
|
dao := dal.NewETCDServiceDAO(db.AppEtcdClient, config.AppConfig.App.LocalAddr)
|
|
err := dao.RegisterService(config.AppConfig.App.Name)
|
|
return err
|
|
}
|