如何判断启动activityMain的按钮来源，并据此从数据库传递对应数据？

嘿，这个需求其实挺普遍的，核心思路就是给每个启动按钮打上唯一的标识，在MainActivity里根据标识去数据库拉对应的数据，我给你拆解成具体步骤和代码示例：

第一步：给每个按钮添加带标识的启动逻辑 #

在每个发起启动的Activity里，给不同按钮的点击事件中，通过Intent的putExtra方法添加一个“来源标识”，用来区分是哪个按钮触发的跳转。

比如在ActivityA里有两个按钮：

// ActivityA中的按钮1点击事件
btnFromA1.setOnClickListener(v -> {
    Intent goToMain = new Intent(ActivityA.this, MainActivity.class);
    // 用自定义的常量作为key，传递唯一标识字符串
    goToMain.putExtra(MainActivity.EXTRA_BUTTON_SOURCE, "A_BUTTON_1");
    startActivity(goToMain);
});

// ActivityA中的按钮2点击事件
btnFromA2.setOnClickListener(v -> {
    Intent goToMain = new Intent(ActivityA.this, MainActivity.class);
    goToMain.putExtra(MainActivity.EXTRA_BUTTON_SOURCE, "A_BUTTON_2");
    startActivity(goToMain);
});

再比如ActivityB里的按钮：

// ActivityB中的按钮点击事件
btnFromB.setOnClickListener(v -> {
    Intent goToMain = new Intent(ActivityB.this, MainActivity.class);
    goToMain.putExtra(MainActivity.EXTRA_BUTTON_SOURCE, "B_BUTTON_1");
    startActivity(goToMain);
});

第二步：在MainActivity中接收标识并处理 #

在MainActivity里，我们需要从Intent中取出这个标识，然后根据不同标识去数据库查询对应的数据。

1. 定义常量和接收参数 #

public class MainActivity extends AppCompatActivity {
    // 定义一个全局常量作为Extra的key，避免拼写错误
    public static final String EXTRA_BUTTON_SOURCE = "extra_button_source";

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        // 获取传递过来的来源标识
        String buttonSource = getIntent().getStringExtra(EXTRA_BUTTON_SOURCE);
        if (buttonSource != null) {
            // 根据标识加载对应数据
            loadTargetData(buttonSource);
        }
    }

2. 处理重复启动的情况（可选） #

如果你的MainActivity设置了singleTop或singleTask这类启动模式，当它已经在栈顶时，再次启动不会走onCreate，而是走onNewIntent，所以需要补充这个方法的逻辑：

@Override
    protected void onNewIntent(Intent intent) {
        super.onNewIntent(intent);
        // 更新当前的Intent，确保后续能拿到最新的参数
        setIntent(intent);
        String buttonSource = getIntent().getStringExtra(EXTRA_BUTTON_SOURCE);
        if (buttonSource != null) {
            loadTargetData(buttonSource);
        }
    }

3. 根据标识查询数据库 #

private void loadTargetData(String buttonSource) {
        // 根据不同的按钮标识，执行对应的数据库查询逻辑
        switch (buttonSource) {
            case "A_BUTTON_1":
                // 查询按钮A1对应的数据，比如：
                // List<Data> data = dbHelper.queryDataByCondition("condition_for_a1");
                updateUIWithData(data);
                break;
            case "A_BUTTON_2":
                // 查询按钮A2对应的数据
                break;
            case "B_BUTTON_1":
                // 查询按钮B1对应的数据
                break;
            default:
                // 处理未知来源的情况，比如加载默认数据
                break;
        }
    }

    private void updateUIWithData(List<Data> data) {
        // 把查询到的数据展示到UI上
    }
}

优化建议：用枚举替代字符串（更安全） #

为了避免字符串拼写错误导致的bug，建议用枚举类来定义所有的按钮来源：

// 定义一个枚举类，包含所有可能的按钮来源
public enum ButtonSource {
    ACTIVITY_A_BUTTON1,
    ACTIVITY_A_BUTTON2,
    ACTIVITY_B_BUTTON1
}

然后传递参数时用枚举的name()方法：

goToMain.putExtra(MainActivity.EXTRA_BUTTON_SOURCE, ButtonSource.ACTIVITY_A_BUTTON1.name());

接收和转换时用Enum.valueOf()：

String sourceStr = getIntent().getStringExtra(EXTRA_BUTTON_SOURCE);
if (sourceStr != null) {
    ButtonSource source = ButtonSource.valueOf(sourceStr);
    loadTargetData(source);
}

// 对应的loadTargetData方法改成接收枚举参数
private void loadTargetData(ButtonSource source) {
    switch (source) {
        case ACTIVITY_A_BUTTON1:
            // 处理逻辑
            break;
        // ...其他case
    }
}

这样能在编译阶段就发现拼写错误，比直接用字符串更可靠～

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