fix: use wrapped context err when <-ctx.Done() case is executed

Author: Max Maximov

CHANGELOG.md

@@ -14,6 +14,8 @@ Versioning](http://semver.org/spec/v2.0.0.html) except to the first release.
 
 ### Fixed
 
+- Use wrapped context err when <-ctx.Done() case is executed
+
 ## [v2.4.0] - 2025-07-11
 
 This release focuses on adding schema/user/session operations, synchronous transaction

connection.go

@@ -36,7 +36,7 @@ type ConnLogKind int
 
 var (
 	errUnknownRequest = errors.New("the passed connected request doesn't belong " +
-		"to the current connection or connection pool")
+			"to the current connection or connection pool")
 )
 
 const (
@@ -113,7 +113,7 @@ func (d defaultLogger) Report(event ConnLogKind, conn *Connection, v ...interfac
 	case LogUnexpectedResultId:
 		header := v[0].(Header)
 		log.Printf("tarantool: connection %s got unexpected request ID (%d) in response "+
-			"(probably cancelled request)",
+				"(probably cancelled request)",
 			conn.Addr(), header.RequestId)
 	case LogWatchEventReadFailed:
 		err := v[0].(error)
@@ -515,8 +515,8 @@ func (conn *Connection) dial(ctx context.Context) error {
 
 	// Subscribe shutdown event to process graceful shutdown.
 	if conn.shutdownWatcher == nil &&
-		isFeatureInSlice(iproto.IPROTO_FEATURE_WATCHERS,
-			conn.serverProtocolInfo.Features) {
+			isFeatureInSlice(iproto.IPROTO_FEATURE_WATCHERS,
+				conn.serverProtocolInfo.Features) {
 		watcher, werr := conn.newWatcherImpl(shutdownEventKey, shutdownEventCallback)
 		if werr != nil {
 			return werr
@@ -528,7 +528,7 @@ func (conn *Connection) dial(ctx context.Context) error {
 }
 
 func pack(h *smallWBuf, enc *msgpack.Encoder, reqid uint32,
-	req Request, streamId uint64, res SchemaResolver) (err error) {
+		req Request, streamId uint64, res SchemaResolver) (err error) {
 	const uint32Code = 0xce
 	const uint64Code = 0xcf
 	const streamBytesLenUint64 = 10
@@ -665,7 +665,7 @@ func (conn *Connection) runReconnects(ctx context.Context) error {
 				return err
 			}
 			if clientErr, ok := err.(ClientError); ok &&
-				clientErr.Code == ErrConnectionClosed {
+					clientErr.Code == ErrConnectionClosed {
 				return err
 			}
 		} else {
@@ -984,7 +984,10 @@ func (conn *Connection) newFuture(req Request) (fut *Future) {
 	if ctx != nil {
 		select {
 		case <-ctx.Done():
-			fut.SetError(fmt.Errorf("context is done (request ID %d)", fut.requestId))
+			fut.SetError(fmt.Errorf(
+				"context is done (request ID %d): %w",
+				fut.requestId, context.Cause(ctx),
+			))
 			shard.rmut.Unlock()
 			return
 		default:
@@ -1026,7 +1029,10 @@ func (conn *Connection) contextWatchdog(fut *Future, ctx context.Context) {
 	case <-fut.done:
 		return
 	default:
-		conn.cancelFuture(fut, fmt.Errorf("context is done (request ID %d)", fut.requestId))
+		conn.cancelFuture(fut, fmt.Errorf(
+			"context is done (request ID %d): %w",
+			fut.requestId, context.Cause(ctx),
+		))
 	}
 }
 
@@ -1228,9 +1234,9 @@ func read(r io.Reader, lenbuf []byte) (response []byte, err error) {
 		return
 	}
 	length = (uint64(lenbuf[1]) << 24) +
-		(uint64(lenbuf[2]) << 16) +
-		(uint64(lenbuf[3]) << 8) +
-		uint64(lenbuf[4])
+			(uint64(lenbuf[2]) << 16) +
+			(uint64(lenbuf[3]) << 8) +
+			uint64(lenbuf[4])
 
 	switch {
 	case length == 0:
@@ -1279,8 +1285,8 @@ func (conn *Connection) ConfiguredTimeout() time.Duration {
 func (conn *Connection) SetSchema(s Schema) {
 	sCopy := s.copy()
 	spaceAndIndexNamesSupported :=
-		isFeatureInSlice(iproto.IPROTO_FEATURE_SPACE_AND_INDEX_NAMES,
-			conn.serverProtocolInfo.Features)
+			isFeatureInSlice(iproto.IPROTO_FEATURE_SPACE_AND_INDEX_NAMES,
+				conn.serverProtocolInfo.Features)
 
 	conn.mutex.Lock()
 	defer conn.mutex.Unlock()
@@ -1463,7 +1469,7 @@ func (conn *Connection) NewWatcher(key string, callback WatchCallback) (Watcher,
 	if !isFeatureInSlice(iproto.IPROTO_FEATURE_WATCHERS,
 		conn.serverProtocolInfo.Features) {
 		err := fmt.Errorf("the feature %s must be supported by connection "+
-			"to create a watcher", iproto.IPROTO_FEATURE_WATCHERS)
+				"to create a watcher", iproto.IPROTO_FEATURE_WATCHERS)
 		return nil, err
 	}