如何正确地通过DialogFragment传递外部视图的点击？

public class MainActivity extends AppCompatActivity implements OnViewClickListener {

    // ...

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

        // ...

        // 设置需要传递点击事件的视图的点击监听器
        Button button = findViewById(R.id.button);
        button.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                if (mOnViewClickListener != null) {
                    mOnViewClickListener.onViewClick(v);
                }
            }
        });
    }

    // 实现OnViewClickListener接口的onViewClick()方法，根据被点击的视图进行处理
    @Override
    public void onViewClick(View view) {
        if (view.getId() == R.id.button) {
            // 处理按钮点击事件
        }
    }

    // ...
}

public class MyDialogFragment extends DialogFragment {

    private OnViewClickListener mOnViewClickListener;

    // ...

    @Override
    public void onAttach(@NonNull Context context) {
        super.onAttach(context);

        // 将外部视图的点击监听器赋值给DialogFragment的成员变量
        if (context instanceof OnViewClickListener) {
            mOnViewClickListener = (OnViewClickListener) context;
        }
    }

    @Nullable
    @Override
    public View onCreateView(@NonNull LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) {
        View rootView = inflater.inflate(R.layout.fragment_dialog, container, false);

        // 找到需要接收点击事件的视图，并为其设置点击监听器
        Button button = rootView.findViewById(R.id.button);
        button.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                if (mOnViewClickListener != null) {
                    mOnViewClickListener.onViewClick(v);
                }
            }
        });

        return rootView;
    }

    // ...
}