JAMES-1982 Add a quota probe
Project: http://git-wip-us.apache.org/repos/asf/james-project/repo Commit: http://git-wip-us.apache.org/repos/asf/james-project/commit/f41a3cf5 Tree: http://git-wip-us.apache.org/repos/asf/james-project/tree/f41a3cf5 Diff: http://git-wip-us.apache.org/repos/asf/james-project/diff/f41a3cf5 Branch: refs/heads/master Commit: f41a3cf550af35f35856dc2641dfb4f818cc75d4 Parents: 336d02e Author: benwa <btell...@linagora.com> Authored: Fri Mar 31 19:09:40 2017 +0700 Committer: benwa <btell...@linagora.com> Committed: Mon Apr 3 18:09:32 2017 +0700 ---------------------------------------------------------------------- .../org/apache/james/modules/MailboxModule.java | 4 +- .../apache/james/modules/QuotaProbesImpl.java | 102 +++++++++++++++++++ 2 files changed, 105 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/james-project/blob/f41a3cf5/server/container/guice/mailbox/src/main/java/org/apache/james/modules/MailboxModule.java ---------------------------------------------------------------------- diff --git a/server/container/guice/mailbox/src/main/java/org/apache/james/modules/MailboxModule.java b/server/container/guice/mailbox/src/main/java/org/apache/james/modules/MailboxModule.java index b342aac..48a9077 100644 --- a/server/container/guice/mailbox/src/main/java/org/apache/james/modules/MailboxModule.java +++ b/server/container/guice/mailbox/src/main/java/org/apache/james/modules/MailboxModule.java @@ -27,7 +27,9 @@ public class MailboxModule extends AbstractModule { @Override protected void configure() { - Multibinder.newSetBinder(binder(), GuiceProbe.class).addBinding().to(MailboxProbeImpl.class); + Multibinder<GuiceProbe> probeMultiBinder = Multibinder.newSetBinder(binder(), GuiceProbe.class); + probeMultiBinder.addBinding().to(MailboxProbeImpl.class); + probeMultiBinder.addBinding().to(QuotaProbesImpl.class); } } http://git-wip-us.apache.org/repos/asf/james-project/blob/f41a3cf5/server/container/guice/mailbox/src/main/java/org/apache/james/modules/QuotaProbesImpl.java ---------------------------------------------------------------------- diff --git a/server/container/guice/mailbox/src/main/java/org/apache/james/modules/QuotaProbesImpl.java b/server/container/guice/mailbox/src/main/java/org/apache/james/modules/QuotaProbesImpl.java new file mode 100644 index 0000000..2277f1a --- /dev/null +++ b/server/container/guice/mailbox/src/main/java/org/apache/james/modules/QuotaProbesImpl.java @@ -0,0 +1,102 @@ +/**************************************************************** + * Licensed to the Apache Software Foundation (ASF) under one * + * or more contributor license agreements. See the NOTICE file * + * distributed with this work for additional information * + * regarding copyright ownership. The ASF licenses this file * + * to you under the Apache License, Version 2.0 (the * + * "License"); you may not use this file except in compliance * + * with the License. You may obtain a copy of the License at * + * * + * http://www.apache.org/licenses/LICENSE-2.0 * + * * + * Unless required by applicable law or agreed to in writing, * + * software distributed under the License is distributed on an * + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * + * KIND, either express or implied. See the License for the * + * specific language governing permissions and limitations * + * under the License. * + ****************************************************************/ + +package org.apache.james.modules; + +import javax.inject.Inject; + +import org.apache.james.mailbox.exception.MailboxException; +import org.apache.james.mailbox.model.MailboxPath; +import org.apache.james.mailbox.quota.MaxQuotaManager; +import org.apache.james.mailbox.quota.QuotaManager; +import org.apache.james.mailbox.quota.QuotaRootResolver; +import org.apache.james.mailbox.store.mail.model.SerializableQuota; +import org.apache.james.mailbox.store.probe.QuotaProbe; +import org.apache.james.utils.GuiceProbe; + +public class QuotaProbesImpl implements QuotaProbe, GuiceProbe { + + private final MaxQuotaManager maxQuotaManager; + private final QuotaRootResolver quotaRootResolver; + private final QuotaManager quotaManager; + + @Inject + public QuotaProbesImpl(MaxQuotaManager maxQuotaManager, QuotaRootResolver quotaRootResolver, QuotaManager quotaManager) { + this.maxQuotaManager = maxQuotaManager; + this.quotaRootResolver = quotaRootResolver; + this.quotaManager = quotaManager; + } + + @Override + public String getQuotaRoot(String namespace, String user, String name) throws MailboxException { + return quotaRootResolver.getQuotaRoot(new MailboxPath(namespace, user, name)).getValue(); + } + + @Override + public SerializableQuota getMessageCountQuota(String quotaRoot) throws MailboxException { + return new SerializableQuota(quotaManager.getMessageQuota(quotaRootResolver.createQuotaRoot(quotaRoot))); + } + + @Override + public SerializableQuota getStorageQuota(String quotaRoot) throws MailboxException { + return new SerializableQuota(quotaManager.getStorageQuota(quotaRootResolver.createQuotaRoot(quotaRoot))); + } + + @Override + public long getMaxMessageCount(String quotaRoot) throws MailboxException { + return maxQuotaManager.getMaxMessage(quotaRootResolver.createQuotaRoot(quotaRoot)); + } + + @Override + public long getMaxStorage(String quotaRoot) throws MailboxException { + return maxQuotaManager.getMaxStorage(quotaRootResolver.createQuotaRoot(quotaRoot)); + } + + @Override + public long getDefaultMaxMessageCount() throws MailboxException { + return maxQuotaManager.getDefaultMaxMessage(); + } + + @Override + public long getDefaultMaxStorage() throws MailboxException { + return maxQuotaManager.getDefaultMaxStorage(); + } + + @Override + public void setMaxMessageCount(String quotaRoot, long maxMessageCount) throws MailboxException { + maxQuotaManager.setMaxMessage(quotaRootResolver.createQuotaRoot(quotaRoot), + maxMessageCount); + } + + @Override + public void setMaxStorage(String quotaRoot, long maxSize) throws MailboxException { + maxQuotaManager.setMaxStorage(quotaRootResolver.createQuotaRoot(quotaRoot), + maxSize); + } + + @Override + public void setDefaultMaxMessageCount(long maxDefaultMessageCount) throws MailboxException { + maxQuotaManager.setDefaultMaxMessage(maxDefaultMessageCount); + } + + @Override + public void setDefaultMaxStorage(long maxDefaultSize) throws MailboxException { + maxQuotaManager.setDefaultMaxStorage(maxDefaultSize); + } +} --------------------------------------------------------------------- To unsubscribe, e-mail: server-dev-unsubscr...@james.apache.org For additional commands, e-mail: server-dev-h...@james.apache.org
