middleware/logger: trim callback data delimiter

heilkit · Oct 20, 2020 · 1755696 · 1755696
1 parent 8805f24
commit 1755696
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/middleware/logger.go b/middleware/logger.go
@@ -23,7 +23,12 @@ func Logger(logger *logrus.Logger, fieldsFunc ...LoggerFieldsFunc) tele.Middlewa
 		return func(c tele.Context) error {
 			data := c.Data()
 			if clb := c.Callback(); clb != nil {
-				data = clb.Unique + "|" + data
+				uniq := clb.Unique
+				if data == "" {
+					data = uniq
+				} else {
+					data = uniq + "|" + data
+				}
 			} else if data == "" {
 				data = c.Text()
 			}