fix hosts strings

fix hosts strings
This commit is contained in:
eshujiushiwo 2016-11-28 16:03:39 +08:00 committed by GitHub
parent 3450eff288
commit 5ee7fdc0cc

View file

@ -30,13 +30,13 @@ type CassandraStore struct {
session *gocql.Session
}
func NewCassandraStore(keyspace string, hosts ...string) (c *CassandraStore, err error) {
func NewCassandraStore(keyspace string, hosts string) (c *CassandraStore, err error) {
c = &CassandraStore{}
s := strings.Split(hosts, ",")
if len(s) == 1 {
glog.V(2).Info("Only one cassandra node to connect!A Cluster is Proposed!Now using:", string(hosts))
c.cluster = gocql.NewCluster(hosts...)
c.cluster = gocql.NewCluster(hosts)
} else if len(s) > 1 {
c.cluster = gocql.NewCluster(s[0], s[1])
}