mirror of
https://github.com/mainflux/mainflux.git
synced 2025-04-27 13:48:49 +08:00
5202d734c9
* Add username, password and port env var Signed-off-by: Aleksandar Novakovic <anovakovic01@gmail.com> * Update existing readme files with new env vars Signed-off-by: Aleksandar Novakovic <anovakovic01@gmail.com>
36 lines
699 B
Go
36 lines
699 B
Go
//
|
|
// Copyright (c) 2018
|
|
// Mainflux
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package cassandra
|
|
|
|
import (
|
|
"github.com/gocql/gocql"
|
|
)
|
|
|
|
// DBConfig contains Cassandra DB specific parameters.
|
|
type DBConfig struct {
|
|
Hosts []string
|
|
Keyspace string
|
|
Username string
|
|
Password string
|
|
Port int
|
|
}
|
|
|
|
// Connect establishes connection to the Cassandra cluster.
|
|
func Connect(cfg DBConfig) (*gocql.Session, error) {
|
|
cluster := gocql.NewCluster(cfg.Hosts...)
|
|
cluster.Keyspace = cfg.Keyspace
|
|
cluster.Consistency = gocql.Quorum
|
|
cluster.Authenticator = gocql.PasswordAuthenticator{
|
|
Username: cfg.Username,
|
|
Password: cfg.Password,
|
|
}
|
|
cluster.Port = cfg.Port
|
|
|
|
return cluster.CreateSession()
|
|
}
|