Some TG-specific improvements with possible other methods as well
This commit is contained in:
parent
1e89fea274
commit
079e32e893
27
main.go
27
main.go
|
@ -30,12 +30,13 @@ func (self TgSender) Send(msg GenericMessage) error {
|
||||||
}
|
}
|
||||||
b := strings.Builder{}
|
b := strings.Builder{}
|
||||||
for _, msg := range msg.msgs {
|
for _, msg := range msg.msgs {
|
||||||
b.WriteString(msg.String())
|
b.WriteString(msg.TgString())
|
||||||
b.WriteRune('\n')
|
b.WriteRune('\n')
|
||||||
}
|
}
|
||||||
message := b.String()
|
message := b.String()
|
||||||
for _, chatId := range *chatIds {
|
for _, chatId := range *chatIds {
|
||||||
toSend := tgbotapi.NewMessage(chatId, message)
|
toSend := tgbotapi.NewMessage(chatId, message)
|
||||||
|
toSend.ParseMode = "HTML"
|
||||||
_, err := self.bot.Send(toSend)
|
_, err := self.bot.Send(toSend)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -44,14 +45,22 @@ func (self TgSender) Send(msg GenericMessage) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func sendTgMessage(tgEvents chan GenericMessage, msg string, chatId int64) {
|
func sendTgMessage(tgEvents chan GenericMessage, msg MsgContent, chatId int64) {
|
||||||
tgEvents <- GenericMessage{TgChatId{chatId}, []MsgContent{{msg}}}
|
tgEvents <- GenericMessage{TgChatId{chatId}, []MsgContent{msg}}
|
||||||
}
|
}
|
||||||
|
|
||||||
type RealSleeper struct {
|
type RealSleeper struct {
|
||||||
duration time.Duration
|
duration time.Duration
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type SatelMsgContent struct {
|
||||||
|
ev satel.Event
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self SatelMsgContent) TgString() string {
|
||||||
|
return fmt.Sprint("<b>", self.ev.Type, "</b>, <i>index</i>:<b>", self.ev.Index, "</b>, value:<b>", self.ev.Value, "</b>")
|
||||||
|
}
|
||||||
|
|
||||||
func (self RealSleeper) Sleep(ch chan<- interface{}) {
|
func (self RealSleeper) Sleep(ch chan<- interface{}) {
|
||||||
go func() {
|
go func() {
|
||||||
time.Sleep(self.duration)
|
time.Sleep(self.duration)
|
||||||
|
@ -110,11 +119,17 @@ func main() {
|
||||||
logger.Print("Created Telegram Bot API client")
|
logger.Print("Created Telegram Bot API client")
|
||||||
|
|
||||||
tgSender := TgSender{bot}
|
tgSender := TgSender{bot}
|
||||||
Consume(SendToTg(tgSenderWorker(tgEvents, &wg, sleeper, log.New(os.Stderr, "TgSender", log.Lmicroseconds)),
|
|
||||||
tgSender, &wg, log.New(os.Stderr, "SendToTg", log.Lmicroseconds)))
|
Consume(
|
||||||
|
SendToTg(
|
||||||
|
tgSenderWorker(tgEvents, &wg, sleeper, log.New(os.Stderr, "TgSender", log.Lmicroseconds)),
|
||||||
|
tgSender, &wg, log.New(os.Stderr, "SendToTg", log.Lmicroseconds)))
|
||||||
|
|
||||||
for e := range FilterByLastSeen(s.Events, "hs_wro_last_seen.bin", log.New(os.Stderr, "FilterByLastSeen", log.Lmicroseconds)) {
|
for e := range FilterByLastSeen(s.Events, "hs_wro_last_seen.bin", log.New(os.Stderr, "FilterByLastSeen", log.Lmicroseconds)) {
|
||||||
logger.Print("Received change from SATEL: ", e)
|
logger.Print("Received change from SATEL: ", e)
|
||||||
sendTgMessage(tgEvents, fmt.Sprint("Change from SATEL: type:", e.Type, ", index:", e.Index, ", value:", e.Value), chatIds[0])
|
for _, chatId := range chatIds {
|
||||||
|
sendTgMessage(tgEvents, SatelMsgContent{e}, chatId)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
close(tgEvents)
|
close(tgEvents)
|
||||||
|
|
|
@ -32,11 +32,15 @@ func Consume(events <-chan GenericMessage) {
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
type MsgContent struct {
|
type MsgContent interface {
|
||||||
|
TgString() string
|
||||||
|
}
|
||||||
|
|
||||||
|
type StringMsgContent struct {
|
||||||
msg string
|
msg string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (self MsgContent) String() string { return self.msg }
|
func (self StringMsgContent) TgString() string { return self.msg }
|
||||||
|
|
||||||
type GenericMessage struct {
|
type GenericMessage struct {
|
||||||
chatIds ChatId
|
chatIds ChatId
|
||||||
|
|
|
@ -30,6 +30,14 @@ func (self *MockSleeper) Sleep(ch chan<- interface{}) {
|
||||||
self.callCount += 1
|
self.callCount += 1
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type FakeChatId struct {
|
||||||
|
FakeId int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (self FakeChatId) GetTgIds() *[]int64 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func TestMessageThrottling(t *testing.T) {
|
func TestMessageThrottling(t *testing.T) {
|
||||||
testEvents := make(chan GenericMessage)
|
testEvents := make(chan GenericMessage)
|
||||||
wg := sync.WaitGroup{}
|
wg := sync.WaitGroup{}
|
||||||
|
@ -37,18 +45,21 @@ func TestMessageThrottling(t *testing.T) {
|
||||||
mockSleeper := MockSleeper{nil, 0}
|
mockSleeper := MockSleeper{nil, 0}
|
||||||
Consume(SendToTg(tgSenderWorker(testEvents, &wg, &mockSleeper, log.New(io.Discard, "", log.Ltime)),
|
Consume(SendToTg(tgSenderWorker(testEvents, &wg, &mockSleeper, log.New(io.Discard, "", log.Ltime)),
|
||||||
&mockSender, &wg, log.New(io.Discard, "", log.Ltime)))
|
&mockSender, &wg, log.New(io.Discard, "", log.Ltime)))
|
||||||
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test1"}}}
|
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{StringMsgContent{"test1"}}}
|
||||||
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{{"test3"}}}
|
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{StringMsgContent{"test3"}}}
|
||||||
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test2"}}}
|
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{StringMsgContent{"test2"}}}
|
||||||
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{{"test4"}}}
|
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{StringMsgContent{"test4"}}}
|
||||||
|
testEvents <- GenericMessage{FakeChatId{123}, []MsgContent{StringMsgContent{"testFake"}}}
|
||||||
assert.Equal(t, 1, mockSleeper.callCount)
|
assert.Equal(t, 1, mockSleeper.callCount)
|
||||||
*mockSleeper.ch <- nil
|
*mockSleeper.ch <- nil
|
||||||
assert.Equal(t, 1, mockSleeper.callCount)
|
assert.Equal(t, 1, mockSleeper.callCount)
|
||||||
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test5"}}}
|
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{StringMsgContent{"test5"}}}
|
||||||
close(testEvents)
|
close(testEvents)
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
assert.Equal(t, 2, mockSleeper.callCount)
|
assert.Equal(t, 2, mockSleeper.callCount)
|
||||||
|
|
||||||
assert.Len(t, mockSender.messages, 3)
|
assert.Len(t, mockSender.messages, 4)
|
||||||
assert.Contains(t, mockSender.messages, GenericMessage{TgChatId{123}, []MsgContent{{"test1"}, {"test2"}}})
|
assert.Contains(t, mockSender.messages, GenericMessage{TgChatId{123}, []MsgContent{StringMsgContent{"test1"}, StringMsgContent{"test2"}}})
|
||||||
|
assert.Contains(t, mockSender.messages, GenericMessage{TgChatId{124}, []MsgContent{StringMsgContent{"test3"}, StringMsgContent{"test4"}}})
|
||||||
|
assert.Contains(t, mockSender.messages, GenericMessage{FakeChatId{123}, []MsgContent{StringMsgContent{"testFake"}}})
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue