Merge pull request #1666 from borzovplus/pass-error-to-onDead

Pass error to onDead callback args

Author: ernado

telegram/client.go

@@ -62,7 +62,7 @@ type Client struct {
 	connsCounter   atomic.Int64
 	create         connConstructor        // immutable
 	resolver       dcs.Resolver           // immutable
-	onDead         func()                 // immutable
+	onDead         func(error)            // immutable
 	newConnBackoff func() backoff.BackOff // immutable
 	defaultMode    manager.ConnMode       // immutable
 

telegram/conn_builder.go

@@ -75,7 +75,7 @@ func (c *Client) createConn(
 	id int64,
 	mode manager.ConnMode,
 	setup manager.SetupCallback,
-	onDead func(),
+	onDead func(error),
 ) pool.Conn {
 	opts, s := c.session.Options(c.opts)
 	opts.Logger = c.log.Named("conn").With(

telegram/internal/manager/conn.go

@@ -54,7 +54,7 @@ type Conn struct {
 	setup SetupCallback // nilable
 
 	// onDead is called on connection death.
-	onDead func()
+	onDead func(error)
 
 	// Wrappers for external world, like logs or PRNG.
 	// Should be immutable.
@@ -127,7 +127,7 @@ func (c *Conn) Run(ctx context.Context) (err error) {
 		if err != nil && ctx.Err() == nil {
 			c.log.Debug("Connection dead", zap.Error(err))
 			if c.onDead != nil {
-				c.onDead()
+				c.onDead(err)
 			}
 		}
 	}()

telegram/internal/manager/create.go

@@ -22,7 +22,7 @@ type ConnOptions struct {
 	Device  DeviceConfig
 	Handler Handler
 	Setup   SetupCallback
-	OnDead  func()
+	OnDead  func(error)
 	Backoff func(ctx context.Context) backoff.BackOff
 }
 

telegram/options.go

@@ -49,7 +49,7 @@ type Options struct {
 	// ReconnectionBackoff configures and returns reconnection backoff object.
 	ReconnectionBackoff func() backoff.BackOff
 	// OnDead will be called on connection dead.
-	OnDead func()
+	OnDead func(error)
 	// MigrationTimeout configures migration timeout.
 	MigrationTimeout time.Duration