// New db adaptor
func NewAdaptor(urls []string, username, password string) *Adaptor {
	cluster := gocql.NewCluster(urls...)
	cluster.Keyspace = "events"
	cluster.ProtoVersion = 3

	// cluster.Authenticator = gocql.PasswordAuthenticator{
	// 	Username: username,
	// 	Password: password,
	// }

	session, err := cluster.CreateSession()
	if err != nil {
		log.Fatalf("Cassandra conn error: %s", err.Error())
	}
	qe := gocassa.GoCQLSessionToQueryExecutor(session)
	conn := gocassa.NewConnection(qe)
	gocassaSession := conn.KeySpace("events")

	return &Adaptor{
		Session:            gocassaSession,
		urls:               urls,
		eventTimeDateRange: 7 * 24 * time.Hour,
		Username:           username,
		Password:           password,
	}
}
func (a *Adaptor) ReestablishConnection() {
	log.Println("Reestablishing cql connection")
	cluster := gocql.NewCluster(a.urls[0])
	cluster.ProtoVersion = 3
	cluster.Authenticator = gocql.PasswordAuthenticator{
		Username: a.Username,
		Password: a.Password,
	}
	session, err := cluster.CreateSession()
	if err != nil {
		log.Fatalf("Cassandra conn error: %s", err.Error())
	}
	qe := gocassa.GoCQLSessionToQueryExecutor(session)
	a.Session = gocassa.NewConnection(qe).KeySpace("events")
}
示例#3
0
func gocqlConnector(ks string) gocassa.Connection {
	ksConnectionsMtx.RLock()
	conn, ok := ksConnections[ks]
	ksConnectionsMtx.RUnlock()
	if ok {
		return conn
	}

	ksConnectionsMtx.Lock()
	defer ksConnectionsMtx.Unlock()
	if conn, ok = ksConnections[ks]; !ok { // Guard against race
		conn = gocassa.NewConnection(&gocqlExecutor{
			ks: ks,
		})
		ksConnections[ks] = conn
	}
	return conn
}