github/dbx
4
0
Fork 0
mirror of https://github.com/t8y2/dbx.git synced 2026-10-21 06:00:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(iotdb): omit tree database from session handshake

Browse Source 
Closes #5826
This commit is contained in:
t8y2
2026-08-12 15:45:24 +00:00
parent 9fb70bcfc9
commit 847afca6fe
2 changed files with 27 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agents/drivers/iotdb/driver.go
+3 -6
View File
@@ -215,10 +215,9 @@ func parseConnectionConfig(params connectParams) (connectionConfig, error) {
func newSessionClient(config connectionConfig) (*sessionClient, error) {
var session client.Session
var err error
database := config.Database
if config.Dialect == client.TableSqlDialect {
database = ""
}
// DBX applies a table database with USE after switching dialects. Do not
// include a tree database in openSession: IoTDB 2.x rejects it there, while
// DBX still retains it for metadata and path qualification.
if len(config.NodeURLs) > 1 {
session, err = client.NewClusterSession(&client.ClusterConfig{
NodeUrls: config.NodeURLs,
@@ -227,7 +226,6 @@ func newSessionClient(config connectionConfig) (*sessionClient, error) {
FetchSize: config.FetchSize,
TimeZone: config.TimeZone,
ConnectRetryMax: config.ConnectRetryMax,
Database: database,
TLSConfig: config.TLSConfig,
})
if err == nil {
@@ -242,7 +240,6 @@ func newSessionClient(config connectionConfig) (*sessionClient, error) {
FetchSize: config.FetchSize,
TimeZone: config.TimeZone,
ConnectRetryMax: config.ConnectRetryMax,
Database: database,
TLSConfig: config.TLSConfig,
})
err = session.Open(config.EnableCompression, config.ConnectTimeoutMS)
agents/drivers/iotdb/integration_test.go
+24
View File
@@ -87,6 +87,30 @@ func TestLiveIoTDBAgentTreeAndTable(t *testing.T) {
}
}
func TestLiveIoTDBAgentTreeDatabaseConnection(t *testing.T) {
if os.Getenv("DBX_IOTDB_LIVE") != "1" {
t.Skip("set DBX_IOTDB_LIVE=1 to run against a real IoTDB server")
}
database := "root.dbx_go_connection_" + strconv.Itoa(os.Getpid())
bootstrap := liveIoTDBServer(t, connectParams{})
defer bootstrap.disconnect()
mustExecuteNonQuery(t, bootstrap, "CREATE DATABASE "+database, "")
defer func() { _ = bootstrap.executeNonQuery("DELETE DATABASE "+database, "", 0) }()
scoped := liveIoTDBServer(t, connectParams{Database: database})
defer scoped.disconnect()
if err := scoped.validateConnection(); err != nil {
t.Fatal(err)
}
tables, err := scoped.listTables(database, metadataListConstraints{})
if err != nil {
t.Fatal(err)
}
if tables == nil {
t.Fatal("expected an initialized tree table list")
}
}
func TestLiveIoTDBAgentProcessMultiSessionAndCancellation(t *testing.T) {
if os.Getenv("DBX_IOTDB_LIVE") != "1" {
t.Skip("set DBX_IOTDB_LIVE=1 to run against a real IoTDB server")