ProfileInfo not availble when using Batch.Send()

[iyuroch]

Observed

1. When providing context with profile info callback function to batch after Send() profileinfo not available
2. Doing the same for select query result in profileinfo

Expected behaviour

Profile info available

Code example

clickhouse.Context(ctx, clickhouse.WithProgress(func(p *clickhouse.Progress) {
		fmt.Println("progress: ", p)
	}), clickhouse.WithProfileInfo(func(p *clickhouse.ProfileInfo) {
		fmt.Println("profile info: ", p)
	}), clickhouse.WithLogs(func(log *clickhouse.Log) {
		fmt.Println("log info: ", log)
	}))

	insert := fmt.Sprintf("INSERT INTO %s (%s)", tableName, strings.Join(cols, ", "))
	chBatch, _ := conn.PrepareBatch(ctx, insert)
	chBatch.Send()

Details

Environment

• clickhouse-go version: github.com/ClickHouse/clickhouse-go/v2 v2.23.0
• Interface: ClickHouse API compatible driver
• Go version:
• Operating system:
• ClickHouse version:
• Is it a ClickHouse Cloud?
• ClickHouse Server non-default settings, if any:
• CREATE TABLE statements for tables involved: (not related)
• Sample data for all these tables, use [clickhouse-obfuscator] (not related)(https://github.com/ClickHouse/ClickHouse/blob/master/programs/obfuscator/Obfuscator.cpp#L42-L80) if necessary

[iyuroch]

I can see that in the code we are going to end of the stream from the server, is there any other way how I can get profile info for batch.Send()? One way I can think of is passing query_id as an option and then checking in query log