如何在 Java 中使用“任务”重构此代码？

Question

问题是关于this code（在Android应用程序中）：

private Task<QuerySnapshot> getVotesFromDB() {
    res = new int[]{0, 0, 0};   // a private class-member
    return FirebaseFirestore.getInstance().collection("Votes")
            .whereEqualTo("proposition_key", curr_proposition.getKey())
            .get().addOnCompleteListener(task -> {
                if (task.isSuccessful()) {
                    for (QueryDocumentSnapshot document : task.getResult()) {
                        String userChoice = (String) document.get("user_choice");
                        int choice;
                        switch (userChoice) {
                            case "against":
                                choice = 0;
                                break;
                            case "impossible":
                                choice = 1;
                                break;
                            case "agreement":
                                choice = 2;
                                break;
                            default:
                                throw new IllegalStateException("Unexpected value: " + userChoice);
                        }
                        res[choice]++;
                    }
                }
            });
}

一般来说，代码会从 Firestore 集合中读取一些行，并对它们应用一些“业务逻辑”（计算每种类型的字符串数）。未来，业务逻辑可能会变得比单纯的计数要复杂得多。所以我正在寻找一种方法来重构代码，以便可以将业务逻辑与数据库分开编写和测试。我想要的是具有某种形式的功能：

int[] countVotes(Generator<String> strings) {
     res = new int[3];
     (for String userChoice: strings)  {
          // Update res as above
     }
     return res;
}

可以在不需要数据库连接的情况下进行单元测试。那么，上面的函数可以重构如下：

private Generator<String> getVotesFromDB() {
    return FirebaseFirestore.getInstance().collection("Votes")
            .whereEqualTo("proposition_key", curr_proposition.getKey())
            .get().addOnCompleteListener(task -> {
                if (task.isSuccessful()) {
                    for (QueryDocumentSnapshot document : task.getResult()) {
                        userChoice = (String) document.get("user_choice");
                        yield userChoice;
                    }
                }
            });
}

然后运行类似的东西：

countVotes(getVotesFromDB())

问题是，我不知道如何使用异步函数调用来做到这一点。有没有办法以类似或更好的方式重构代码？

Answer 1

您可以将结果收集到一个数组中，然后使用单独的方法对其进行处理。然后，如果处理方法很复杂，则可以针对各种不同的结果列表单独进行单元测试。

private void getVotesFromDB() {
    FirebaseFirestore.getInstance().collection("Votes")
        .whereEqualTo("proposition_key", curr_proposition.getKey())
        .get().addOnCompleteListener(task -> {
            if (task.isSuccessful()) {
                ArrayList<String> results = new ArrayList();
                for (QueryDocumentSnapshot document : task.getResult()) {
                    String userChoice = (String) document.get("user_choice");
                    results.add(userChoice);
                }

                // assign the result to a class member
                res = countVotes(results);

                // do something to signal to the rest of the code
                // that results have been processed (e.g. post to
                // LiveData or call another "showResults" method)
            }
        });
}

然后，任何复杂的计数逻辑都可以独立于 firebase 调用。

int[] countVotes(ArrayList<String> choices) {
    int[] res = new int[]{0,0,0}; 

    // count the votes

    return res;
}