fabian/telegram_backup
1
0
Fork 0
mirror of https://github.com/fabianonline/telegram_backup.git synced 2024-11-22 16:56:16 +00:00
Code Issues Releases Wiki Activity

File size-aware file downloading. Fixes #1.

Browse Source
This commit is contained in:
Fabian Schlenz 2016-07-02 19:43:01 +02:00
parent d75d409f87
commit a2d3dadf7b
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -127,7 +127,7 @@ class DownloadManager {
throw new RuntimeException("Could not find a size for a photo."); throw new RuntimeException("Could not find a size for a photo.");
} }
if (size.getLocation() instanceof TLFileLocation) { if (size.getLocation() instanceof TLFileLocation) {
boolean res = this.downloadPhoto(msg.getId(), (TLFileLocation)size.getLocation()); boolean res = this.downloadPhoto(msg.getId(), (TLFileLocation)size.getLocation(), size.getSize());
prog.onMediaDownloadedPhoto(res); prog.onMediaDownloadedPhoto(res);
} }
} else { } else {
@ -198,27 +198,27 @@ class DownloadManager {
return a; return a;
} }
private boolean downloadPhoto(int msgId, TLFileLocation src) throws RpcErrorException, IOException { private boolean downloadPhoto(int msgId, TLFileLocation src, int size) throws RpcErrorException, IOException {
TLInputFileLocation loc = new TLInputFileLocation(); TLInputFileLocation loc = new TLInputFileLocation();
loc.setVolumeId(src.getVolumeId()); loc.setVolumeId(src.getVolumeId());
loc.setLocalId(src.getLocalId()); loc.setLocalId(src.getLocalId());
loc.setSecret(src.getSecret()); loc.setSecret(src.getSecret());
return this.downloadFile(this.makeFilename(msgId, "jpg"), loc); return this.downloadFile(this.makeFilename(msgId, "jpg"), loc, size);
} }
private boolean downloadDocument(String filename, TLDocument doc) throws RpcErrorException, IOException { private boolean downloadDocument(String filename, TLDocument doc) throws RpcErrorException, IOException {
TLInputDocumentFileLocation loc = new TLInputDocumentFileLocation(); TLInputDocumentFileLocation loc = new TLInputDocumentFileLocation();
loc.setId(doc.getId()); loc.setId(doc.getId());
loc.setAccessHash(doc.getAccessHash()); loc.setAccessHash(doc.getAccessHash());
return this.downloadFileFromDc(filename, loc, doc.getDcId()); return this.downloadFileFromDc(filename, loc, doc.getDcId(), doc.getSize());
} }
private boolean downloadVideo(String filename, TLVideo vid) throws RpcErrorException, IOException { private boolean downloadVideo(String filename, TLVideo vid) throws RpcErrorException, IOException {
TLInputDocumentFileLocation loc = new TLInputDocumentFileLocation(); TLInputDocumentFileLocation loc = new TLInputDocumentFileLocation();
loc.setId(vid.getId()); loc.setId(vid.getId());
loc.setAccessHash(vid.getAccessHash()); loc.setAccessHash(vid.getAccessHash());
return this.downloadFileFromDc(filename, loc, vid.getDcId()); return this.downloadFileFromDc(filename, loc, vid.getDcId(), vid.getSize());
} }
private String makeFilename(int id, String ext) { private String makeFilename(int id, String ext) {
@ -230,11 +230,11 @@ class DownloadManager {
return path + id + ".dat"; return path + id + ".dat";
} }
private boolean downloadFile(String target, TLAbsInputFileLocation loc) throws RpcErrorException, IOException { private boolean downloadFile(String target, TLAbsInputFileLocation loc, int size) throws RpcErrorException, IOException {
return downloadFileFromDc(target, loc, null); return downloadFileFromDc(target, loc, null, size);
} }
private boolean downloadFileFromDc(String target, TLAbsInputFileLocation loc, Integer dcID) throws RpcErrorException, IOException { private boolean downloadFileFromDc(String target, TLAbsInputFileLocation loc, Integer dcID, int size) throws RpcErrorException, IOException {
// Don't download already existing files. // Don't download already existing files.
if (new File(target).isFile()) return false; if (new File(target).isFile()) return false;
@ -248,12 +248,16 @@ class DownloadManager {
if (dcID==null) { if (dcID==null) {
response = (TLFile)this.client.executeRpcQuery(req); response = (TLFile)this.client.executeRpcQuery(req);
} else { } else {
response = (TLFile)this.client.executeRpcQuery(req, dcID); response = (TLFile) this.client.executeRpcQuery(req, dcID);
} }
offset += Config.FILE_DOWNLOAD_BLOCK_SIZE; offset += response.getBytes().getData().length;
fos.write(response.getBytes().getData()); fos.write(response.getBytes().getData());
try { Thread.sleep(Config.DELAY_AFTER_GET_FILE); } catch(InterruptedException e) {} try { Thread.sleep(Config.DELAY_AFTER_GET_FILE); } catch(InterruptedException e) {}
} while(response.getBytes().getLength() == Config.FILE_DOWNLOAD_BLOCK_SIZE); } while(offset < size && response.getBytes().getData().length>0);
if (offset < size) {
System.out.println("Requested file " + target + " with " + size + " bytes, but got only " + offset + " bytes.");
System.exit(1);
}
fos.close(); fos.close();
return true; return true;
} catch (java.io.IOException ex) { } catch (java.io.IOException ex) {