这很简单
首先，您需要监听充电状态的变化。这可以通过以下方式完成：

//Create receiver
BroadcastReceiver powerChangeReceiver = new BroadcastReceiver() {
    @Override
    public void onReceive(Context context, Intent intent) {
        String action = intent.getAction();
        
        if(Intent.ACTION_POWER_CONNECTED.equals(action)){
            turnScreenOn();
        } else if(Intent.ACTION_POWER_DISCONNECTED.equals(action)){
            turnScreenOff();
        }

    }
};

//Register receiver
IntentFilter i = new IntentFilter();
i.addAction(Intent.ACTION_POWER_CONNECTED);
i.addAction(Intent.ACTION_POWER_DISCONNECTED);
registerReceiver(powerChangeReceiver, i);

现在您需要做的就是实现turnScreenOn()和turnScreenOff()方法：

public void turnScreenOn(){
    runOnUiThread(() -> {
        PowerManager pm = (PowerManager) getApplicationContext().getSystemService(POWER_SERVICE);
        @SuppressWarnings("deprecation")
        PowerManager.WakeLock wl = pm.newWakeLock(PowerManager.SCREEN_BRIGHT_WAKE_LOCK |
                PowerManager.ACQUIRE_CAUSES_WAKEUP, "example:tag");
        wl.acquire();
    });
}

public void turnScreenOff(){
    if(policyManager.isDeviceOwnerApp(getPackageName())) {
        runOnUiThread(() -> {
            policyManager.lockNow();
        });
    }
}

不要忘记向清单添加WAKE_LOCK权限：

<uses-permission android:name="android.permission.WAKE_LOCK" />

希望能有所帮助！