Add sessionID support and BeforeExecuteWithTime to telemetry

- Add sessionID field to metricContext struct
- Update BeforeExecute to accept sessionID parameter
- Add BeforeExecuteWithTime method for custom start times
- Update connection.go to pass sessionID in BeforeExecute call

This enables proper session tracking in telemetry and allows
capturing accurate execution times by providing a custom start time.

Author: samikshya-db

connection.go

@@ -200,7 +200,7 @@ func (c *conn) QueryContext(ctx context.Context, query string, args []driver.Nam
 	var statementID string
 	if c.telemetry != nil && exStmtResp != nil && exStmtResp.OperationHandle != nil && exStmtResp.OperationHandle.OperationId != nil {
 		statementID = client.SprintGuid(exStmtResp.OperationHandle.OperationId.GUID)
-		ctx = c.telemetry.BeforeExecute(ctx, statementID)
+		ctx = c.telemetry.BeforeExecute(ctx, c.id, statementID)
 		defer func() {
 			c.telemetry.AfterExecute(ctx, err)
 			c.telemetry.CompleteStatement(ctx, statementID, err != nil)

telemetry/interceptor.go

@@ -16,6 +16,7 @@ type Interceptor struct {
 
 // metricContext holds metric collection state in context.
 type metricContext struct {
+	sessionID   string
 	statementID string
 	startTime   time.Time
 	tags        map[string]interface{}
@@ -49,12 +50,13 @@ func getMetricContext(ctx context.Context) *metricContext {
 // BeforeExecute is called before statement execution.
 // Returns a new context with metric tracking attached.
 // Exported for use by the driver package.
-func (i *Interceptor) BeforeExecute(ctx context.Context, statementID string) context.Context {
+func (i *Interceptor) BeforeExecute(ctx context.Context, sessionID string, statementID string) context.Context {
 	if !i.enabled {
 		return ctx
 	}
 
 	mc := &metricContext{
+		sessionID:   sessionID,
 		statementID: statementID,
 		startTime:   time.Now(),
 		tags:        make(map[string]interface{}),
@@ -63,6 +65,24 @@ func (i *Interceptor) BeforeExecute(ctx context.Context, statementID string) con
 	return withMetricContext(ctx, mc)
 }
 
+// BeforeExecuteWithTime is called before statement execution with a custom start time.
+// This is useful when the statement ID is not known until after execution starts.
+// Exported for use by the driver package.
+func (i *Interceptor) BeforeExecuteWithTime(ctx context.Context, sessionID string, statementID string, startTime time.Time) context.Context {
+	if !i.enabled {
+		return ctx
+	}
+
+	mc := &metricContext{
+		sessionID:   sessionID,
+		statementID: statementID,
+		startTime:   startTime,
+		tags:        make(map[string]interface{}),
+	}
+
+	return withMetricContext(ctx, mc)
+}
+
 // AfterExecute is called after statement execution.
 // Records the metric with timing and error information.
 // Exported for use by the driver package.
@@ -86,6 +106,7 @@ func (i *Interceptor) AfterExecute(ctx context.Context, err error) {
 	metric := &telemetryMetric{
 		metricType:  "statement",
 		timestamp:   mc.startTime,
+		sessionID:   mc.sessionID,
 		statementID: mc.statementID,
 		latencyMs:   time.Since(mc.startTime).Milliseconds(),
 		tags:        mc.tags,