// LockKeyspace will lock the keyspace in the topology server. // UnlockKeyspace should be called if this returns no error. func (n *ActionNode) LockKeyspace(ctx context.Context, ts topo.Server, keyspace string, lockTimeout time.Duration, interrupted chan struct{}) (lockPath string, err error) { log.Infof("Locking keyspace %v for action %v", keyspace, n.Action) span := trace.NewSpanFromContext(ctx) span.StartClient("TopoServer.LockKeyspaceForAction") span.Annotate("action", n.Action) span.Annotate("keyspace", keyspace) defer span.Finish() return ts.LockKeyspaceForAction(keyspace, n.ToJson(), lockTimeout, interrupted) }
func CheckKeyspaceLock(t *testing.T, ts topo.Server) { if err := ts.CreateKeyspace("test_keyspace", &topo.Keyspace{}); err != nil { t.Fatalf("CreateKeyspace: %v", err) } interrupted := make(chan struct{}, 1) lockPath, err := ts.LockKeyspaceForAction("test_keyspace", "fake-content", 5*time.Second, interrupted) if err != nil { t.Fatalf("LockKeyspaceForAction: %v", err) } // test we can't take the lock again if _, err := ts.LockKeyspaceForAction("test_keyspace", "unused-fake-content", time.Second/10, interrupted); err != topo.ErrTimeout { t.Errorf("LockKeyspaceForAction(again): %v", err) } // test we can interrupt taking the lock go func() { time.Sleep(time.Second / 10) close(interrupted) }() if _, err := ts.LockKeyspaceForAction("test_keyspace", "unused-fake-content", 5*time.Second, interrupted); err != topo.ErrInterrupted { t.Errorf("LockKeyspaceForAction(interrupted): %v", err) } if err := ts.UnlockKeyspaceForAction("test_keyspace", lockPath, "fake-results"); err != nil { t.Errorf("UnlockKeyspaceForAction(): %v", err) } // test we can't unlock again if err := ts.UnlockKeyspaceForAction("test_keyspace", lockPath, "fake-results"); err == nil { t.Error("UnlockKeyspaceForAction(again) worked") } // test we can't lock a non-existing keyspace interrupted = make(chan struct{}, 1) if _, err := ts.LockKeyspaceForAction("test_keyspace_666", "fake-content", 5*time.Second, interrupted); err == nil { t.Fatalf("LockKeyspaceForAction(test_keyspace_666) worked for non-existing keyspace") } }