max_retry 回数までリトライするように変更する

amazon_transcribe_handler.go

@@ -163,7 +163,7 @@ func (h *AmazonTranscribeHandler) Handle(ctx context.Context, reader io.Reader)
 					Send()
 
 				// リトライしない設定の場合はクライアントにエラーを返し、再度接続するかはクライアント側で判断する
-				if !*at.Config.Retry {
+				if *at.Config.MaxRetry < 1 {
 					if err := encoder.Encode(NewSuzuErrorResponse(err)); err != nil {
 						zlog.Error().
 							Err(err).

handler.go

@@ -143,8 +143,8 @@ func (s *Server) createSpeechHandler(serviceType string, onResultFunc func(conte
 					Str("connection_id", h.SoraConnectionID).
 					Send()
 				if err, ok := err.(*SuzuError); ok {
-					if *s.config.Retry {
-						if err.IsRetry() {
+					if err.IsRetry() {
+						if *s.config.MaxRetry > retryCount {
 							retryCount += 1
 
 							zlog.Debug().
@@ -184,7 +184,7 @@ func (s *Server) createSpeechHandler(serviceType string, onResultFunc func(conte
 							Send()
 						return err
 					} else if errors.Is(err, ErrServerDisconnected) {
-						if *s.config.Retry {
+						if *s.config.MaxRetry > retryCount {
 							// サーバから切断されたが再度接続できる可能性があるため、接続を試みる
 							retryCount += 1
 
@@ -215,6 +215,9 @@ func (s *Server) createSpeechHandler(serviceType string, onResultFunc func(conte
 					return err
 				}
 
+				// 1 度でも接続結果を受け取れた場合はリトライ回数をリセットする
+				retryCount = 0
+
 				// メッセージが空でない場合はクライアントに結果を送信する
 				if n > 0 {
 					if _, err := c.Response().Write(buf[:n]); err != nil {

speech_to_text_handler.go

@@ -142,7 +142,7 @@ func (h *SpeechToTextHandler) Handle(ctx context.Context, reader io.Reader) (*io
 						Send()
 
 					// リトライしない設定の場合はクライアントにエラーを返し、再度接続するかはクライアント側で判断する
-					if !*stt.Config.Retry {
+					if *stt.Config.MaxRetry < 1 {
 						if err := encoder.Encode(NewSuzuErrorResponse(err)); err != nil {
 							zlog.Error().
 								Err(err).