fix: stop subscribe query runner after client disconnect

tigrisdata-archive · Aug 17, 2022 · cb61b7d · cb61b7d
1 parent 872adf7
commit cb61b7d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/services/v1/query_runner.go b/server/services/v1/query_runner.go
@@ -834,6 +834,11 @@ func (runner *SubscribeQueryRunner) Run(ctx context.Context, tx transaction.Tx,
 		if ep.Code != 1004 && ep.Code != 1031 && ulog.E(err) {
 			return nil, ctx, err
 		}
+
+		// check for client disconnect
+		if runner.streaming.Context().Err() != nil {
+			break
+		}
 	}
 
 	return &Response{}, ctx, nil