fabian
/
telegram_backup
mirror of https://github.com/fabianonline/telegram_backup.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Added timeout detection to media downloads as well.

This commit is contained in:
Fabian Schlenz 2017-02-18 12:56:56 +01:00
parent e0668926c9
commit 6738e20fe4
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/de/fabianonline/telegram_backup/DownloadManager.java
View File

@ -319,15 +319,26 @@ public class DownloadManager {
}
logger.trace("offset before the loop is {}", offset);
fos = new FileOutputStream(temp_filename, true);
TLFile response;
TLFile response = null;
boolean try_again;
do {
try_again = false;
int block_size = size;
logger.trace("offset: {} block_size: {} size: {}", offset, block_size, size);
TLRequestUploadGetFile req = new TLRequestUploadGetFile(loc, offset, block_size);
if (dcID==null) {
response = (TLFile) download_client.executeRpcQuery(req);
} else {
response = (TLFile) download_client.executeRpcQuery(req, dcID);
try {
if (dcID==null) {
response = (TLFile) download_client.executeRpcQuery(req);
} else {
response = (TLFile) download_client.executeRpcQuery(req, dcID);
}
} catch (RpcErrorException e) {
if (e.getTag().startsWith("420: FLOOD_WAIT_")) {
try_again = true;
Utils.obeyFloodWaitException(e);
} else {
throw e;
}
}
offset += response.getBytes().getData().length;
@ -336,7 +347,7 @@ public class DownloadManager {
fos.write(response.getBytes().getData());
fos.flush();
try { TimeUnit.MILLISECONDS.sleep(Config.DELAY_AFTER_GET_FILE); } catch(InterruptedException e) {}
} while(offset < size && response.getBytes().getData().length>0);
} while(offset < size && (response.getBytes().getData().length>0 || try_again));
fos.close();
if (offset < size) {
System.out.println("Requested file " + target + " with " + size + " bytes, but got only " + offset + " bytes.");