Examples with AdminWaitSetRequest com.zimbra.soap.admin.message.AdminWaitSetRequest used on opensource projects

Example 6 with AdminWaitSetRequest

use of com.zimbra.soap.admin.message.AdminWaitSetRequest in project zm-mailbox by Zimbra.

the class ImapServerListener method restoreWaitSet.

private synchronized void restoreWaitSet() throws ServiceException {
    ZimbraLog.imap.debug("Attempting to restore admin waitset for all registered listeners.");
    if (wsID != null) {
        ZimbraLog.imap.debug("Another thread has already restored waitset.");
        // another thread has already restored waitset
        return;
    }
    cancelPendingRequest();
    AdminCreateWaitSetRequest req = new AdminCreateWaitSetRequest("all", false);
    checkAuth();
    AdminCreateWaitSetResponse resp;
    resp = soapProv.invokeJaxbAsAdminWithRetry(req, server);
    if (resp == null) {
        throw ServiceException.FAILURE("Received null response from AdminCreateWaitSetRequest", null);
    }
    wsID = resp.getWaitSetId();
    setWaitSetIdOnMailboxes();
    lastSequence.set(resp.getSequence());
    ZimbraLog.imap.debug("Created new waitset to replace lost or cancelled one. WaitSet ID: %s", wsID);
    // send non-blocking synchronous WaitSetRequest. This way the caller has certainty that listeners were added on remote server
    AdminWaitSetRequest waitSetReq = new AdminWaitSetRequest(wsID, lastSequence.toString());
    waitSetReq.setBlock(false);
    waitSetReq.setExpand(true);
    Enumeration<String> accountIds = this.sessionMap.keys();
    while (accountIds.hasMoreElements()) {
        String accountId = accountIds.nextElement();
        WaitSetAddSpec updateOrAdd = new WaitSetAddSpec();
        updateOrAdd.setId(accountId);
        Enumeration<Integer> folderIDs = this.sessionMap.get(accountId).keys();
        while (folderIDs.hasMoreElements()) {
            updateOrAdd.addFolderInterest(folderIDs.nextElement());
            ZimbraLog.imap.debug("Adding account %s to waitset %s", accountId, wsID);
            waitSetReq.addAddAccount(updateOrAdd);
        }
    }
    ZimbraLog.imap.debug("Sending initial AdminWaitSetRequest. WaitSet ID: %s", wsID);
    AdminWaitSetResponse wsResp = soapProv.invokeJaxbAsAdminWithRetry(waitSetReq, server);
    try {
        processAdminWaitSetResponse(wsResp);
    } catch (Exception e) {
        throw ServiceException.FAILURE("Failed to process initial AdminWaitSetResponse", e);
    }
}

Example 7 with AdminWaitSetRequest

use of com.zimbra.soap.admin.message.AdminWaitSetRequest in project zm-mailbox by Zimbra.

the class ImapServerListener method initWaitSet.

private synchronized void initWaitSet(String accountId, boolean alreadyListening) throws ServiceException {
    if (wsID == null && this.sessionMap.containsKey(accountId)) {
        AdminCreateWaitSetRequest req = new AdminCreateWaitSetRequest("all", false);
        checkAuth();
        AdminCreateWaitSetResponse resp;
        resp = soapProv.invokeJaxbAsAdminWithRetry(req, server);
        if (resp == null) {
            throw ServiceException.FAILURE("Received null response from AdminCreateWaitSetRequest", null);
        }
        wsID = resp.getWaitSetId();
        setWaitSetIdOnMailboxes();
        lastSequence.set(resp.getSequence());
    } else {
        cancelPendingRequest();
    }
    ZimbraLog.imap.debug("Current waitset ID is %s", wsID);
    // send non-blocking synchronous WaitSetRequest. This way the caller has certainty that listener was added on remote server
    AdminWaitSetRequest waitSetReq = new AdminWaitSetRequest(wsID, lastSequence.toString());
    waitSetReq.setBlock(false);
    waitSetReq.setExpand(true);
    if (!this.sessionMap.containsKey(accountId) || this.sessionMap.get(accountId).isEmpty()) {
        ZimbraLog.imap.debug("Removing accout %s from waitset %s", accountId, wsID);
        waitSetReq.addRemoveAccount(new Id(accountId));
    } else {
        WaitSetAddSpec updateOrAdd = new WaitSetAddSpec();
        updateOrAdd.setId(accountId);
        Enumeration<Integer> folderIDs = this.sessionMap.get(accountId).keys();
        while (folderIDs.hasMoreElements()) {
            updateOrAdd.addFolderInterest(folderIDs.nextElement());
        }
        if (alreadyListening) {
            ZimbraLog.imap.debug("Updating folder interests for account %s in waitset %s", accountId, wsID);
            waitSetReq.addUpdateAccount(updateOrAdd);
        } else {
            ZimbraLog.imap.debug("Adding account %s to waitset %s", accountId, wsID);
            waitSetReq.addAddAccount(updateOrAdd);
        }
    }
    try {
        ZimbraLog.imap.debug("Sending initial AdminWaitSetRequest. WaitSet ID: %s", wsID);
        AdminWaitSetResponse wsResp = soapProv.invokeJaxbAsAdminWithRetry(waitSetReq, server);
        processAdminWaitSetResponse(wsResp);
    } catch (SoapFaultException e) {
        if (AdminServiceException.NO_SUCH_WAITSET.equalsIgnoreCase(e.getCode())) {
            // waitset is gone. Create a new one
            ZimbraLog.imap.warn("AdminWaitSet %s does not exist anymore", wsID);
            wsID = null;
            unsetWaitSetIdOnMailboxes();
            lastSequence.set(0);
            restoreWaitSet();
        } else {
            throw ServiceException.FAILURE("Failed to process initial AdminWaitSetResponse", e);
        }
    } catch (Exception e) {
        throw ServiceException.FAILURE("Failed to process initial AdminWaitSetResponse", e);
    }
}

