support ignoring workload lock in RawEngine operation (#636)

* support ignoring workload lock in RawEngine operation

* add UT

cluster/calcium/raw_engine.go

@@ -15,10 +15,18 @@ func (c *Calcium) RawEngine(ctx context.Context, opts *types.RawEngineOptions) (
  wg.Add(1)
  _ = c.pool.Invoke(func() {
  defer wg.Done()
- err = c.withWorkloadLocked(ctx, ID, func(ctx context.Context, workload *types.Workload) error {
+ if opts.IgnoreLock {
+ var workload *types.Workload
+ if workload, err = c.store.GetWorkload(ctx, ID); err != nil {
+ return
+ }
  msg, err = workload.RawEngine(ctx, opts)
- return err
- })
+ } else {
+ err = c.withWorkloadLocked(ctx, ID, func(ctx context.Context, workload *types.Workload) error {
+ msg, err = workload.RawEngine(ctx, opts)
+ return err
+ })
+ }
 
  if err == nil {
  logger.Infof(ctx, "Workload %s", ID)

cluster/calcium/raw_engine_test.go

@@ -32,3 +32,19 @@ func TestRawEngine(t *testing.T) {
  _, err := c.RawEngine(ctx, &types.RawEngineOptions{ID: "id1", Op: "xxxx"})
  assert.NoError(t, err)
 }
+
+func TestRawEngineIgnoreLock(t *testing.T) {
+ c := NewTestCluster()
+ ctx := context.Background()
+ store := c.store.(*storemocks.Store)
+ workload := &types.Workload{
+ ID: "id1",
+ Privileged: true,
+ }
+ engine := &enginemocks.API{}
+ workload.Engine = engine
+ store.On("GetWorkload", mock.Anything, mock.Anything).Return(workload, nil)
+ engine.On("RawEngine", mock.Anything, mock.Anything).Return(&enginetypes.RawEngineResult{}, nil).Once()
+ _, err := c.RawEngine(ctx, &types.RawEngineOptions{ID: "id1", Op: "xxxx", IgnoreLock: true})
+ assert.NoError(t, err)
+}