search/main.go

48 lines
992 B
Go

package main
import (
"context"
"os"
"os/signal"
"time"
"myschools.me/suguo/search/exceptionless"
"myschools.me/suguo/search/model"
"myschools.me/suguo/search/mysql"
"myschools.me/suguo/search/service"
)
func main() {
// 日志初始化
exceptionless.Init(&exceptionless.Config{
ApiKey: os.Getenv("exceptionless_apikey"),
ServerURL: os.Getenv("exceptionless_url"),
})
//mysql初始化
mysql.Init(&mysql.Config{
ConnString: os.Getenv("mysql_dsn"),
ConnMaxLifetime: 1,
MaxIdleConns: 2,
MaxOpenConns: 300,
})
db, err := mysql.New()
if err != nil {
panic(err)
}
if err := db.AutoMigrate(&model.Site{}, &model.Domain{}, &model.SiteObj{}); err != nil {
panic(err)
}
go service.Probe()
// 服务停止相应
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
<-c
_, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
exceptionless.SubmitLog("search service shutting down", "info")
os.Exit(0)
}