Example 8 with AdminWaitSetRequest

use of com.zimbra.soap.admin.message.AdminWaitSetRequest in project zm-mailbox by Zimbra.

the class TestWaitSetRequest method adminWSfolderInterestChanges.

/**
 * Test that can change the info on which folders we're interested in several times.
 * Introduced for ZCS-2220 although discovered that shouldn't need ADMIN SOAP sessions
 * for IMAP AdminWaitSet code, so counting them is a bit redundant.  Kept test
 * as some assurance that lots of folder interest changes works OK.
 */
@Test
public void adminWSfolderInterestChanges() throws Exception {
    int numFolders = LC.zimbra_session_limit_admin.intValue() + 15;
    String user1Name = testInfo.getMethodName().toLowerCase() + "user1";
    acc1 = TestUtil.createAccount(user1Name);
    ZMailbox zMbox1 = TestUtil.getZMailbox(user1Name);
    Set<String> accountIds = new HashSet<String>();
    String acct1Id = zMbox1.getAccountId();
    accountIds.add(acct1Id);
    String adminAuthToken = TestUtil.getAdminSoapTransport().getAuthToken().getValue();
    List<ZFolder> folders = new ArrayList<>(numFolders);
    for (int fnum = 1; fnum <= numFolders; fnum++) {
        folders.add(TestUtil.createFolder(zMbox1, String.format("FOLDER_%s", fnum)));
    }
    int numSess = countActiveSessionsForAdmin();
    SessionCache.getAllSessions(acct1Id);
    AdminCreateWaitSetResponse resp = createAdminWaitSet(accountIds, adminAuthToken, false);
    assertNotNull(resp);
    waitSetId = resp.getWaitSetId();
    int seq = resp.getSequence();
    for (int fnum = 0; fnum < numFolders; fnum++) {
        AdminWaitSetRequest waitSet = new AdminWaitSetRequest(waitSetId, Integer.toString(seq));
        Set<Integer> folderInterest = Sets.newHashSet();
        folderInterest.add(folders.get(fnum).getFolderIdInOwnerMailbox());
        waitSet.addUpdateAccount(createWaitSetAddSpec(acct1Id, folderInterest));
        AdminWaitSetResponse wsResp = sendReq(envelope(adminAuthToken, jaxbToString(waitSet), "urn:zimbra"), TestUtil.getAdminSoapUrl() + "AdminWaitSetRequest");
        seq = Integer.parseInt(wsResp.getSeqNo());
        assertEquals("Number of ADMIN sessions (should not have increased)", numSess, countActiveSessionsForAdmin());
    }
}

Example 9 with AdminWaitSetRequest

use of com.zimbra.soap.admin.message.AdminWaitSetRequest in project zm-mailbox by Zimbra.

the class TestWaitSetRequest method testBlockingAdminWait1Account.

@Test
public void testBlockingAdminWait1Account() throws Exception {
    ZimbraLog.test.info("Starting testBlockingAdminWait1Account");
    Set<String> accountIds = new HashSet<String>();
    String user1Name = "testBlockingAdminWait1Account_user1";
    String user2Name = "testBlockingAdminWait1Account_user2";
    acc1 = TestUtil.createAccount(user1Name);
    acc2 = TestUtil.createAccount(user2Name);
    ZMailbox mbox = TestUtil.getZMailbox(user1Name);
    accountIds.add(mbox.getAccountId());
    ZMailbox mbox2 = TestUtil.getZMailbox(user2Name);
    String adminAuthToken = TestUtil.getAdminSoapTransport().getAuthToken().getValue();
    AdminCreateWaitSetResponse resp = createAdminWaitSet(accountIds, adminAuthToken, false);
    assertNotNull(resp);
    waitSetId = resp.getWaitSetId();
    assertNotNull(waitSetId);
    int seq = resp.getSequence();
    AdminWaitSetRequest waitSetReq = new AdminWaitSetRequest(waitSetId, Integer.toString(seq));
    waitSetReq.setBlock(true);
    final CountDownLatch doneSignal = new CountDownLatch(1);
    waitForAccounts(Arrays.asList(mbox.getAccountId()), doneSignal, waitSetReq, "testBlockingAdminWait1Account", false);
    String subject = NAME_PREFIX + " test wait set request 1";
    TestUtil.addMessage(mbox, subject);
    TestUtil.addMessage(mbox2, subject);
    try {
        doneSignal.await(5, TimeUnit.SECONDS);
    } catch (Exception e) {
        fail("Wait interrupted.");
    }
    assertTrue("callback was not triggered.", cbCalled.get());
    assertTrue(failureMessage, success.get());
}