Merge pull request #278 from JensRantil/cassandra-auth

mattes · web-flow · commit 62cddb9e749b · 2017-09-18T10:08:58.000-07:00
feat(cassandra): support for user/pw authentication
diff --git a/database/cassandra/README.md b/database/cassandra/README.md
@@ -16,6 +16,8 @@ system_schema table which comes with 3.X
 | `consistency` | ALL | Migration consistency
 | `protocol` |  | Cassandra protocol version (3 or 4)
 | `timeout` | 1 minute | Migration timeout
+| `username` | nil | Username to use when authenticating. |
+| `password` | nil | Password to use when authenticating. |
 
 
 `timeout` is parsed using [time.ParseDuration(s string)](https://golang.org/pkg/time/#ParseDuration)
diff --git a/database/cassandra/cassandra.go b/database/cassandra/cassandra.go
@@ -65,6 +65,13 @@ func (p *Cassandra) Open(url string) (database.Driver, error) {
 	cluster.Consistency = gocql.All
 	cluster.Timeout = 1 * time.Minute
 
+	if len(u.Query().Get("username")) > 0 && len(u.Query().Get("password")) > 0 {
+		authenticator := gocql.PasswordAuthenticator{
+			Username: u.Query().Get("username"),
+			Password: u.Query().Get("password"),
+		}
+		cluster.Authenticator = authenticator
+	}
 
 	// Retrieve query string configuration
 	if len(u.Query().Get("consistency")) > 0 {
