Firebase putFile：上传按钮禁用与启用的时机探讨

这个场景我之前做项目时也碰到过，咱们一步步梳理清楚：

首先回应你关于按钮重新启用位置的疑问：
你说得很对，OnCompleteListener确实存在触发盲区——比如设备彻底离线时，Firebase的UploadTask会进入本地重试队列，此时任务不会立即进入"完成"状态，onComplete自然不会被调用，这就会导致按钮一直锁死，用户体验很差。

所以正确的按钮启用逻辑必须覆盖所有任务终止的场景：

• 上传成功（OnSuccessListener）
• 上传失败（OnFailureListener）
• 任务被主动取消（如果有取消功能的话）

你可以把恢复UI的逻辑抽成单独方法，避免重复代码，改造后的上传方法如下：

private void uploadFile() {
    if (mImageUri != null) {
        StorageReference sR = mStorageRef.child(System.currentTimeMillis() + "." + getFileExtension(mImageUri));
        mButtonUpload.setClickable(false);
        sR.putFile(mImageUri)
                .addOnSuccessListener(new OnSuccessListener<UploadTask.TaskSnapshot>() {
                    @Override
                    public void onSuccess(UploadTask.TaskSnapshot taskSnapshot) {
                        Toast.makeText(MainActivity.this, "Upload successful", Toast.LENGTH_SHORT).show();
                        Upload upload = new Upload(mEditTextFileName.getText().toString().trim(), taskSnapshot.getDownloadUrl().toString());
                        String uploadId = mDatabaseRef.push().getKey();
                        mDatabaseRef.child(uploadId).setValue(upload);
                        
                        // 上传成功后恢复UI
                        resetUploadUI();
                    }
                })
                .addOnFailureListener(new OnFailureListener() {
                    @Override
                    public void onFailure(@NonNull Exception e) {
                        Toast.makeText(MainActivity.this, e.getMessage(), Toast.LENGTH_SHORT).show();
                        
                        // 上传失败后恢复UI
                        resetUploadUI();
                    }
                })
                .addOnProgressListener(new OnProgressListener<UploadTask.TaskSnapshot>() {
                    @Override
                    public void onProgress(UploadTask.TaskSnapshot taskSnapshot) {
                        double progress = (100.0 * taskSnapshot.getBytesTransferred()) / taskSnapshot.getTotalByteCount();
                        mProgressBar.setProgress((int) progress);
                    }
                });
    } else {
        Toast.makeText(MainActivity.this, "No file selected", Toast.LENGTH_SHORT).show();
    }
}

// 抽成单独方法，统一处理UI重置
private void resetUploadUI() {
    mButtonUpload.setClickable(true);
    Handler handler = new Handler();
    handler.postDelayed(() -> mProgressBar.setProgress(0), 500);
}

接下来聊聊你提到的用成员变量存储UploadTask并判断任务状态的方案——这不仅可行，还是更推荐的进阶做法！

因为Firebase的UploadTask会在离线时自动加入重试队列，哪怕你恢复了按钮的可点击状态，用户再次点击还是会触发新的上传任务，导致重复上传。而通过成员变量跟踪任务状态，能从源头阻止重复操作：

private UploadTask mUploadTask;

// 初始化按钮点击事件
mButtonUpload.setOnClickListener(v -> {
    if (mUploadTask != null && mUploadTask.isInProgress()) {
        Toast.makeText(MainActivity.this, "Upload in progress", Toast.LENGTH_SHORT).show();
    } else {
        uploadFile();
    }
});

// 修改uploadFile方法，绑定任务到成员变量
private void uploadFile() {
    if (mImageUri != null) {
        StorageReference sR = mStorageRef.child(System.currentTimeMillis() + "." + getFileExtension(mImageUri));
        // 将任务实例赋值给成员变量
        mUploadTask = sR.putFile(mImageUri);
        mButtonUpload.setClickable(false);
        
        mUploadTask.addOnSuccessListener(taskSnapshot -> {
            Toast.makeText(MainActivity.this, "Upload successful", Toast.LENGTH_SHORT).show();
            Upload upload = new Upload(mEditTextFileName.getText().toString().trim(), taskSnapshot.getDownloadUrl().toString());
            String uploadId = mDatabaseRef.push().getKey();
            mDatabaseRef.child(uploadId).setValue(upload);
            
            resetUploadUI();
            // 任务完成后清空成员变量，不影响下一次上传
            mUploadTask = null;
        })
        .addOnFailureListener(e -> {
            Toast.makeText(MainActivity.this, e.getMessage(), Toast.LENGTH_SHORT).show();
            resetUploadUI();
            // 任务失败后同样清空成员变量
            mUploadTask = null;
        })
        .addOnProgressListener(taskSnapshot -> {
            double progress = (100.0 * taskSnapshot.getBytesTransferred()) / taskSnapshot.getTotalByteCount();
            mProgressBar.setProgress((int) progress);
        });
    } else {
        Toast.makeText(MainActivity.this, "No file selected", Toast.LENGTH_SHORT).show();
    }
}

这种方案的优势很明显：

• 不管任务是正在上传、等待重试还是已经结束，都能准确判断状态，从源头避免重复触发
• 任务完成（成功/失败）后清空成员变量，不影响下一次正常上传
• 结合按钮的setClickable(false)，形成双重保障，彻底杜绝重复点击问题

最后补充一点：如果你的App支持主动取消上传的功能，记得在取消操作时也要调用resetUploadUI()并清空mUploadTask，确保状态始终准确。

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