update cassandra connections

update cassandra connections
This commit is contained in:
eshujiushiwo 2016-11-28 11:27:40 +08:00 committed by GitHub
parent dcaf1796fe
commit 5385bf9555

View file

@ -32,7 +32,12 @@ type CassandraStore struct {
func NewCassandraStore(keyspace string, hosts ...string) (c *CassandraStore, err error) {
c = &CassandraStore{}
c.cluster = gocql.NewCluster(hosts...)
s := strings.Split(hosts, ",")
if len(s) == 1 {
c.cluster = gocql.NewCluster(hosts...)
} else if len(s) > 1 {
c.cluster = gocql.NewCluster(s[0], s[1])
}
c.cluster.Keyspace = keyspace
c.cluster.Consistency = gocql.Quorum
c.session, err = c.cluster.CreateSession()