idk/muwire
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

key by infohash

Browse Source
This commit is contained in:
Zlatin Balevsky
2019-05-29 11:40:01 +01:00
parent 380ec45bba
commit 7618e2d0fb
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
core/src/main/groovy/com/muwire/core/files/FileManager.groovy
View File

@@ -1,16 +1,20 @@
package com.muwire.core.files package com.muwire.core.files
import com.muwire.core.EventBus import com.muwire.core.EventBus
import com.muwire.core.InfoHash
import com.muwire.core.SharedFile import com.muwire.core.SharedFile
import com.muwire.core.search.ResultsEvent import com.muwire.core.search.ResultsEvent
import com.muwire.core.search.SearchEvent import com.muwire.core.search.SearchEvent
import com.muwire.core.search.SearchIndex import com.muwire.core.search.SearchIndex
import groovy.util.logging.Log
@Log
class FileManager { class FileManager {
final EventBus eventBus final EventBus eventBus
final Map<byte[], Set<SharedFile>> rootToFiles = Collections.synchronizedMap(new HashMap<>()) final Map<InfoHash, Set<SharedFile>> rootToFiles = Collections.synchronizedMap(new HashMap<>())
final Map<File, SharedFile> fileToSharedFile = Collections.synchronizedMap(new HashMap<>()) final Map<File, SharedFile> fileToSharedFile = Collections.synchronizedMap(new HashMap<>())
final Map<String, Set<File>> nameToFiles = new HashMap<>() final Map<String, Set<File>> nameToFiles = new HashMap<>()
final SearchIndex index = new SearchIndex() final SearchIndex index = new SearchIndex()
@@ -33,11 +37,13 @@ class FileManager {
} }
private void addToIndex(SharedFile sf) { private void addToIndex(SharedFile sf) {
byte [] root = sf.getInfoHash().getRoot() log.info("Adding shared file " + sf.getFile())
Set<SharedFile> existing = rootToFiles.get(root) InfoHash infoHash = sf.getInfoHash()
Set<SharedFile> existing = rootToFiles.get(infoHash)
if (existing == null) { if (existing == null) {
log.info("adding new root")
existing = new HashSet<>() existing = new HashSet<>()
rootToFiles.put(root, existing); rootToFiles.put(infoHash, existing);
} }
existing.add(sf) existing.add(sf)
fileToSharedFile.put(sf.file, sf) fileToSharedFile.put(sf.file, sf)
@@ -55,12 +61,12 @@ class FileManager {
void onFileUnsharedEvent(FileUnsharedEvent e) { void onFileUnsharedEvent(FileUnsharedEvent e) {
SharedFile sf = e.unsharedFile SharedFile sf = e.unsharedFile
byte [] root = sf.getInfoHash().getRoot() InfoHash infoHash = sf.getInfoHash()
Set<SharedFile> existing = rootToFiles.get(root) Set<SharedFile> existing = rootToFiles.get(infoHash)
if (existing != null) { if (existing != null) {
existing.remove(sf) existing.remove(sf)
if (existing.isEmpty()) { if (existing.isEmpty()) {
rootToFiles.remove(root) rootToFiles.remove(infoHash)
} }
} }
@@ -85,7 +91,7 @@ class FileManager {
} }
Set<SharedFile> getSharedFiles(byte []root) { Set<SharedFile> getSharedFiles(byte []root) {
return rootToFiles.get(root) return rootToFiles.get(new InfoHash(root))
} }
void onSearchEvent(SearchEvent e) { void onSearchEvent(SearchEvent e) {
@@ -93,7 +99,7 @@ class FileManager {
ResultsEvent re = null ResultsEvent re = null
if (e.searchHash != null) { if (e.searchHash != null) {
Set<SharedFile> found Set<SharedFile> found
found = rootToFiles.get e.searchHash found = rootToFiles.get new InfoHash(e.searchHash)
if (found != null && !found.isEmpty()) if (found != null && !found.isEmpty())
re = new ResultsEvent(results: found.asList(), uuid: e.uuid) re = new ResultsEvent(results: found.asList(), uuid: e.uuid)
} else { } else {