Add ShowSpaces and Execute for Session (#315)

Add Show spaces and for

Co-authored-by: Anqi <[email protected]>

session.go

@@ -185,6 +185,32 @@ func (session *Session) ExecuteJsonWithParameter(stmt string, params map[string]
 	return resp.([]byte), err
 }
 
+func (session *Session) ExecuteAndCheck(stmt string) (*ResultSet, error) {
+	rs, err := session.Execute(stmt)
+	if err != nil {
+		return nil, err
+	}
+
+	if !rs.IsSucceed() {
+		errMsg := rs.GetErrorMsg()
+		return nil, fmt.Errorf("Fail to execute query. %s", errMsg)
+	}
+
+	return rs, nil
+}
+
+func (session *Session) ShowSpaces() ([]SpaceName, error) {
+	rs, err := session.ExecuteAndCheck("SHOW SPACES;")
+	if err != nil {
+		return nil, err
+	}
+
+	var names []SpaceName
+	rs.Scan(&names)
+
+	return names, nil
+}
+
 func (session *Session) reConnect() error {
 	newConnection, err := session.connPool.getIdleConn()
 	if err != nil {

session_pool.go

@@ -278,8 +278,7 @@ func (pool *SessionPool) ExecuteAndCheck(q string) (*ResultSet, error) {
 		return nil, err
 	}
 
-	errCode := rs.GetErrorCode()
-	if errCode != ErrorCode_SUCCEEDED {
+	if !rs.IsSucceed() {
 		errMsg := rs.GetErrorMsg()
 		return nil, fmt.Errorf("fail to execute query. %s", errMsg)
 	}

session_test.go

@@ -117,3 +117,24 @@ func TestSession_Recover(t *testing.T) {
 	}
 	assert.Equal(t, 1, pool.getActiveConnCount()+pool.getIdleConnCount())
 }
+
+func TestSession_ShowSpaces(t *testing.T) {
+	config := GetDefaultConf()
+	host := HostAddress{address, port}
+	pool, err := NewConnectionPool([]HostAddress{host}, config, DefaultLogger{})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	sess, err := pool.GetSession("root", "nebula")
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.Equal(t, 1, pool.getActiveConnCount()+pool.getIdleConnCount())
+
+	spaceNames, err := sess.ShowSpaces()
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.LessOrEqual(t, 1, len(spaceNames))
+}