Fix CSeqDB::GetDiskUsage to include top level BLASTDB files common to…

… entire database JIRA SB-2084 git-svn-id: https://anonsvn.ncbi.nlm.nih.gov/repos/v1/trunk/c++@101660 78c7ea69-d796-4a43-9a09-de51944f1b03
ncbi · Feb 3, 2024 · a536b45 · a536b45
1 parent 2cdfe03
commit a536b45
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/objtools/blast/seqdb_reader/seqdb.cpp b/src/objtools/blast/seqdb_reader/seqdb.cpp
@@ -1472,6 +1472,7 @@ Int8 CSeqDB::GetDiskUsage() const
     vector<string> extn;
     const bool is_protein(GetSequenceType() == CSeqDB::eProtein);
     SeqDB_GetFileExtensions(is_protein, extn, GetBlastDbVersion());
+    string blastdb_dirname;
 
     ITERATE(vector<string>, path, paths) {
         ITERATE(vector<string>, ext, extn) {
@@ -1480,6 +1481,37 @@ Int8 CSeqDB::GetDiskUsage() const
                 Int8 length = file.GetLength();
                 if (length != -1) {
                     retval += length;
+                    LOG_POST(Trace << "File " << file.GetPath() << " " << length << " bytes");
+                    blastdb_dirname = file.GetDir();
+                } else {
+                    ERR_POST(Error << "Error retrieving file size for "
+                                   << file.GetPath());
+                }
+            }
+        }
+    }
+    // For multi-volume databases, take into account files that apply to the
+    // entire BLASTDB
+    if (paths.size() > 1) {
+        _ASSERT( !blastdb_dirname.empty() );
+        auto dbname = GetDBNameList();
+        vector<string> dblist;
+        NStr::Split(dbname, " ", dblist, NStr::fSplit_Tokenize);
+        if (dblist.size() > 1) {
+            CNcbiOstrstream oss;
+            oss << "Cannot compute disk usage for multiple BLASTDBs (i.e.: '"
+                << dbname << "') at once. Please try again using one BLASTDB "
+                << "at a time.";
+            NCBI_THROW(CSeqDBException, eArgErr, CNcbiOstrstreamToString(oss));
+        }
+
+        for (const auto& ext: extn) {
+            CFile file(CDirEntry::MakePath(blastdb_dirname, dbname, ext));
+            if (file.Exists()) {
+                Int8 length = file.GetLength();
+                if (length != -1) {
+                    retval += length;
+                    LOG_POST(Trace << "File " << file.GetPath() << " " << length << " bytes");
                 } else {
                     ERR_POST(Error << "Error retrieving file size for "
                                    << file.GetPath());