Fix(stdin-handler): Handle using reader instead of scanf

This commit is contained in:
2022-09-10 02:36:18 +08:00
parent 95ff3dabab
commit bcc99e523e

View File

@@ -1,6 +1,7 @@
package main
import (
"bufio"
"fmt"
"log"
"net/http"
@@ -43,17 +44,22 @@ func main() {
func handleStdIn(webhookFlag *bool, webhookURL *string) {
for {
var err error
var currStr string
_, err := fmt.Scanln(&currStr)
if err != nil {
return
}
// Handle string
if *webhookFlag {
err = sendWebhook(&currStr, webhookURL)
reader := bufio.NewReader(os.Stdin)
for err == nil {
currStr, err = reader.ReadString('\n')
if err != nil {
log.Println(err)
return
}
// Handle string
if *webhookFlag {
err = sendWebhook(&currStr, webhookURL)
if err != nil {
log.Println(err)
}
}
}
}