'去测试'返回错误" gocql:在超时期限内没有从cassandra收到任何响应。"

时间:2017-05-21 23:16:17

标签: go cassandra connection-timeout gocql

所以我想在我的本地机器上测试一个与Cassandra交互的API。在我的func TestMain(m *testing.M)函数中,我想在运行测试之前清除表。 TestMain函数看起来像这样......

func TestMain(m *testing.M) {
    keyspace = "staging"
    cassandra.SetKeyspace(keyspace)

    client = http.DefaultClient

    // Ensure all tables are empty before tests run
    err := ClearAllTables()
    if err != nil {
        logrus.Errorf("Failed to clear all tables: %v.", err)
        os.Exit(1)
    }

    // Run tests
    statusCode := m.Run()

    os.Exit(statusCode)
}

ClearAllTables函数看起来像这样......

func ClearAllTables() (err error) {

    // Create a DB session
    session := cassandra.CreateSession()
    defer session.Close()

    // Get names of all existing tables
    tableNameList := []string{"cliques", "users"}

    // Remove all rows from each table
    var count int
    for _, tableName := range tableNameList {

        if err := session.Query(`TRUNCATE TABLE ` + tableName).Exec(); err != nil {
            return err
        }
    }
    return nil
}

出于某种原因,当我尝试TRUNCATE表时,Cassandra超时,我得到了错误......

level=error msg="Failed to clear all tables: gocql: no response received from cassandra within timeout period."

这似乎只在我测试程序时发生。我在main函数中编写了一段代码,工作正常。

func main () {
    session := cassandra.CreateSession()
    defer session.Close()

    if err := session.Query(`TRUNCATE TABLE cliques`).Exec(); err != nil {
        return
    }
    fmt.Println("Table truncated") //works
}

我还写了一段代码片段,它从数据库中返回一些行,并且在main函数中也可以正常工作。

以下是我如何创建我的cassandra会话......

// CreateSession connect to a cassandra instance
func CreateSession() *gocql.Session {
    // Connect to the cluster
    cluster := gocql.NewCluster("127.0.0.1")
    cluster.Keyspace = "staging"
    cluster.ProtoVersion = 4
    cluster.CQLVersion = "3.0.0"
    cluster.Consistency = gocql.One

    session, err := cluster.CreateSession()
    if err != nil {
        logrus.Errorf("Failed to connect to Cassandra: %v.", err)
    }
    return session
}

我错过了什么,为什么Cassandra可以与go run main.go一起使用,但不能与go test一起使用?

0 个答案:

没有答案