com.marklogic.client.impl.TransactionImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2016 MarkLogic Corporation
 *
 * 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 com.marklogic.client.impl;

import com.marklogic.client.FailedRequestException;
import com.marklogic.client.ForbiddenUserException;
import com.marklogic.client.Transaction;
import com.marklogic.client.io.marker.StructureReadHandle;

import javax.ws.rs.core.NewCookie;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;

class TransactionImpl implements Transaction {
	final static int DEFAULT_TIMELIMIT = -1;

	private RESTServices    services;
	private String          transactionId;
	private String          hostId;
	// we keep cookies scoped with each tranasaction to work with load balancers
	// that need to keep requests for one transaction on a specific MarkLogic Server host
	private List cookies = new ArrayList();
	private Calendar        created = Calendar.getInstance();

	TransactionImpl(RESTServices services, String transactionId, List cookies) {
		this.services      = services;
		this.transactionId = transactionId;
		if ( cookies != null ) {
			for ( NewCookie cookie : cookies ) {
				// make a clone to ensure we're not holding on to any resources
				// related to an HTTP connection that need to be released
				this.cookies.add(new NewCookie(cookie));
				if ( "HostId".equalsIgnoreCase(cookie.getName()) ) {
					hostId =  cookie.getValue();
				}
			}
		}
	}

	public String getTransactionId() {
		return transactionId;
	}
	public void setTransactionId(String transactionId) {
		this.transactionId = transactionId;
	}

	public List getCookies() {
		return cookies;
	}

	public String getHostId() {
		return hostId;
	}
	protected void setHostId(String hostId) {
		this.hostId = hostId;
	}

	public Calendar getCreatedTimestamp() {
		return created;
	}

	@Override
    @SuppressWarnings({ "rawtypes", "unchecked" })
	public  T readStatus(T handle) throws ForbiddenUserException, FailedRequestException {
		if (handle == null)
			throw new IllegalArgumentException("reading transaction status with null handle");

		HandleImplementation handleBase = HandleAccessor.checkHandle(handle, "structure");

		handleBase.receiveContent(
				services.getValue(
						null,
						"transactions",
						getTransactionId(),
						false,
						handleBase.getMimetype(),
						handleBase.receiveAs()
						)
				);

		return handle;
	}

	@Override
	public void commit() throws ForbiddenUserException, FailedRequestException {
		services.commitTransaction(this);
	}

	@Override
	public void rollback() throws ForbiddenUserException, FailedRequestException {
		services.rollbackTransaction(this);
	}

}