ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-08-31 21:24:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Rework stats

Browse Source
This commit is contained in:
9seconds
2021-03-27 22:04:30 +03:00
parent 5eca6ecb05
commit bc2bd4510a
19 changed files with 273 additions and 232 deletions
Download Patch File Download Diff File Expand all files Collapse all files
events/event_stream.go
+2 -2
View File
@@ -73,8 +73,8 @@ func eventStreamProcessor(ctx context.Context, eventChan <-chan mtglib.Event, ob
observer.EventStart(typedEvt)
case mtglib.EventConnectedToDC:
observer.EventConnectedToDC(typedEvt)
case mtglib.EventTraffic:
observer.EventTraffic(typedEvt)
case mtglib.EventTelegramTraffic:
observer.EventTelegramTraffic(typedEvt)
case mtglib.EventFinish:
observer.EventFinish(typedEvt)
case mtglib.EventIPBlocklisted:
events/event_stream_test.go
+4 -4
View File
@@ -90,8 +90,8 @@ func (suite *EventStreamTestSuite) TestEventConnectedToDC() {
time.Sleep(100 * time.Millisecond)
}
func (suite *EventStreamTestSuite) TestEventTraffic() {
evt := mtglib.EventTraffic{
func (suite *EventStreamTestSuite) TestEventTelegramTraffic() {
evt := mtglib.EventTelegramTraffic{
CreatedAt: time.Now(),
ConnID: "connID",
Traffic: 1024,
@@ -100,10 +100,10 @@ func (suite *EventStreamTestSuite) TestEventTraffic() {
for _, v := range []*ObserverMock{suite.observerMock1, suite.observerMock2} {
v.
On("EventTraffic", mock.Anything).
On("EventTelegramTraffic", mock.Anything).
Once().
Run(func(args mock.Arguments) {
caught := args.Get(0).(mtglib.EventTraffic)
caught := args.Get(0).(mtglib.EventTelegramTraffic)
suite.Equal(evt.CreatedAt, caught.CreatedAt)
suite.Equal(evt.ConnID, caught.ConnID)
events/init.go
+1 -1
View File
@@ -6,7 +6,7 @@ type Observer interface {
EventStart(mtglib.EventStart)
EventFinish(mtglib.EventFinish)
EventConnectedToDC(mtglib.EventConnectedToDC)
EventTraffic(mtglib.EventTraffic)
EventTelegramTraffic(mtglib.EventTelegramTraffic)
EventConcurrencyLimited(mtglib.EventConcurrencyLimited)
EventIPBlocklisted(mtglib.EventIPBlocklisted)
events/init_test.go
+1 -1
View File
@@ -17,7 +17,7 @@ func (o *ObserverMock) EventConnectedToDC(evt mtglib.EventConnectedToDC) {
o.Called(evt)
}
func (o *ObserverMock) EventTraffic(evt mtglib.EventTraffic) {
func (o *ObserverMock) EventTelegramTraffic(evt mtglib.EventTelegramTraffic) {
o.Called(evt)
}
events/multi_observer.go
+2 -2
View File
@@ -40,7 +40,7 @@ func (m multiObserver) EventConnectedToDC(evt mtglib.EventConnectedToDC) {
wg.Wait()
}
func (m multiObserver) EventTraffic(evt mtglib.EventTraffic) {
func (m multiObserver) EventTelegramTraffic(evt mtglib.EventTelegramTraffic) {
wg := &sync.WaitGroup{}
wg.Add(len(m.observers))
@@ -48,7 +48,7 @@ func (m multiObserver) EventTraffic(evt mtglib.EventTraffic) {
go func(obs Observer) {
defer wg.Done()
obs.EventTraffic(evt)
obs.EventTelegramTraffic(evt)
}(v)
}
events/noop.go
+1 -1
View File
@@ -19,7 +19,7 @@ type noopObserver struct{}
func (n noopObserver) EventStart(_ mtglib.EventStart) {}
func (n noopObserver) EventConnectedToDC(_ mtglib.EventConnectedToDC) {}
func (n noopObserver) EventTraffic(_ mtglib.EventTraffic) {}
func (n noopObserver) EventTelegramTraffic(_ mtglib.EventTelegramTraffic) {}
func (n noopObserver) EventFinish(_ mtglib.EventFinish) {}
func (n noopObserver) EventConcurrencyLimited(_ mtglib.EventConcurrencyLimited) {}
func (n noopObserver) EventIPBlocklisted(_ mtglib.EventIPBlocklisted) {}
events/noop_test.go
+1 -1
View File
@@ -31,7 +31,7 @@ func (suite *NoopTestSuite) SetupSuite() {
RemoteIP: net.ParseIP("127.1.0.1"),
DC: 2,
},
"traffic": mtglib.EventTraffic{
"telegram-traffic": mtglib.EventTelegramTraffic{
CreatedAt: time.Now(),
ConnID: "connID",
Traffic: 1000,