001/* 002 * (C) Copyright 2006-2012 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * <a href="mailto:tdelprat@nuxeo.com">Tiry</a> 018 * <a href="mailto:tmartins@nuxeo.com">Thierry Martins</a> 019 */ 020 021package org.nuxeo.ecm.quota.size; 022 023import static org.nuxeo.ecm.quota.size.QuotaAwareDocument.DOCUMENTS_SIZE_STATISTICS_FACET; 024 025import org.nuxeo.ecm.core.api.DocumentModel; 026import org.nuxeo.ecm.core.api.adapter.DocumentAdapterFactory; 027import org.nuxeo.ecm.quota.QuotaUtils; 028 029/** 030 * Simple factory for {@link QuotaAwareDocument} document model adapter 031 * 032 * @author <a href="mailto:tdelprat@nuxeo.com">Tiry</a> 033 * @since 5.6 034 */ 035public class QuotaAwareDocumentFactory implements DocumentAdapterFactory { 036 037 public static QuotaAware make(DocumentModel doc) { 038 if (!doc.hasFacet(DOCUMENTS_SIZE_STATISTICS_FACET)) { 039 doc.addFacet(DOCUMENTS_SIZE_STATISTICS_FACET); 040 } 041 return new QuotaAwareDocument(doc); 042 } 043 044 public static void unmake(DocumentModel doc) { 045 if (doc.hasFacet(DOCUMENTS_SIZE_STATISTICS_FACET)) { 046 new QuotaAwareDocument(doc).resetInfos(); 047 saveDocument(doc); // first save to flush the reset 048 doc.removeFacet(DOCUMENTS_SIZE_STATISTICS_FACET); 049 saveDocument(doc); // second save to flush facet removal 050 } 051 } 052 053 protected static DocumentModel saveDocument(DocumentModel doc) { 054 doc.putContextData(DocumentsSizeUpdater.DISABLE_QUOTA_CHECK_LISTENER, Boolean.TRUE); 055 // set flags to disable listeners 056 QuotaUtils.disableListeners(doc); 057 DocumentModel origDoc = doc; 058 doc = doc.getCoreSession().saveDocument(doc); 059 // remove flags as they could be kept in the document for a long time otherwise 060 QuotaUtils.clearContextData(doc); 061 // also remove flags from original doc, which the caller still references 062 QuotaUtils.clearContextData(origDoc); 063 return doc; 064 } 065 066 @Override 067 public Object getAdapter(DocumentModel doc, Class<?> adapter) { 068 if (doc.hasFacet(DOCUMENTS_SIZE_STATISTICS_FACET)) { 069 return adapter.cast(new QuotaAwareDocument(doc)); 070 } 071 return null; 072 } 073}