strolch/li.strolch.persistence.xml/src/main/java/li/strolch/persistence/xml/XmlAuditDao.java

/*
 * Copyright 2013 Robert von Burg <eitch@eitchnet.ch>
 * 
 * Licensed 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 li.strolch.persistence.xml;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import li.strolch.model.Tags;
import li.strolch.model.audit.Audit;
import li.strolch.model.audit.AuditQuery;
import li.strolch.model.audit.AuditVisitor;
import li.strolch.persistence.api.AuditDao;
import li.strolch.persistence.api.StrolchTransaction;
import ch.eitchnet.utils.collections.DateRange;
import ch.eitchnet.xmlpers.api.PersistenceContext;
import ch.eitchnet.xmlpers.api.PersistenceTransaction;
import ch.eitchnet.xmlpers.objref.IdOfSubTypeRef;
import ch.eitchnet.xmlpers.objref.ObjectRef;
import ch.eitchnet.xmlpers.objref.SubTypeRef;
import ch.eitchnet.xmlpers.objref.TypeRef;

/**
 * @author Robert von Burg <eitch@eitchnet.ch>
 */
public class XmlAuditDao implements AuditDao {

	private PersistenceTransaction tx;

	public XmlAuditDao(StrolchTransaction tx) {
		XmlStrolchTransaction strolchTx = (XmlStrolchTransaction) tx;
		this.tx = strolchTx.getTx();
	}

	protected String getClassType() {
		return Tags.AUDIT;
	}

	protected IdOfSubTypeRef getIdRef(String type, Long id) {
		IdOfSubTypeRef idRef = this.tx.getObjectRefCache().getIdOfSubTypeRef(getClassType(), type, id.toString());
		return idRef;
	}

	protected SubTypeRef getTypeRef(String type) {
		SubTypeRef typeRef = this.tx.getObjectRefCache().getSubTypeRef(getClassType(), type);
		return typeRef;
	}

	@Override
	public boolean hasElement(String type, Long id) {
		IdOfSubTypeRef ref = getIdRef(type, id);
		return this.tx.getObjectDao().hasElement(ref);
	}

	@Override
	public long querySize(DateRange dateRange) {
		long size = 0;
		Set<String> types = queryTypes();
		for (String type : types) {
			// SubTypeRef subTypeRef = getTypeRef(type);
			// size += this.tx.getMetadataDao().querySize(subTypeRef);

			size += querySize(type, dateRange);
		}
		return size;
	}

	@Override
	public long querySize(String type, DateRange dateRange) {
		long size = 0;

		// TODO re-think this nonsense... this might have a huge performance penalty
		SubTypeRef subTypeRef = getTypeRef(type);
		Set<String> keySet = this.tx.getMetadataDao().queryKeySet(subTypeRef);
		for (String key : keySet) {
			ObjectRef objectRef = subTypeRef.getChildIdRef(this.tx, key);
			Audit audit = this.tx.getObjectDao().queryById(objectRef);
			if (dateRange.contains(audit.getDate()))
				size++;
		}

		// return this.tx.getMetadataDao().querySize(subTypeRef);
		return size;
	}

	@Override
	public Set<String> queryTypes() {
		TypeRef typeRef = this.tx.getObjectRefCache().getTypeRef(getClassType());
		Set<String> types = this.tx.getMetadataDao().queryTypeSet(typeRef);
		return types;
	}

	@Override
	public Audit queryBy(String type, Long id) {
		Audit t = this.tx.getObjectDao().queryById(getIdRef(type, id));
		return t;
	}

	@Override
	public List<Audit> queryAll(String type, DateRange dateRange) {

		List<Audit> audits = new ArrayList<>();

		SubTypeRef subTypeRef = getTypeRef(type);
		Set<String> keySet = this.tx.getMetadataDao().queryKeySet(subTypeRef);
		for (String key : keySet) {
			ObjectRef objectRef = subTypeRef.getChildIdRef(this.tx, key);
			Audit audit = this.tx.getObjectDao().queryById(objectRef);
			if (dateRange.contains(audit.getDate()))
				audits.add(audit);
		}

		// this.tx.getObjectDao().queryAll(getTypeRef(type));
		return audits;
	}

	@Override
	public void save(Audit audit) {
		PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
		this.tx.getFileDao().performCreate(ctx);
	}

	@Override
	public void saveAll(List<Audit> audits) {
		for (Audit audit : audits) {
			PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
			this.tx.getFileDao().performCreate(ctx);
		}
	}

	@Override
	public void update(Audit audit) {
		PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
		this.tx.getFileDao().performUpdate(ctx);
	}

	@Override
	public void updateAll(List<Audit> audits) {
		for (Audit audit : audits) {
			PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
			this.tx.getFileDao().performUpdate(ctx);
		}
	}

	@Override
	public void remove(Audit audit) {
		PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
		this.tx.getFileDao().performDelete(ctx);
	}

	@Override
	public void removeAll(List<Audit> audits) {
		for (Audit audit : audits) {
			PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(audit);
			this.tx.getFileDao().performDelete(ctx);
		}
	}

	@Override
	public long removeAll(String type, DateRange dateRange) {

		long removed = 0L;

		SubTypeRef subTypeRef = getTypeRef(type);
		Set<String> keySet = this.tx.getMetadataDao().queryKeySet(subTypeRef);
		for (String key : keySet) {
			ObjectRef objectRef = subTypeRef.getChildIdRef(this.tx, key);
			Audit audit = this.tx.getObjectDao().queryById(objectRef);
			if (dateRange.contains(audit.getDate())) {
				PersistenceContext<Audit> ctx = this.tx.getObjectDao().createCtx(objectRef);
				this.tx.getFileDao().performDelete(ctx);
				removed++;
			}
		}

		return removed;
	}

	@Override
	public <U> List<U> doQuery(AuditQuery query, AuditVisitor<U> auditVisitor) {
		// TODO implement XML file based querying...
		throw new UnsupportedOperationException("not yet implemented!");
	}
}