Hide floating rotation button upon click

Fixes: 132201048
Test: manual
Change-Id: I8ef65a110e863fbcc752e04b728fdd1e74854f1f

1 files changed, 4 insertions, 1 deletions

diff --git a/packages/SystemUI/src/com/android/systemui/statusbar/phone/FloatingRotationButton.java b/packages/SystemUI/src/com/android/systemui/statusbar/phone/FloatingRotationButton.java
index 6286d4d7febf..a79b6251a3d0 100644
--- a/packages/SystemUI/src/com/android/systemui/statusbar/phone/FloatingRotationButton.java
+++ b/packages/SystemUI/src/com/android/systemui/statusbar/phone/FloatingRotationButton.java
@@ -136,7 +136,10 @@ public class FloatingRotationButton implements RotationButton {
 
     @Override
     public void setOnClickListener(View.OnClickListener onClickListener) {
-        mKeyButtonView.setOnClickListener(onClickListener);
+        mKeyButtonView.setOnClickListener(view -> {
+            hide();
+            onClickListener.onClick(view);
+        });
     }
 
     @Override