Merge branch 'master' into webapp-go

* Берём список каналов из конфига (пока что хардкод)
* Добавлен канальчик webapp-go для новой реализации веб-клиента (пока команды cat и кис)

aleesa-misc-lib.go

@@ -95,7 +95,7 @@ func msgParser(ctx context.Context, msg string) {
 
 		for _, command := range cmds {
 			if cmd == command {
-				sendTo = "phrases"
+				sendTo = config.ForwardChannels.Phrases
 
 				// Костыль для кармы
 				if cmd == "karma" || cmd == "карма" {
@@ -109,14 +109,28 @@ func msgParser(ctx context.Context, msg string) {
 
 		// Не угадали? акей, как насчёт модуля webapp?
 		if !done {
-			cmds = []string{"buni", "anek", "анек", "анекдот", "cat", "кис", "drink", "праздник", "fox", "лис", "frog",
-				"лягушка", "horse", "лошадь", "лошадка", "monkeyuser", "rabbit", "bunny", "кролик", "snail", "улитка",
-				"owl", "сова", "сыч", "xkcd", "tits", "boobs", "tities", "boobies", "сиси", "сисечки", "butt", "booty",
-				"ass", "попа", "попка"}
+			cmds = []string{"frog",
+				"лягушка", "horse", "лошадь", "лошадка", "rabbit", "bunny", "кролик", "snail", "улитка",
+				"owl", "сова", "сыч"}
 
 			for _, command := range cmds {
 				if cmd == command {
-					sendTo = "webapp"
+					sendTo = config.ForwardChannels.Webapp
+					done = true
+					break
+				}
+			}
+		}
+
+		// Не угадали? акей, как насчёт модуля webapp-go?
+		if !done {
+			cmds = []string{"cat", "кис", "fox", "лис", "buni", "anek", "анек", "анекдот", "xkcd", "monkeyuser", "tits",
+				"boobs", "tities", "boobies", "сиси", "сисечки", "butt", "booty", "ass", "попа", "попка", "drink",
+				"праздник"}
+
+			for _, command := range cmds {
+				if cmd == command {
+					sendTo = config.ForwardChannels.WebappGo
 					done = true
 					break
 				}
@@ -129,7 +143,7 @@ func msgParser(ctx context.Context, msg string) {
 
 			for _, command := range cmds {
 				if cmd == command {
-					sendTo = "games"
+					sendTo = config.ForwardChannels.Games
 					done = true
 					break
 				}
@@ -144,7 +158,7 @@ func msgParser(ctx context.Context, msg string) {
 
 			for _, command := range cmds {
 				if cmdLen > len(command) && cmd[0:len(command)] == command {
-					sendTo = "webapp"
+					sendTo = config.ForwardChannels.WebappGo
 					done = true
 					break
 				}
@@ -160,7 +174,7 @@ func msgParser(ctx context.Context, msg string) {
 
 			for _, command := range cmds {
 				if cmdLen > len(command) && cmd[0:len(command)] == command {
-					sendTo = "phrases"
+					sendTo = config.ForwardChannels.Phrases
 
 					if command == "karma " || command == "карма " {
 						// Костыль для кармы
@@ -180,7 +194,7 @@ func msgParser(ctx context.Context, msg string) {
 			if j.Message[msgLen-len("--"):msgLen] == "--" || j.Message[msgLen-len("++"):msgLen] == "++" {
 				// Предполагается, что менять карму мы будем для одной фразы, то есть для 1 строки
 				if len(strings.Split(j.Message, "\n")) == 1 {
-					sendTo = "phrases"
+					sendTo = config.ForwardChannels.Phrases
 
 					// Костыль для кармы
 					j.Misc.Answer = 1
@@ -296,6 +310,10 @@ func readConfig() {
 			sampleConfig.Port = 6379
 		}
 
+		if sampleConfig.Timeout == 0 {
+			sampleConfig.Timeout = 10
+		}
+
 		if sampleConfig.Loglevel == "" {
 			sampleConfig.Loglevel = "info"
 		}
@@ -319,6 +337,10 @@ func readConfig() {
 			sampleConfig.ForwardChannels.Webapp = "webapp"
 		}
 
+		if sampleConfig.ForwardChannels.WebappGo == "" {
+			sampleConfig.ForwardChannels.WebappGo = "webapp-go"
+		}
+
 		if sampleConfig.ForwardChannels.Craniac == "" {
 			sampleConfig.ForwardChannels.Craniac = "craniac"
 		}

data/config_sample.json

@@ -5,6 +5,9 @@
     # 6379 if not set
     "port" : 6379,
 
+    # Redis timeout, 10 seconds if not set
+    "timeout" : 10,
+
     # info if not set
     "loglevel" : "info",
 
@@ -25,6 +28,9 @@
        # webapp if not set
        "webapp" : "webapp",
 
+       # webapp-go if not set
+        "webapp-go" : "webapp-go",
+
        # craniac if not set
        "craniac" : "craniac"
     },

main.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"context"
 	"fmt"
 	"os"
 	"os/signal"
@@ -39,13 +38,18 @@ func init() {
 	default:
 		log.SetLevel(log.InfoLevel)
 	}
+
+	// Иницализируем клиента Редиски
+	redisClient = redis.NewClient(&redis.Options{
+		Addr: fmt.Sprintf("%s:%d", config.Server, config.Port),
+	}).WithContext(ctx).WithTimeout(time.Duration(config.Timeout) * time.Second)
+
+	// Обозначим, что хотим после соединения подписаться на события из канала config.Channel
+	subscriber = redisClient.Subscribe(ctx, config.Channel)
 }
 
 // Основная функция программы, не добавить и не убавить
 func main() {
-	// Main context
-	var ctx = context.Background()
-
 	// Откроем лог и скормим его логгеру
 	if config.Log != "" {
 		logfile, err := os.OpenFile(config.Log, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
@@ -57,14 +61,6 @@ func main() {
 		log.SetOutput(logfile)
 	}
 
-	// Иницализируем клиента
-	redisClient = redis.NewClient(&redis.Options{
-		Addr: fmt.Sprintf("%s:%d", config.Server, config.Port),
-	})
-
-	log.Debugf("Lazy connect() to redis at %s:%d", config.Server, config.Port)
-	subscriber = redisClient.Subscribe(ctx, config.Channel)
-
 	// Самое время поставить траппер сигналов
 	signal.Notify(sigChan,
 		syscall.SIGINT,
@@ -73,25 +69,13 @@ func main() {
 
 	go sigHandler()
 
-	// Обработчик событий от редиски
-	for {
-		if shutdown {
-			time.Sleep(1 * time.Second)
-			continue
+	// Начнём выгребать события из редиски (длина ковеера/буфера канала по-умолчанию - 100 сообщений)
+	ch := subscriber.Channel()
+
+	for msg := range ch {
+		if !shutdown {
+			msgParser(ctx, msg.Payload)
 		}
-
-		msg, err := subscriber.ReceiveMessage(ctx)
-
-		if err != nil {
-			if !shutdown {
-				log.Warnf("Unable to connect to redis at %s:%d: %s", config.Server, config.Port, err)
-			}
-
-			time.Sleep(1 * time.Second)
-			continue
-		}
-
-		go msgParser(ctx, msg.Payload)
 	}
 }
 

types.go

@@ -4,14 +4,16 @@ package main
 type myConfig struct {
 	Server          string `json:"server,omitempty"`
 	Port            int    `json:"port,omitempty"`
+	Timeout         int    `json:"timeout,omitempty"`
 	Loglevel        string `json:"loglevel,omitempty"`
 	Log             string `json:"log,omitempty"`
 	Channel         string `json:"channel,omitempty"`
 	ForwardChannels struct {
-		Games   string `json:"games,omitempty"`
-		Phrases string `json:"phrases,omitempty"`
-		Webapp  string `json:"webapp,omitempty"`
-		Craniac string `json:"craniac,omitempty"`
+		Games    string `json:"games,omitempty"`
+		Phrases  string `json:"phrases,omitempty"`
+		Webapp   string `json:"webapp,omitempty"`
+		WebappGo string `json:"webapp-go,omitempty"`
+		Craniac  string `json:"craniac,omitempty"`
 	} `json:"forward_channels"`
 	Csign       string `json:"csign,omitempty"`
 	ForwardsMax int64  `json:"forwards_max,omitempty"`