MXS-2345 Query classifier collects db of USE stmt

qc_get_database_names() will now return the database used in a USE statement.
2019-02-22 11:22:32 +02:00
parent c1d96d79f0
commit 675971bd3c
2 changed files with 67 additions and 36 deletions
--- a/query_classifier/qc_sqlite/qc_sqlite.cc
+++ b/query_classifier/qc_sqlite/qc_sqlite.cc
@ -3047,6 +3047,13 @@ public:
        m_status = QC_QUERY_PARSED;
        m_type_mask = QUERY_TYPE_SESSION_WRITE;
        m_operation = QUERY_OP_CHANGE_DB;
+
+        if (should_collect(QC_COLLECT_DATABASES))
+        {
+            char* zCopy = MXS_STRNDUP_A(pToken->z, pToken->n);
+
+            m_database_names.push_back(zCopy);
+        }
    }

    void set_type_mask(uint32_t type_mask)