Add Purge method for cassandra (jaegertracing#5414)

## Which problem is this PR solving?
-
jaegertracing#5398 (comment)

## Description of the changes
- added purge for method for cassandra

## How was this change tested?
- via integration tests

## Checklist
- [x] I have read
https://github.com/jaegertracing/jaeger/blob/master/CONTRIBUTING_GUIDELINES.md
- [x] I have signed all commits
- [x] I have added unit tests for the new functionality
- [x] I have run lint and test steps successfully
  - for `jaeger`: `make lint test`
  - for `jaeger-ui`: `yarn lint` and `yarn test`

---------

Signed-off-by: Harshvir Potpose <[email protected]>
Signed-off-by: Yuri Shkuro <[email protected]>
Co-authored-by: Yuri Shkuro <[email protected]>

Author: 2 people

plugin/storage/cassandra/factory.go

@@ -16,6 +16,7 @@
 package cassandra
 
 import (
+	"context"
 	"errors"
 	"flag"
 	"io"
@@ -251,3 +252,7 @@ func (f *Factory) Close() error {
 func (f *Factory) PrimarySession() cassandra.Session {
 	return f.primarySession
 }
+
+func (f *Factory) Purge(_ context.Context) error {
+	return f.primarySession.Query("TRUNCATE traces").Exec()
+}

plugin/storage/integration/cassandra_test.go

@@ -16,13 +16,13 @@
 package integration
 
 import (
+	"context"
 	"testing"
 
 	"github.com/stretchr/testify/require"
 	"go.uber.org/zap"
 	"go.uber.org/zap/zaptest"
 
-	dbsession "github.com/jaegertracing/jaeger/pkg/cassandra"
 	"github.com/jaegertracing/jaeger/pkg/config"
 	"github.com/jaegertracing/jaeger/pkg/metrics"
 	"github.com/jaegertracing/jaeger/plugin/storage/cassandra"
@@ -31,8 +31,7 @@ import (
 
 type CassandraStorageIntegration struct {
 	StorageIntegration
-
-	session dbsession.Session
+	factory *cassandra.Factory
 }
 
 func newCassandraStorageIntegration() *CassandraStorageIntegration {
@@ -58,7 +57,7 @@ func newCassandraStorageIntegration() *CassandraStorageIntegration {
 }
 
 func (s *CassandraStorageIntegration) cleanUp(t *testing.T) {
-	require.NoError(t, s.session.Query("TRUNCATE traces").Exec())
+	require.NoError(t, s.factory.Purge(context.Background()))
 }
 
 func (s *CassandraStorageIntegration) initializeCassandraFactory(t *testing.T, flags []string) *cassandra.Factory {
@@ -75,7 +74,7 @@ func (s *CassandraStorageIntegration) initializeCassandra(t *testing.T) {
 	f := s.initializeCassandraFactory(t, []string{
 		"--cassandra.keyspace=jaeger_v1_dc1",
 	})
-	s.session = f.PrimarySession()
+	s.factory = f
 	var err error
 	s.SpanWriter, err = f.CreateSpanWriter()
 	require.NoError(t, err)