github.com/hellofresh/janus@v0.0.0-20230925145208-ce8de8183c67/cassandra/wrapper/sessionretry.go (about)

     1  package wrapper
     2  
     3  import (
     4  	"github.com/gocql/gocql"
     5  	log "github.com/sirupsen/logrus"
     6  )
     7  
     8  // sessionRetry is an implementation of SessionInterface
     9  type sessionRetry struct {
    10  	goCqlSession *gocql.Session
    11  }
    12  
    13  // Query wrapper to be able to return our own QueryInterface
    14  func (s sessionRetry) Query( stmt string, values ...interface{}) QueryInterface {
    15  	log.Debug("running SessionRetry Query() method")
    16  
    17  	return queryRetry{goCqlQuery: s.goCqlSession.Query(stmt, values...)}
    18  }
    19  
    20  // Close wrapper to be able to run goCql method
    21  func (s sessionRetry) Close() {
    22  	log.Debug("running SessionRetry Close() method")
    23  
    24  	s.goCqlSession.Close()
    25  }