MediaStreamer でエラーを返却する場合にも同じようにリトライするように修正

Author: nobuo-kobayashi

dConnectDevicePlugin/dConnectDeviceHost/app/src/main/java/org/deviceconnect/android/deviceplugin/host/recorder/AbstractRTMPBroadcaster.java

@@ -103,6 +103,8 @@ public void onError(MediaEncoderException e) {
                 if (mOnBroadcasterEventListener != null) {
                     mOnBroadcasterEventListener.onError(e);
                 }
+
+                AbstractRTMPBroadcaster.this.stop();
             }
 
             @Override
@@ -111,7 +113,6 @@ public void onConnected() {
 
             @Override
             public void onDisconnected() {
-                AbstractRTMPBroadcaster.this.stop();
             }
 
             @Override

dConnectDevicePlugin/dConnectDeviceHost/app/src/main/java/org/deviceconnect/android/deviceplugin/host/recorder/AbstractSRTBroadcaster.java

@@ -103,6 +103,8 @@ public void onError(MediaEncoderException e) {
                 if (mOnBroadcasterEventListener != null) {
                     mOnBroadcasterEventListener.onError(e);
                 }
+
+                AbstractSRTBroadcaster.this.stop();
             }
 
             @Override
@@ -111,7 +113,6 @@ public void onConnected() {
 
             @Override
             public void onDisconnected() {
-                AbstractSRTBroadcaster.this.stop();
             }
 
             @Override

dConnectSDK/dConnectLibStreaming/libmedia/src/main/java/org/deviceconnect/android/libmedia/streaming/rtmp/RtmpClient.java

@@ -1,7 +1,7 @@
 package org.deviceconnect.android.libmedia.streaming.rtmp;
 
 import android.os.Handler;
-import android.os.Looper;
+import android.os.HandlerThread;
 
 import org.deviceconnect.android.libmedia.streaming.MediaEncoderException;
 import org.deviceconnect.android.libmedia.streaming.MediaStreamer;
@@ -23,7 +23,7 @@ public class RtmpClient {
     /**
      * ハンドラ.
      */
-    private final Handler mHandler = new Handler(Looper.getMainLooper());
+    private Handler mRetryHandler;
 
     /**
      * リトライ数.
@@ -33,12 +33,12 @@ public class RtmpClient {
     /**
      * リトライ回数.
      */
-    private int mMaxRetryCount = 3;
+    private int mMaxRetryCount = 0;
 
     /**
      * リトライを行うインターバル.
      */
-    private int mRetryInterval = 2000;
+    private int mRetryInterval = 4000;
 
     /**
      * リトライフラグ.
@@ -49,16 +49,100 @@ public class RtmpClient {
     private boolean mRetryFlag;
 
     /**
-     * 起動フラグ.
+     * イベントを送信するリスナー.
      */
-    private boolean mRunnableFlag;
+    private OnEventListener mOnEventListener;
 
     /**
      * コンストラクタ.
      */
     public RtmpClient(String broadcastURI) {
         mRtmpMuxer = new RtmpMuxer(broadcastURI);
+        mRtmpMuxer.setOnEventListener(new RtmpMuxer.OnEventListener() {
+            @Override
+            public void onConnected() {
+                if (mOnEventListener != null) {
+                    mOnEventListener.onConnected();
+                }
+            }
+
+            @Override
+            public void onDisconnected() {
+                if (mOnEventListener != null) {
+                    mOnEventListener.onDisconnected();
+                }
+            }
+
+            @Override
+            public void onNewBitrate(long bitrate) {
+                if (mOnEventListener != null) {
+                    mOnEventListener.onNewBitrate(bitrate);
+                }
+            }
+
+            @Override
+            public void onError(MediaEncoderException e) {
+                error(e);
+            }
+        });
+
         mMediaStreamer = new MediaStreamer(mRtmpMuxer);
+        mMediaStreamer.setOnEventListener(new MediaStreamer.OnEventListener() {
+            @Override
+            public void onStarted() {
+                mRetryCount = 0;
+
+                if (mOnEventListener != null) {
+                    mOnEventListener.onStarted();
+                }
+            }
+
+            @Override
+            public void onStopped() {
+                if (mOnEventListener != null) {
+                    mOnEventListener.onStopped();
+                }
+            }
+
+            @Override
+            public void onError(MediaEncoderException e) {
+                error(e);
+            }
+        });
+    }
+
+    /**
+     * エラー処理を行います.
+     *
+     * @param e エラー原因の例外
+     */
+    private synchronized void error(MediaEncoderException e) {
+        if (mRetryFlag) {
+            // リトライ処理中なので、連続で発生したエラーは無視する。
+            return;
+        }
+
+        if (mRetryCount < mMaxRetryCount) {
+            mRetryCount++;
+            mRetryFlag = true;
+
+            mMediaStreamer.stop();
+
+            if (mRetryHandler != null) {
+                mRetryHandler.postDelayed(() -> {
+                    if (mRetryHandler != null) {
+                        mMediaStreamer.start();
+                    }
+                    mRetryFlag = false;
+                }, mRetryInterval);
+            }
+        } else {
+            stop();
+
+            if (mOnEventListener != null) {
+                mOnEventListener.onError(e);
+            }
+        }
     }
 
     /**
@@ -103,63 +187,7 @@ public int getRetryInterval() {
      * @param listener リスナー
      */
     public void setOnEventListener(OnEventListener listener) {
-        mMediaStreamer.setOnEventListener(listener);
-        mRtmpMuxer.setOnEventListener(new RtmpMuxer.OnEventListener() {
-            @Override
-            public void onConnected() {
-                // 接続できたのでリトライ数を初期化
-                mRetryCount = 0;
-                if (listener != null) {
-                    listener.onConnected();
-                }
-            }
-
-            @Override
-            public void onDisconnected() {
-                if (mRetryFlag) {
-                    // エラーが発生して切断された時も、このイベントが呼び出されるので
-                    // リトライの処理を行っている間はリスナーに通知しないようにブロックする。
-                    return;
-                }
-
-                if (listener != null) {
-                    listener.onDisconnected();
-                }
-            }
-
-            @Override
-            public void onNewBitrate(long bitrate) {
-                if (listener != null) {
-                    listener.onNewBitrate(bitrate);
-                }
-            }
-
-            @Override
-            public void onError(MediaEncoderException e) {
-                if (mRetryFlag) {
-                    return;
-                }
-
-                mMediaStreamer.stop();
-
-                if (mRetryCount < mMaxRetryCount) {
-                    mRetryCount++;
-                    mRetryFlag = true;
-
-                    mHandler.postDelayed(() -> {
-                        mRetryFlag = false;
-                        if (mRunnableFlag) {
-                            mMediaStreamer.start();
-                        }
-                    }, mRetryInterval);
-                } else {
-                    mRunnableFlag = false;
-                    if (listener != null) {
-                        listener.onError(e);
-                    }
-                }
-            }
-        });
+        mOnEventListener = listener;
     }
 
     /**
@@ -168,23 +196,33 @@ public void onError(MediaEncoderException e) {
      * @return 動作中の場合は true、それ以外は false
      */
     public boolean isRunning() {
-        return mMediaStreamer.isRunning();
+        return mRetryHandler != null;
     }
 
     /**
      * RTMP のセッションを開始します.
      */
     public synchronized void start() {
+        if (mRetryHandler != null) {
+            return;
+        }
+
+        HandlerThread thread = new HandlerThread("rtmp-retry-thread");
+        thread.start();
+        mRetryHandler = new Handler(thread.getLooper());
         mRetryCount = mMaxRetryCount;
-        mRunnableFlag = true;
+        mRetryFlag = false;
         mMediaStreamer.start();
     }
 
     /**
      * RTMP のセッションを停止します.
      */
     public synchronized void stop() {
-        mRunnableFlag = false;
+        if (mRetryHandler != null) {
+            mRetryHandler.getLooper().quit();
+            mRetryHandler = null;
+        }
         mMediaStreamer.stop();
     }
 
@@ -283,6 +321,9 @@ public void restartAudioEncoder() {
         }
     }
 
+    /**
+     * RtmpClient で発生したイベントを通知するリスナー.
+     */
     public interface OnEventListener extends MediaStreamer.OnEventListener, RtmpMuxer.OnEventListener {
     }
 }