Merge pull request scylladb#366 from Zariel/batch-consistency

Inherit consistency from session in NewBatch
guangminglion · Apr 15, 2015 · 1ff8661 · 1ff8661
2 parents 75dba14 + cd7d08d
commit 1ff8661
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/session.go b/session.go
@@ -734,7 +734,10 @@ func NewBatch(typ BatchType) *Batch {
 
 // NewBatch creates a new batch operation using defaults defined in the cluster
 func (s *Session) NewBatch(typ BatchType) *Batch {
-	return &Batch{Type: typ, rt: s.cfg.RetryPolicy, serialCons: s.cfg.SerialConsistency}
+	s.mu.RLock()
+	batch := &Batch{Type: typ, rt: s.cfg.RetryPolicy, serialCons: s.cfg.SerialConsistency, Cons: s.cons}
+	s.mu.RUnlock()
+	return batch
 }
 
 // Attempts returns the number of attempts made to execute the batch.