forked from JabRef/jabref
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Logic for counting views is now working correctly (JabRef#10418)
- Loading branch information
1 parent
5367b87
commit 707e3c0
Showing
3 changed files
with
140 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
92 changes: 92 additions & 0 deletions
92
src/main/java/org/jabref/model/groups/PopularityGroup.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,92 @@ | ||
| package org.jabref.model.groups; | ||
|
|
||
| import java.nio.file.Path; | ||
| import java.util.Objects; | ||
|
|
||
| import org.jabref.model.entry.BibEntry; | ||
| import org.jabref.model.entry.field.StandardField; | ||
|
|
||
| import org.h2.mvstore.MVMap; | ||
| import org.h2.mvstore.MVStore; | ||
|
|
||
| import static org.jabref.gui.desktop.os.NativeDesktop.getOtherDataDir; | ||
|
|
||
| /** | ||
| * This group contains entries that have been viewed more than once. | ||
| * {@link StandardField#GROUPS}. | ||
| */ | ||
| public class PopularityGroup extends AbstractGroup { | ||
| private static MVStore mvStore; | ||
|
|
||
| public PopularityGroup(String name, GroupHierarchyType context) { | ||
| super(name, context); | ||
| mvStore = getMVStore(); | ||
| } | ||
|
|
||
| public boolean contains(BibEntry entry) { | ||
| int viewCount = getEntryViewCount(entry); | ||
| return viewCount > 0; | ||
| } | ||
|
|
||
| @Override | ||
| public AbstractGroup deepCopy() { | ||
| // Create a copy of this popularity group with the same tracker and context | ||
| return new PopularityGroup(getName(), getHierarchicalContext()); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean isDynamic() { | ||
| // This group is dynamic since it changes based on view count | ||
| return true; | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object o) { | ||
| if (this == o) { | ||
| return true; | ||
| } | ||
| if (!(o instanceof PopularityGroup other)) { | ||
| return false; | ||
| } | ||
| return Objects.equals(getName(), other.getName()) | ||
| && Objects.equals(getHierarchicalContext(), other.getHierarchicalContext()); | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(getName(), getHierarchicalContext()); | ||
| } | ||
|
|
||
| /** | ||
| * Retrieves the view count for a given BibEntry. | ||
| * | ||
| * @param entry The BibEntry whose view count is to be fetched. | ||
| * @return The number of views for the BibEntry, or 0 if not found. | ||
| */ | ||
| public static int getEntryViewCount(BibEntry entry) { | ||
| MVMap<String, Integer> viewCounts = mvStore.openMap("entryViewCounts"); | ||
| String uniqueKey = getUniqueKeyForEntry(entry); | ||
|
|
||
| // Log the entry and unique key | ||
| System.out.println("BibEntry: " + entry); | ||
| System.out.println("UniqueKey: " + uniqueKey); | ||
|
|
||
| return viewCounts.getOrDefault(uniqueKey, 0); | ||
| } | ||
|
|
||
| public static String getUniqueKeyForEntry(BibEntry entry) { | ||
| return entry.getField(StandardField.DOI).orElse( | ||
| entry.getField(StandardField.KEY).orElse( | ||
| String.valueOf(entry.hashCode()) | ||
| ) | ||
| ); | ||
| } | ||
|
|
||
| public static synchronized MVStore getMVStore() { | ||
| if (mvStore == null) { | ||
| Path mvStorePath = getOtherDataDir().resolve("tracking.mv"); | ||
| mvStore = new MVStore.Builder().fileName(mvStorePath.toString()).open(); | ||
| } | ||
| return mvStore; | ||
| } | ||
| } |