参数校验

2020-06-09 17:11:48 +08:00
parent 599547d627
commit 7e56fe2ee0
1 changed files with 4 additions and 0 deletions
@@ -2271,6 +2271,10 @@ public class MemoryMessagesStore implements IMessagesStore {

    @Override
    public ErrorCode blackUserRequest(String fromUser, String targetUserId, int state, long[] heads) {
+        if(state != 0 && state != 1 && state != 2){
+            return ErrorCode.INVALID_PARAMETER;
+        }
+
        if (state == 2) {
            state = 1;
        }