CASSGO-103: Expose GetSerialConsistency on Query and Batch

Return the serial consistency as *Consistency so callers can detect "not set" (nil) and avoid using the zero value ANY for serial operations, which is invalid and can cause panics.

patch by tejaslodayadd; reviewed by millerjp, joao-r-reis for CASSGO-103.

Author: tejaslodayadd

session.go

@@ -1272,6 +1272,15 @@ func (q *Query) SerialConsistency(cons Consistency) *Query {
 	return q
 }
 
+// GetSerialConsistency returns the currently configured serial consistency level
+// for the query. If the serial consistency level has not been set, it returns nil.
+func (q *Query) GetSerialConsistency() *Consistency {
+	if !q.serialCons.isSerial() {
+		return nil
+	}
+	return &q.serialCons
+}
+
 // PageState sets the paging state for the query to resume paging from a specific
 // point in time. Setting this will disable to query paging for this query, and
 // must be used for all subsequent pages.
@@ -2079,6 +2088,15 @@ func (b *Batch) SerialConsistency(cons Consistency) *Batch {
 	return b
 }
 
+// GetSerialConsistency returns the currently configured serial consistency level
+// for the batch. If the serial consistency level has not been set, it returns nil.
+func (b *Batch) GetSerialConsistency() *Consistency {
+	if !b.serialCons.isSerial() {
+		return nil
+	}
+	return &b.serialCons
+}
+
 // DefaultTimestamp will enable the with default timestamp flag on the query.
 // If enable, this will replace the server side assigned
 // timestamp as default timestamp. Note that a timestamp in the query itself

session_test.go

@@ -108,6 +108,15 @@ func TestQueryBasicAPI(t *testing.T) {
 		t.Fatalf("expected Query.GetConsistency to return 'All', got '%s'", qry.GetConsistency())
 	}
 
+	if sc := qry.GetSerialConsistency(); sc != nil {
+		t.Fatalf("expected Query.GetSerialConsistency to return nil when not set, got '%s'", sc)
+	}
+
+	qry.SerialConsistency(Serial)
+	if sc := qry.GetSerialConsistency(); sc == nil || *sc != Serial {
+		t.Fatalf("expected Query.GetSerialConsistency to return 'Serial', got '%v'", sc)
+	}
+
 	trace := &traceWriter{}
 	qry.Trace(trace)
 	if qry.trace != trace {
@@ -194,6 +203,16 @@ func TestBatchBasicAPI(t *testing.T) {
 		t.Fatalf("expected batch.GetConsistency() to return 'One', got '%s'", b.GetConsistency())
 	}
 
+	// Test Serial Consistency
+	if sc := b.GetSerialConsistency(); sc != nil {
+		t.Fatalf("expected batch.GetSerialConsistency() to return nil when not set, got '%s'", sc)
+	}
+
+	b.SerialConsistency(Serial)
+	if sc := b.GetSerialConsistency(); sc == nil || *sc != Serial {
+		t.Fatalf("expected batch.GetSerialConsistency() to return 'Serial', got '%v'", sc)
+	}
+
 	trace := &traceWriter{}
 	b.Trace(trace)
 	if b.trace != trace {