SERVER-44167 Added ability to kill operations by key

There are two patches here really. One of which makes killOp fast to use
and visible. The other adds OperationKey to various places and maps it
to an internal OpId.

1 files changed, 18 insertions, 3 deletions

diff --git a/src/mongo/db/operation_context.h b/src/mongo/db/operation_context.h
index 8fc158a7b81..c1e224bfee7 100644
--- a/src/mongo/db/operation_context.h
+++ b/src/mongo/db/operation_context.h
@@ -79,7 +79,7 @@ class OperationContext : public Interruptible, public Decorable<OperationContext
     OperationContext& operator=(const OperationContext&) = delete;
 
 public:
-    OperationContext(Client* client, unsigned int opId);
+    OperationContext(Client* client, OperationId opId);
     virtual ~OperationContext();
 
     bool shouldParticipateInFlowControl() const {
@@ -166,11 +166,25 @@ public:
     /**
      * Returns the operation ID associated with this operation.
      */
-    unsigned int getOpID() const {
+    OperationId getOpID() const {
         return _opId;
     }
 
     /**
+     * Returns the operation UUID associated with this operation or boost::none.
+     */
+    const boost::optional<OperationKey>& getOperationKey() const {
+        return _opKey;
+    }
+
+    /**
+     * Sets the operation UUID associated with this operation.
+     *
+     * This function may only be called once per OperationContext.
+     */
+    void setOperationKey(OperationKey opKey);
+
+    /**
      * Returns the session ID associated with this operation, if there is one.
      */
     const boost::optional<LogicalSessionId>& getLogicalSessionId() const {
@@ -490,7 +504,8 @@ private:
 
     Client* const _client;
 
-    const unsigned int _opId;
+    const OperationId _opId;
+    boost::optional<OperationKey> _opKey;
 
     boost::optional<LogicalSessionId> _lsid;
     boost::optional<TxnNumber> _txnNumber;