Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TASK-3836 - Fix CellBase data release usage #2242

Merged
merged 2 commits into from
Jan 27, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,8 @@ public OpenCGAResult<GeneCoverageStats> coverageStats(String studyIdStr, String
// Get species and assembly from catalog
OpenCGAResult<Project> projectQueryResult = catalogManager.getProjectManager().search(
new Query(ProjectDBAdaptor.QueryParams.STUDY.key(), studyIdStr),
new QueryOptions(QueryOptions.INCLUDE, ProjectDBAdaptor.QueryParams.ORGANISM.key()), token);
new QueryOptions(QueryOptions.INCLUDE, Arrays.asList(ProjectDBAdaptor.QueryParams.ORGANISM.key(),
ProjectDBAdaptor.QueryParams.CELLBASE.key())), token);
if (projectQueryResult.getNumResults() != 1) {
throw new CatalogException("Error getting species and assembly from catalog");
}
Expand Down Expand Up @@ -438,7 +439,9 @@ public List<Region> mergeRegions(List<Region> regions, List<String> genes, boole
// Get species and assembly from catalog
OpenCGAResult<Project> projectQueryResult = catalogManager.getProjectManager().search(
new Query(ProjectDBAdaptor.QueryParams.STUDY.key(), study),
new QueryOptions(QueryOptions.INCLUDE, ProjectDBAdaptor.QueryParams.ORGANISM.key()), token);
new QueryOptions(QueryOptions.INCLUDE, Arrays.asList(
ProjectDBAdaptor.QueryParams.ORGANISM.key(),
ProjectDBAdaptor.QueryParams.CELLBASE.key())), token);
if (projectQueryResult.getNumResults() != 1) {
throw new CatalogException("Error getting species and assembly from catalog");
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ private void annotate(String projectStr, List<String> studies, String loadFileSt
}

private void synchronizeProjectMetadata(String projectStr, String token) throws CatalogException, StorageEngineException {
Project project = catalogManager.getProjectManager().get(projectStr, null, token).first();
Project project = catalogManager.getProjectManager().get(projectStr, QueryOptions.empty(), token).first();
ProjectOrganism organism = project.getOrganism();
int currentRelease = project.getCurrentRelease();
CatalogStorageMetadataSynchronizer.updateProjectMetadata(variantStorageEngine.getMetadataManager(), organism, currentRelease,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,7 @@

import static org.opencb.opencga.catalog.db.api.FileDBAdaptor.QueryParams.ID;
import static org.opencb.opencga.catalog.db.api.FileDBAdaptor.QueryParams.UID;
import static org.opencb.opencga.catalog.db.api.ProjectDBAdaptor.QueryParams.CURRENT_RELEASE;
import static org.opencb.opencga.catalog.db.api.ProjectDBAdaptor.QueryParams.ORGANISM;
import static org.opencb.opencga.catalog.db.api.ProjectDBAdaptor.QueryParams.*;

/**
* Created by imedina on 17/08/16.
Expand Down Expand Up @@ -163,7 +162,8 @@ private void updateProject(String studyFqn, String token) throws CatalogExceptio
Project project = catalogManager
.getProjectManager()
.get(projectFqn,
new QueryOptions(QueryOptions.INCLUDE, Arrays.asList(CURRENT_RELEASE.key(), ORGANISM.key())), token).first();
new QueryOptions(QueryOptions.INCLUDE, Arrays.asList(CURRENT_RELEASE.key(), ORGANISM.key(), CELLBASE.key())),
token).first();
release = project.getCurrentRelease();

// Add species, assembly and release
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,8 @@ public static void updateProjectMetadata(CatalogManager catalog, VariantStorageM
throws CatalogException, StorageEngineException {
final Project p = catalog.getProjectManager().get(project,
new QueryOptions(QueryOptions.INCLUDE, Arrays.asList(
ProjectDBAdaptor.QueryParams.ORGANISM.key(), ProjectDBAdaptor.QueryParams.CURRENT_RELEASE.key())),
ProjectDBAdaptor.QueryParams.ORGANISM.key(), ProjectDBAdaptor.QueryParams.CURRENT_RELEASE.key(),
ProjectDBAdaptor.QueryParams.CELLBASE.key())),
sessionId)
.first();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ protected void run() throws Exception {
// StorageEngineFactory engineFactory = StorageEngineFactory.get(storageConfiguration);
// VariantStorageManager variantStorageManager = new VariantStorageManager(catalogManager, engineFactory);

for (Project project : catalogManager.getProjectManager().search(new Query(), new QueryOptions(), token).getResults()) {
for (Project project : catalogManager.getProjectManager().search(new Query(), QueryOptions.empty(), token).getResults()) {
if (project.getInternal() == null || project.getCellbase() == null) {
String userToken = catalogManager.getUserManager()
.getNonExpiringToken(project.getFqn().split(USER_PROJECT_SEPARATOR)[0], token);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ public class AddCellbaseDataRelease extends StorageMigrationTool {
protected void run() throws Exception {
VariantStorageManager variantStorageManager = getVariantStorageManager();
for (String projectFqn : getVariantStorageProjects()) {
Project project = catalogManager.getProjectManager().get(projectFqn, new QueryOptions(), token).first();
Project project = catalogManager.getProjectManager().get(projectFqn, QueryOptions.empty(), token).first();
CellBaseConfiguration cellbase = project.getCellbase();
CellBaseUtils cellBaseUtils = getVariantStorageEngineByProject(projectFqn).getCellBaseUtils();
boolean updateCellbase = false;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ public void before() throws Exception {
protected void loadFiles() throws Exception {
variantStorageEngine.getConfiguration().getCellbase().setUrl(ParamConstants.CELLBASE_URL);
variantStorageEngine.getConfiguration().getCellbase().setVersion("v5.1");
variantStorageEngine.getConfiguration().getCellbase().setDataRelease("2");
variantStorageEngine.getOptions().put(VariantStorageOptions.ASSEMBLY.key(), "grch38");
variantStorageEngine.reloadCellbaseConfiguration();

Expand Down