io.pivotal.arca.provider.DatabaseProvider Maven / Gradle / Ivy

Go to download
/* 
 * Copyright (C) 2014 Pivotal Software, Inc. 
 * 
 * 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 io.pivotal.arca.provider;

import android.content.ContentProviderOperation;
import android.content.ContentProviderResult;
import android.content.OperationApplicationException;
import android.database.sqlite.SQLiteDatabase;
import android.net.Uri;

import io.pivotal.arca.provider.DatabaseConfiguration.DefaultDatabaseConfiguration;

import java.util.ArrayList;
import java.util.Collection;

public abstract class DatabaseProvider extends DatasetProvider {

	private volatile SQLiteDatabase mDatabase;

	protected SQLiteDatabase getDatabase() {
		synchronized (this) {
			if (mDatabase == null) {
				mDatabase = createDatabase();
			}
		}
		return mDatabase;
	}

	protected void closeDatabase() {
		synchronized (this) {
			if (mDatabase != null) {
				mDatabase.close();
				mDatabase = null;
			}
		}
	}

	protected SQLiteDatabase createDatabase() {
		final DatabaseConfiguration configuration = onCreateDatabaseConfiguration();
		final DatabaseHelper helper = DatabaseHelper.create(getContext(), configuration, getSQLiteDatasets());
		return helper.getWritableDatabase();
	}

	public DatabaseConfiguration onCreateDatabaseConfiguration() {
		return new DefaultDatabaseConfiguration(getContext());
	}

	private Collection getSQLiteDatasets() {
		final Collection datasets = getDatasets();
		return CollectionUtils.filter(datasets, SQLiteDataset.class);
	}

	@Override
	protected Dataset getDatasetOrThrowException(final Uri uri) {
		final Dataset dataset = super.getDatasetOrThrowException(uri);
		if (dataset instanceof SQLiteDataset) {
			final SQLiteDatabase database = getDatabase();
			((SQLiteDataset) dataset).setDatabase(database);
		}
		return dataset;
	}

	@Override
	public ContentProviderResult[] applyBatch(final ArrayList operations) throws OperationApplicationException {
		final SQLiteDatabase database = getDatabase();
		database.beginTransaction();
		try {
			final ContentProviderResult[] results = super.applyBatch(operations);
			database.setTransactionSuccessful();
			return results;
		} finally {
			database.endTransaction();
		}
	}
}