From dc4eadbb371a38cb0352630f229ff4b4e989483d Mon Sep 17 00:00:00 2001
From: "Christian Mauduit (DataDog)" <christian.mauduit@datadoghq.com>
Date: Wed, 19 Jun 2024 19:13:25 +0200
Subject: [PATCH] [contrib/gocql] adding support for MapScanCAS in gocql
 (cassandra)

---
 contrib/gocql/gocql/gocql.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/contrib/gocql/gocql/gocql.go b/contrib/gocql/gocql/gocql.go
index 7881392ba4..fd39fe454e 100644
--- a/contrib/gocql/gocql/gocql.go
+++ b/contrib/gocql/gocql/gocql.go
@@ -205,6 +205,14 @@ func (tq *Query) MapScan(m map[string]interface{}) error {
 	return err
 }
 
+// MapScanCAS wraps in a span query.MapScanCAS call.
+func (tq *Query) MapScanCAS(m map[string]interface{}) (applied bool, err error) {
+	span := tq.newChildSpan(tq.ctx)
+	applied, err = tq.Query.MapScanCAS(m)
+	tq.finishSpan(span, err)
+	return applied, err
+}
+
 // Scan wraps in a span query.Scan call.
 func (tq *Query) Scan(dest ...interface{}) error {
 	span := tq.newChildSpan(tq.ctx)