Merge pull request #1490 from hilimd/master

Add cassandra authenticator mode
This commit is contained in:
Chris Lu 2020-09-24 21:30:20 -07:00 committed by GitHub
commit b1ac324c44
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -28,11 +28,16 @@ func (store *CassandraStore) Initialize(configuration util.Configuration, prefix
return store.initialize( return store.initialize(
configuration.GetString(prefix+"keyspace"), configuration.GetString(prefix+"keyspace"),
configuration.GetStringSlice(prefix+"hosts"), configuration.GetStringSlice(prefix+"hosts"),
configuration.GetString(prefix+"username"),
configuration.GetString(prefix+"password"),
) )
} }
func (store *CassandraStore) initialize(keyspace string, hosts []string) (err error) { func (store *CassandraStore) initialize(keyspace string, hosts []string, username string, password string) (err error) {
store.cluster = gocql.NewCluster(hosts...) store.cluster = gocql.NewCluster(hosts...)
if username != "" && password != "" {
store.cluster.Authenticator = gocql.PasswordAuthenticator{Username: username, Password: password}
}
store.cluster.Keyspace = keyspace store.cluster.Keyspace = keyspace
store.cluster.Consistency = gocql.LocalQuorum store.cluster.Consistency = gocql.LocalQuorum
store.session, err = store.cluster.CreateSession() store.session, err = store.cluster.CreateSession()