diff --git a/handler/beatleader/beatleader.go b/handler/beatleader/beatleader.go index 78d135c..f4a4f26 100644 --- a/handler/beatleader/beatleader.go +++ b/handler/beatleader/beatleader.go @@ -185,9 +185,9 @@ func getMyBLPic(msg model.Message) (reply *model.Reply) { var noUpdate bool = false if len(tokens) == 2 { userIdStr = tokens[1] + noUpdate = true } else { userIdStr = strconv.Itoa(int(msg.UserId)) - noUpdate = true } var data *beatleader.PlayerDataLite var lastData *beatleader.PlayerDataLite diff --git a/handler/scoresaber/score.go b/handler/scoresaber/score.go index ca441f1..b9d4e3c 100644 --- a/handler/scoresaber/score.go +++ b/handler/scoresaber/score.go @@ -210,8 +210,9 @@ func getMySSPic(msg model.Message) (reply *model.Reply) { noUpdate = false ) var userIdStr string - if len(msg.RawMsg) > len("查ss ") { - userIdStr = strings.Split(msg.RawMsg, " ")[1] + tokens := util.SplitN(msg.RawMsg, 2) + if len(tokens) == 2 { + userIdStr = tokens[1] noUpdate = true } else { userIdStr = strconv.Itoa(int(msg.UserId))