org.grails.datastore.mapping.cassandra.CassandraSession.java Source code

Introduction

Here is the source code for org.grails.datastore.mapping.cassandra.CassandraSession.java

Source

/* Copyright (C) 2010 SpringSource
 *
 * 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 org.grails.datastore.mapping.cassandra;

import me.prettyprint.cassandra.service.CassandraClient;
import me.prettyprint.cassandra.service.CassandraClientPool;

import org.springframework.dao.DataAccessResourceFailureException;
import org.grails.datastore.mapping.cassandra.engine.CassandraEntityPersister;
import org.grails.datastore.mapping.core.AbstractSession;
import org.grails.datastore.mapping.core.Datastore;
import org.grails.datastore.mapping.engine.Persister;
import org.grails.datastore.mapping.model.MappingContext;
import org.grails.datastore.mapping.model.PersistentEntity;
import org.grails.datastore.mapping.transactions.Transaction;
import org.springframework.transaction.TransactionSystemException;

/**
 * @author Graeme Rocher
 * @since 1.0
 */
public class CassandraSession extends AbstractSession<CassandraClient> {
    private CassandraClient cassandraClient;
    private CassandraClientPool connectionPool;

    public CassandraSession(Datastore ds, MappingContext context, CassandraClientPool connectionPool,
            CassandraClient client) {
        super(ds, context);
        this.connectionPool = connectionPool;
        this.cassandraClient = client;
    }

    @Override
    protected Persister createPersister(Class cls, MappingContext mappingContext) {
        PersistentEntity entity = mappingContext.getPersistentEntity(cls.getName());
        if (entity != null) {
            return new CassandraEntityPersister(mappingContext, entity, this, cassandraClient);
        }
        return null;
    }

    public boolean isConnected() {
        return !cassandraClient.isReleased();
    }

    @Override
    public void disconnect() {
        try {
            connectionPool.releaseClient(cassandraClient);
        } catch (Exception e) {
            throw new DataAccessResourceFailureException(
                    "Failed to release Cassandra client session: " + e.getMessage(), e);
        } finally {
            super.disconnect();
        }
    }

    @Override
    protected Transaction beginTransactionInternal() {
        throw new TransactionSystemException("Transactions are not supported by Cassandra");
    }

    public CassandraClient getNativeInterface() {
        return cassandraClient;
    }
}