Index: dm/tools/dmtool/vizQuery.go |
diff --git a/dm/tools/dmtool/vizQuery.go b/dm/tools/dmtool/vizQuery.go |
index e12f431a191616688dbab79be920f700dd04ef04..6699430e0a001857fd36f1dae8fcc061071be04d 100644 |
--- a/dm/tools/dmtool/vizQuery.go |
+++ b/dm/tools/dmtool/vizQuery.go |
@@ -234,7 +234,7 @@ func (r *visQueryRun) Run(a subcommands.Application, args []string, env subcomma |
for c.Err() == nil { |
gdata, err := runQuery(c, dc, query) |
if err != nil { |
- if errors.Contains(err, context.Canceled) { |
+ if errors.Any(err, func(err error) bool { return err == context.Canceled }) { |
return 0 |
} |
logging.WithError(err).Errorf(c, "error running query") |