如何在Firebase实时数据库中实现安全的点赞/点踩计数器

数据库结构设计 #

需要补充两个节点的字段，用来记录用户操作和站点投票数：

• Users/{userId}/votes/{stationId}: 存储用户对该站点的投票类型，值为"like"、"dislike"或null（未投票状态）
• STATIONS/{stationId}: 新增upvote（点赞数，整数）和downvote（点踩数，整数）字段，初始值设为0

核心代码实现 #

初始化按钮与基础引用 #

private DatabaseReference mUserVotesRef;
private DatabaseReference mStationVotesRef;
private String currentUserId; // 当前登录用户ID
private String currentStationId; // 当前操作的站点ID

// 初始化按钮
mlike = root.findViewById(R.id.btn_like);
mdislike = root.findViewById(R.id.btn_dislike);

// 获取当前用户ID和站点ID（根据你的业务场景替换，比如从登录状态/Intent获取）
currentUserId = FirebaseAuth.getInstance().getCurrentUser().getUid();
currentStationId = "目标站点ID";

// 初始化数据库引用
mUserVotesRef = FirebaseDatabase.getInstance().getReference().child("Users").child(currentUserId).child("votes").child(currentStationId);
mStationVotesRef = FirebaseDatabase.getInstance().getReference().child("STATIONS").child(currentStationId);

// 加载用户已有投票状态，设置按钮初始状态
loadUserVoteStatus();

// 设置按钮点击监听
mlike.setOnClickListener(v -> likeStations());
mdislike.setOnClickListener(v -> dislikeStations());

加载用户投票状态（控制按钮可用性） #

private void loadUserVoteStatus() {
    mUserVotesRef.addListenerForSingleValueEvent(new ValueEventListener() {
        @Override
        public void onDataChange(DataSnapshot snapshot) {
            String voteType = snapshot.getValue(String.class);
            // 根据投票状态设置按钮可点击性和样式
            if ("like".equals(voteType)) {
                mlike.setEnabled(false);
                mdislike.setEnabled(true);
                mlike.setImageResource(R.drawable.ic_liked); // 替换为你的已点赞图标
                mdislike.setImageResource(R.drawable.ic_dislike_normal); // 替换为你的未点踩图标
            } else if ("dislike".equals(voteType)) {
                mdislike.setEnabled(false);
                mlike.setEnabled(true);
                mdislike.setImageResource(R.drawable.ic_disliked); // 替换为你的已点踩图标
                mlike.setImageResource(R.drawable.ic_like_normal); // 替换为你的未点赞图标
            } else {
                // 未投票状态，两个按钮都可点击
                mlike.setEnabled(true);
                mdislike.setEnabled(true);
                mlike.setImageResource(R.drawable.ic_like_normal);
                mdislike.setImageResource(R.drawable.ic_dislike_normal);
            }
        }

        @Override
        public void onCancelled(DatabaseError error) {
            // 处理数据库读取错误
        }
    });
}

点赞方法实现 #

private void likeStations() {
    mUserVotesRef.addListenerForSingleValueEvent(new ValueEventListener() {
        @Override
        public void onDataChange(DataSnapshot snapshot) {
            String existingVote = snapshot.getValue(String.class);

            // 用事务保证计数操作的原子性，避免并发错误
            mStationVotesRef.runTransaction(new Transaction.Handler() {
                @NonNull
                @Override
                public Transaction.Result doTransaction(@NonNull MutableData mutableData) {
                    Integer upvote = mutableData.child("upvote").getValue(Integer.class);
                    Integer downvote = mutableData.child("downvote").getValue(Integer.class);

                    if (upvote == null) upvote = 0;
                    if (downvote == null) downvote = 0;

                    if ("dislike".equals(existingVote)) {
                        // 之前点过踩，切换点赞：点踩数减1，点赞数加1
                        downvote -= 1;
                        upvote += 1;
                    } else if (existingVote == null) {
                        // 未投过票，直接增加点赞数
                        upvote += 1;
                    }
                    // 已点过赞的情况不做操作（按钮已禁用）

                    mutableData.child("upvote").setValue(upvote);
                    mutableData.child("downvote").setValue(downvote);
                    return Transaction.success(mutableData);
                }

                @Override
                public void onComplete(@Nullable DatabaseError error, boolean committed, @Nullable DataSnapshot currentData) {
                    if (committed && error == null) {
                        // 更新用户投票记录
                        mUserVotesRef.setValue("like");
                        // 刷新按钮状态
                        loadUserVoteStatus();
                    }
                }
            });
        }

        @Override
        public void onCancelled(DatabaseError error) {
            // 处理错误
        }
    });
}

点踩方法实现 #

private void dislikeStations() {
    mUserVotesRef.addListenerForSingleValueEvent(new ValueEventListener() {
        @Override
        public void onDataChange(DataSnapshot snapshot) {
            String existingVote = snapshot.getValue(String.class);

            mStationVotesRef.runTransaction(new Transaction.Handler() {
                @NonNull
                @Override
                public Transaction.Result doTransaction(@NonNull MutableData mutableData) {
                    Integer upvote = mutableData.child("upvote").getValue(Integer.class);
                    Integer downvote = mutableData.child("downvote").getValue(Integer.class);

                    if (upvote == null) upvote = 0;
                    if (downvote == null) downvote = 0;

                    if ("like".equals(existingVote)) {
                        // 之前点过赞，切换点踩：点赞数减1，点踩数加1
                        upvote -= 1;
                        downvote += 1;
                    } else if (existingVote == null) {
                        // 未投过票，直接增加点踩数
                        downvote += 1;
                    }
                    // 已点过踩的情况不做操作（按钮已禁用）

                    mutableData.child("upvote").setValue(upvote);
                    mutableData.child("downvote").setValue(downvote);
                    return Transaction.success(mutableData);
                }

                @Override
                public void onComplete(@Nullable DatabaseError error, boolean committed, @Nullable DataSnapshot currentData) {
                    if (committed && error == null) {
                        // 更新用户投票记录
                        mUserVotesRef.setValue("dislike");
                        // 刷新按钮状态
                        loadUserVoteStatus();
                    }
                }
            });
        }

        @Override
        public void onCancelled(DatabaseError error) {
            // 处理错误
        }
    });
}

关键说明 #

• 事务操作：使用Firebase事务保证投票计数的原子性，避免多用户并发操作导致的计数错误。
• 用户投票记录：通过Users节点下的votes子节点记录用户对每个站点的投票状态，确保同一用户只能对站点投一次票，切换投票时自动调整对应计数。
• 按钮状态控制：页面加载时根据用户已有投票状态设置按钮的可点击性和样式，从交互层面避免重复操作。

内容的提问来源于stack exchange，提问作者markharrop