dats gey

Author: AmarnathCJD

telegram/client.go

@@ -59,13 +59,14 @@ func (e *exSenders) setTTL() {
 // Client is the main struct of the library
 type Client struct {
 	*mtproto.MTProto
-	Cache      *CACHE
-	clientData clientData
-	dispatcher *UpdateDispatcher
-	wg         sync.WaitGroup
-	stopCh     chan struct{}
-	exSenders  *exSenders
-	Log        *utils.Logger
+	Cache        *CACHE
+	clientData   clientData
+	dispatcher   *UpdateDispatcher
+	wg           sync.WaitGroup
+	stopCh       chan struct{}
+	exSenders    *exSenders
+	exportedKeys map[int]*ExportedAuthParams
+	Log          *utils.Logger
 }
 
 type DeviceConfig struct {

telegram/media.go

@@ -695,15 +695,25 @@ func initializeWorkers(numWorkers int, dc int32, c *Client, w *WorkerPool) {
 
 	var authParams = ExportedAuthParams{}
 	if dc != int32(c.GetDC()) {
-		auth, err := c.AuthExportAuthorization(dc)
-		if err != nil {
-			c.Log.Error(err)
-			return
+		if c.exportedKeys == nil {
+			c.exportedKeys = make(map[int]*ExportedAuthParams)
 		}
 
-		authParams = ExportedAuthParams{
-			ID:    auth.ID,
-			Bytes: auth.Bytes,
+		if exportedKey, ok := c.exportedKeys[int(dc)]; ok {
+			authParams = *exportedKey
+		} else {
+			auth, err := c.AuthExportAuthorization(dc)
+			if err != nil {
+				c.Log.Error(err)
+				return
+			}
+
+			authParams = ExportedAuthParams{
+				ID:    auth.ID,
+				Bytes: auth.Bytes,
+			}
+
+			c.exportedKeys[int(dc)] = &authParams
 		}
 	}