workqueue: Actually implement GetKeyState in real server (#1111)

This was accidentally omitted from the rest of the implementation.

Author: wlynch

modules/workqueue/cmd/inmem/main.go

@@ -102,3 +102,7 @@ func (y *enq) Process(ctx context.Context, req *workqueue.ProcessRequest) (*work
 	}
 	return &workqueue.ProcessResponse{}, nil
 }
+
+func (y *enq) GetKeyState(ctx context.Context, req *workqueue.GetKeyStateRequest) (*workqueue.KeyState, error) {
+	return y.wq.Get(ctx, req.Key)
+}

modules/workqueue/cmd/receiver/main.go

@@ -89,3 +89,7 @@ func (y *enq) Process(ctx context.Context, req *workqueue.ProcessRequest) (*work
 	}
 	return &workqueue.ProcessResponse{}, nil
 }
+
+func (y *enq) GetKeyState(ctx context.Context, req *workqueue.GetKeyStateRequest) (*workqueue.KeyState, error) {
+	return y.wq.Get(ctx, req.Key)
+}