limiting original UNION query results to only 1 result (potentially speeding things up in some cases)

stamparm · stamparm · commit 64c241fe922e · 2012-05-08T13:45:53.000Z
diff --git a/lib/core/agent.py b/lib/core/agent.py
@@ -582,7 +582,12 @@ def forgeInbandQuery(self, query, position, count, comment, prefix, suffix, char
         if query.startswith("SELECT "):
             query = query[len("SELECT "):]
 
-        inbandQuery = self.prefixQuery("UNION ALL SELECT ", prefix=prefix)
+        limitOriginal = ""
+        if where == PAYLOAD.WHERE.ORIGINAL:
+            if Backend.getIdentifiedDbms() in (DBMS.MYSQL, ):
+                limitOriginal = "%s " % (queries[Backend.getIdentifiedDbms()].limit.query % (1, 1))
+
+        inbandQuery = self.prefixQuery("%sUNION ALL SELECT " % limitOriginal, prefix=prefix)
 
         if limited:
             inbandQuery += ",".join(char if _ != position else '(SELECT %s)' % query for _ in xrange(0, count))
