Skip to content

Commit

Permalink
升级框架,解决监听端口无法变更问题
Browse files Browse the repository at this point in the history
  • Loading branch information
mylxsw committed Aug 20, 2020
1 parent ecf8879 commit eda2fd1
Show file tree
Hide file tree
Showing 17 changed files with 49 additions and 40 deletions.
4 changes: 2 additions & 2 deletions agent/api/message.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import (
"github.com/mylxsw/adanos-alert/rpc/protocol"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/mylxsw/glacier/web"
)

Expand All @@ -34,7 +34,7 @@ func (m *MessageController) Register(router *web.Router) {
}

func (m *MessageController) saveMessage(msgRepo store.MessageStore, commonMessage misc.CommonMessage, ctx web.Context) error {
commonMessage.Meta["adanos_agent_version"] = m.cc.MustGet(glacier.VersionKey).(string)
commonMessage.Meta["adanos_agent_version"] = m.cc.MustGet(infra.VersionKey).(string)
commonMessage.Meta["adanos_agent_ip"] = misc.ServerIP()
m.cc.MustResolve(func(db *ledis.DB) {
agentID, _ := db.Get([]byte("agent-id"))
Expand Down
4 changes: 2 additions & 2 deletions agent/api/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"github.com/gorilla/mux"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/mylxsw/glacier/web"
"github.com/prometheus/client_golang/prometheus/promhttp"
)
Expand All @@ -13,7 +13,7 @@ type ServiceProvider struct{}

func (s ServiceProvider) Register(app container.Container) {}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.WebAppRouter(routers(app.Container()))
app.WebAppMuxRouter(func(router *mux.Router) {
// prometheus metrics
Expand Down
4 changes: 2 additions & 2 deletions agent/job/heartbeat.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ import (
"github.com/mylxsw/adanos-alert/rpc/protocol"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
)

func heartbeatJob(cc container.Container, db *ledis.DB, hs protocol.HeartbeatClient) error {
Expand All @@ -19,7 +19,7 @@ func heartbeatJob(cc container.Container, db *ledis.DB, hs protocol.HeartbeatCli
AgentTs: time.Now().Unix(),
AgentIP: misc.ServerIP(),
AgentID: string(agentID),
ClientVersion: cc.MustGet(glacier.VersionKey).(string),
ClientVersion: cc.MustGet(infra.VersionKey).(string),
}

pong, err := hs.Ping(ctx, &pingReq)
Expand Down
4 changes: 2 additions & 2 deletions agent/job/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package job
import (
"github.com/mylxsw/adanos-alert/rpc/protocol"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/cron"
"github.com/mylxsw/glacier/infra"
)

type ServiceProvider struct{}
Expand All @@ -14,7 +14,7 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(protocol.NewHeartbeatClient)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.Cron(func(cr cron.Manager, cc container.Container) error {
cc.Must(cr.Add("sync-message", "@every 5s", messageSyncJob))
cc.Must(cr.Add("heartbeat", "@every 10s", heartbeatJob))
Expand Down
4 changes: 2 additions & 2 deletions agent/store/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ package store

import (
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
)

type ServiceProvider struct{}
Expand All @@ -11,6 +11,6 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(NewMessageStore)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {

}
4 changes: 2 additions & 2 deletions api/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ import (
"github.com/mylxsw/adanos-alert/configs"
_ "github.com/mylxsw/adanos-alert/docs"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/prometheus/client_golang/prometheus/promhttp"
httpSwagger "github.com/swaggo/http-swagger"
)
Expand All @@ -29,7 +29,7 @@ type ServiceProvider struct{}

func (s ServiceProvider) Register(app container.Container) {}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.MustResolve(func(conf *configs.Config) {
app.WebAppRouter(routers(app.Container()))
app.WebAppMuxRouter(func(router *mux.Router) {
Expand Down
13 changes: 10 additions & 3 deletions cmd/agent/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@ import (
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/asteria/writer"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/mylxsw/glacier/listener"
"github.com/mylxsw/glacier/starter/application"
"github.com/mylxsw/glacier/web"
"github.com/mylxsw/go-toolkit/network"
Expand Down Expand Up @@ -53,8 +54,14 @@ func main() {
Usage: "本地数据库存储目录",
Value: "/tmp/adanos-agent",
}))
app.AddFlags(altsrc.NewStringFlag(cli.StringFlag{
Name: "listen",
Usage: "listen address",
EnvVar: "ADANOS_AGENT_LISTEN_ADDR",
Value: "127.0.0.1:29999",
}))

app.WithHttpServer().TCPListenerAddr("127.0.0.1:29999")
app.WithHttpServer(listener.FlagContext("listen"))

app.BeforeServerStart(func(cc container.Container) error {
stackWriter := writer.NewStackWriter()
Expand All @@ -71,7 +78,7 @@ func main() {
})

// Config
app.Singleton(func(c glacier.FlagContext) *config.Config {
app.Singleton(func(c infra.FlagContext) *config.Config {
return &config.Config{
DataDir: c.String("data_dir"),
ServerAddr: c.String("server_addr"),
Expand Down
9 changes: 5 additions & 4 deletions cmd/server/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@ import (

"github.com/mylxsw/adanos-alert/rpc"
"github.com/mylxsw/asteria/writer"
"github.com/mylxsw/glacier/infra"
"github.com/mylxsw/glacier/listener"

"github.com/gorilla/mux"
"github.com/mylxsw/adanos-alert/api"
Expand All @@ -22,7 +24,6 @@ import (
"github.com/mylxsw/asteria/level"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/starter/application"
"github.com/mylxsw/glacier/web"
"github.com/urfave/cli"
Expand All @@ -39,7 +40,7 @@ func main() {
app.AddFlags(altsrc.NewStringFlag(cli.StringFlag{
Name: "listen",
Usage: "http listen addr",
Value: "19999",
Value: ":19999",
}))
app.AddFlags(altsrc.NewStringFlag(cli.StringFlag{
Name: "grpc_listen",
Expand Down Expand Up @@ -149,7 +150,7 @@ func main() {
Usage: "阿里云语音通知模板变量名",
}))

app.WithHttpServer().TCPListenerAddr(":19999")
app.WithHttpServer(listener.FlagContext("listen"))

app.BeforeServerStart(func(cc container.Container) error {
stackWriter := writer.NewStackWriter()
Expand All @@ -165,7 +166,7 @@ func main() {
return nil
})

app.Singleton(func(c glacier.FlagContext) *configs.Config {
app.Singleton(func(c infra.FlagContext) *configs.Config {
aggregationPeriod, err := time.ParseDuration(c.String("aggregation_period"))
if err != nil {
log.Warningf("invalid argument [aggregation_period: %s], using default value", c.String("aggregation_period"))
Expand Down
2 changes: 1 addition & 1 deletion go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ require (
github.com/mylxsw/asteria v0.0.0-20200709071207-26dd334a6e51
github.com/mylxsw/coll v0.0.0-20190810120926-a7a6f0f4bae8
github.com/mylxsw/container v0.0.0-20200525090619-01208c02b074
github.com/mylxsw/glacier v0.0.0-20200814023904-e2c4a8f9ea69
github.com/mylxsw/glacier v0.0.0-20200820103600-705c319bcccc
github.com/mylxsw/go-toolkit v0.0.0-20191208081907-50a06279f988
github.com/mylxsw/graceful v0.0.0-20200605063420-3c53968cf134
github.com/pkg/errors v0.9.1
Expand Down
4 changes: 2 additions & 2 deletions go.sum
Original file line number Diff line number Diff line change
Expand Up @@ -172,8 +172,8 @@ github.com/mylxsw/container v0.0.0-20191208075953-c8ee6e3238cc h1:xBh4hQSO+fcWa/
github.com/mylxsw/container v0.0.0-20191208075953-c8ee6e3238cc/go.mod h1:v2QwNL+V2nI1o7naopTXXalpa1Y6b5E8lCwwANiYfyc=
github.com/mylxsw/container v0.0.0-20200525090619-01208c02b074 h1:8OLLLh/tG7M5kQrkmFIVYFpj2bLUtYZ3q+jp4vcA5gQ=
github.com/mylxsw/container v0.0.0-20200525090619-01208c02b074/go.mod h1:UDbF8EtqT7jB0yDc6g9u4P8ORvw4RbFot1QputpgR5U=
github.com/mylxsw/glacier v0.0.0-20200814023904-e2c4a8f9ea69 h1:HV+OSJ0T+L9xtB3WyHj6SDc3TQa2eXgBu1Mx/3gMhLM=
github.com/mylxsw/glacier v0.0.0-20200814023904-e2c4a8f9ea69/go.mod h1:gR/AjRPG59eaf9O94gcktmuE+cR9F04WzIcPCDVSy2M=
github.com/mylxsw/glacier v0.0.0-20200820103600-705c319bcccc h1:UZ+X4drqKtDR0/OKxuvz5nzHssurkVkJW2vR1qbfVlY=
github.com/mylxsw/glacier v0.0.0-20200820103600-705c319bcccc/go.mod h1:gR/AjRPG59eaf9O94gcktmuE+cR9F04WzIcPCDVSy2M=
github.com/mylxsw/go-toolkit v0.0.0-20191208081907-50a06279f988 h1:eUlVMPyxeWUogeqwAkat6eE+3eY5fjP/UAXyhAf/7Jw=
github.com/mylxsw/go-toolkit v0.0.0-20191208081907-50a06279f988/go.mod h1:aJAUawotu1IMtEPXInIm6qwiAEIMSGRAMH6otakYCWw=
github.com/mylxsw/graceful v0.0.0-20200605063420-3c53968cf134 h1:uYT1HZC8kyKGtAR/Los/azrD8uROgkM8HHuhN2bzPmI=
Expand Down
4 changes: 2 additions & 2 deletions internal/action/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ import (
"github.com/mylxsw/adanos-alert/internal/repository"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/pkg/errors"
)

Expand All @@ -15,7 +15,7 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(NewManager)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.MustResolve(func(manager Manager, queueManager queue.Manager) {
manager.Register("http", NewHttpAction(manager))
manager.Register("dingding", NewDingdingAction(manager))
Expand Down
9 changes: 5 additions & 4 deletions internal/job/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,17 @@ package job

import (
"fmt"
"os"
"sync"

"github.com/mylxsw/adanos-alert/internal/repository"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/glacier/infra"
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson/primitive"
"os"
"sync"

"github.com/mylxsw/adanos-alert/configs"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/cron"
)

Expand All @@ -22,7 +23,7 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(NewTrigger)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.Cron(func(cr cron.Manager, cc container.Container) error {

return cc.Resolve(func(conf *configs.Config, aggregationJob *AggregationJob, alertJob *TriggerJob, lockRepo repository.LockRepo) {
Expand Down
6 changes: 3 additions & 3 deletions internal/queue/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ import (

"github.com/mylxsw/adanos-alert/configs"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
)

type ServiceProvider struct{}
Expand All @@ -16,11 +16,11 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(NewManager)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {

}

func (s ServiceProvider) Daemon(ctx context.Context, app glacier.Glacier) {
func (s ServiceProvider) Daemon(ctx context.Context, app infra.Glacier) {
app.MustResolve(func(manager Manager, conf *configs.Config) {
var wg sync.WaitGroup
wg.Add(conf.QueueWorkerNum)
Expand Down
4 changes: 2 additions & 2 deletions internal/repository/impl/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@ import (
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/coll"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/cron"
"github.com/mylxsw/glacier/infra"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
Expand All @@ -30,7 +30,7 @@ func (s ServiceProvider) Register(app container.Container) {
app.MustSingleton(NewAgentRepo)
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.Cron(func(cr cron.Manager, cc container.Container) error {
return cc.Resolve(func(kvRepo repository.KVRepo, groupRepo repository.MessageGroupRepo, msgRepo repository.MessageRepo, conf *configs.Config) {
_ = cr.Add("kv_repository_gc", "@hourly", func() {
Expand Down
4 changes: 2 additions & 2 deletions migrate/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ package migrate

import (
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
)

type ServiceProvider struct {
Expand All @@ -11,6 +11,6 @@ type ServiceProvider struct {
func (s ServiceProvider) Register(app container.Container) {
}

func (s ServiceProvider) Boot(app glacier.Glacier) {
func (s ServiceProvider) Boot(app infra.Glacier) {
app.MustResolve(initPredefinedTemplates)
}
4 changes: 2 additions & 2 deletions rpc/heartbeat.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ import (
"github.com/mylxsw/adanos-alert/rpc/protocol"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
)

// HeartbeatService is a service server for heartbeat
Expand Down Expand Up @@ -36,6 +36,6 @@ func (h *HeartbeatService) Ping(ctx context.Context, request *protocol.PingReque
})
return &protocol.PongResponse{
ServerTs: time.Now().Unix(),
ServerVersion: h.cc.MustGet(glacier.VersionKey).(string),
ServerVersion: h.cc.MustGet(infra.VersionKey).(string),
}, nil
}
6 changes: 3 additions & 3 deletions rpc/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import (
"github.com/mylxsw/adanos-alert/rpc/protocol"
"github.com/mylxsw/asteria/log"
"github.com/mylxsw/container"
"github.com/mylxsw/glacier"
"github.com/mylxsw/glacier/infra"
"github.com/mylxsw/graceful"
"google.golang.org/grpc"
)
Expand All @@ -36,14 +36,14 @@ func (p Provider) Register(app container.Container) {

}

func (p Provider) Boot(app glacier.Glacier) {
func (p Provider) Boot(app infra.Glacier) {
app.MustResolve(func(serv *grpc.Server) {
protocol.RegisterMessageServer(serv, NewMessageService(app.Container()))
protocol.RegisterHeartbeatServer(serv, NewHeartbeatService(app.Container()))
})
}

func (p Provider) Daemon(_ context.Context, app glacier.Glacier) {
func (p Provider) Daemon(_ context.Context, app infra.Glacier) {
app.MustResolve(func(serv *grpc.Server, conf *configs.Config, gf graceful.Graceful) {
listener, err := net.Listen("tcp", conf.GRPCListen)
if err != nil {
Expand Down

0 comments on commit eda2fd1

Please sign in to comment.