Use span.update_context to avoid DroppedSpan issues in Elasticsearch instrumentation (#1690)

* Use span.update_context to avoid DroppedSpan issues

* CHANGELOG

Author: basepi

CHANGELOG.asciidoc

@@ -29,6 +29,18 @@ endif::[]
 //===== Bug fixes
 //
 
+=== Unreleased
+
+// Unreleased changes go here
+// When the next release happens, nest these changes under the "Python Agent version 6.x" heading
+//[float]
+//===== Features
+//
+[float]
+===== Bug fixes
+
+* Fix error in Elasticsearch instrumentation when spans are dropped {pull}1690[#1690]
+
 [[release-notes-6.x]]
 === Python Agent version 6.x
 

elasticapm/instrumentation/packages/elasticsearch.py

@@ -163,7 +163,7 @@ def call(self, module, method, wrapped, instance, args, kwargs):
 
             hits = self._get_hits(result_data)
             if hits:
-                span.context["db"]["rows_affected"] = hits
+                span.update_context("db", {"rows_affected": hits})
 
             return result_data