mirror of
https://github.com/berkeleydb/je.git
synced 2024-09-09 12:06:12 +00:00
319 lines
11 KiB
Java
319 lines
11 KiB
Java
|
/*-
|
|||
|
* Copyright (C) 2002, 2017, Oracle and/or its affiliates. All rights reserved.
|
|||
|
*
|
|||
|
* This file was distributed by Oracle as part of a version of Oracle Berkeley
|
|||
|
* DB Java Edition made available at:
|
|||
|
*
|
|||
|
* http://www.oracle.com/technetwork/database/database-technologies/berkeleydb/downloads/index.html
|
|||
|
*
|
|||
|
* Please see the LICENSE file included in the top-level directory of the
|
|||
|
* appropriate version of Oracle Berkeley DB Java Edition for a copy of the
|
|||
|
* license and additional information.
|
|||
|
*/
|
|||
|
|
|||
|
package collections.ship.tuple;
|
|||
|
|
|||
|
import java.io.File;
|
|||
|
|
|||
|
import com.sleepycat.bind.serial.ClassCatalog;
|
|||
|
import com.sleepycat.bind.serial.StoredClassCatalog;
|
|||
|
import com.sleepycat.bind.serial.TupleSerialKeyCreator;
|
|||
|
import com.sleepycat.bind.tuple.TupleInput;
|
|||
|
import com.sleepycat.bind.tuple.TupleOutput;
|
|||
|
import com.sleepycat.je.Database;
|
|||
|
import com.sleepycat.je.DatabaseConfig;
|
|||
|
import com.sleepycat.je.DatabaseException;
|
|||
|
import com.sleepycat.je.Environment;
|
|||
|
import com.sleepycat.je.EnvironmentConfig;
|
|||
|
import com.sleepycat.je.ForeignKeyDeleteAction;
|
|||
|
import com.sleepycat.je.SecondaryConfig;
|
|||
|
import com.sleepycat.je.SecondaryDatabase;
|
|||
|
|
|||
|
/**
|
|||
|
* SampleDatabase defines the storage containers, indices and foreign keys
|
|||
|
* for the sample database.
|
|||
|
*
|
|||
|
* @author Mark Hayes
|
|||
|
*/
|
|||
|
public class SampleDatabase {
|
|||
|
|
|||
|
private static final String CLASS_CATALOG = "java_class_catalog";
|
|||
|
private static final String SUPPLIER_STORE = "supplier_store";
|
|||
|
private static final String PART_STORE = "part_store";
|
|||
|
private static final String SHIPMENT_STORE = "shipment_store";
|
|||
|
private static final String SHIPMENT_PART_INDEX = "shipment_part_index";
|
|||
|
private static final String SHIPMENT_SUPPLIER_INDEX =
|
|||
|
"shipment_supplier_index";
|
|||
|
private static final String SUPPLIER_CITY_INDEX = "supplier_city_index";
|
|||
|
|
|||
|
private Environment env;
|
|||
|
private Database partDb;
|
|||
|
private Database supplierDb;
|
|||
|
private Database shipmentDb;
|
|||
|
private SecondaryDatabase supplierByCityDb;
|
|||
|
private SecondaryDatabase shipmentByPartDb;
|
|||
|
private SecondaryDatabase shipmentBySupplierDb;
|
|||
|
private StoredClassCatalog javaCatalog;
|
|||
|
|
|||
|
/**
|
|||
|
* Open all storage containers, indices, and catalogs.
|
|||
|
*/
|
|||
|
public SampleDatabase(String homeDirectory)
|
|||
|
throws DatabaseException {
|
|||
|
|
|||
|
// Open the Berkeley DB environment in transactional mode.
|
|||
|
//
|
|||
|
System.out.println("Opening environment in: " + homeDirectory);
|
|||
|
EnvironmentConfig envConfig = new EnvironmentConfig();
|
|||
|
envConfig.setTransactional(true);
|
|||
|
envConfig.setAllowCreate(true);
|
|||
|
env = new Environment(new File(homeDirectory), envConfig);
|
|||
|
|
|||
|
// Set the Berkeley DB config for opening all stores.
|
|||
|
//
|
|||
|
DatabaseConfig dbConfig = new DatabaseConfig();
|
|||
|
dbConfig.setTransactional(true);
|
|||
|
dbConfig.setAllowCreate(true);
|
|||
|
|
|||
|
// Create the Serial class catalog. This holds the serialized class
|
|||
|
// format for all database records of serial format.
|
|||
|
//
|
|||
|
Database catalogDb = env.openDatabase(null, CLASS_CATALOG, dbConfig);
|
|||
|
javaCatalog = new StoredClassCatalog(catalogDb);
|
|||
|
|
|||
|
// Open the Berkeley DB database for the part, supplier and shipment
|
|||
|
// stores. The stores are opened with no duplicate keys allowed.
|
|||
|
//
|
|||
|
partDb = env.openDatabase(null, PART_STORE, dbConfig);
|
|||
|
|
|||
|
supplierDb = env.openDatabase(null, SUPPLIER_STORE, dbConfig);
|
|||
|
|
|||
|
shipmentDb = env.openDatabase(null, SHIPMENT_STORE, dbConfig);
|
|||
|
|
|||
|
// Open the SecondaryDatabase for the city index of the supplier store,
|
|||
|
// and for the part and supplier indices of the shipment store.
|
|||
|
// Duplicate keys are allowed since more than one supplier may be in
|
|||
|
// the same city, and more than one shipment may exist for the same
|
|||
|
// supplier or part. A foreign key constraint is defined for the
|
|||
|
// supplier and part indices to ensure that a shipment only refers to
|
|||
|
// existing part and supplier keys. The CASCADE delete action means
|
|||
|
// that shipments will be deleted if their associated part or supplier
|
|||
|
// is deleted.
|
|||
|
//
|
|||
|
SecondaryConfig secConfig = new SecondaryConfig();
|
|||
|
secConfig.setTransactional(true);
|
|||
|
secConfig.setAllowCreate(true);
|
|||
|
secConfig.setSortedDuplicates(true);
|
|||
|
|
|||
|
secConfig.setKeyCreator(new SupplierByCityKeyCreator(javaCatalog,
|
|||
|
SupplierData.class));
|
|||
|
supplierByCityDb = env.openSecondaryDatabase(null, SUPPLIER_CITY_INDEX,
|
|||
|
supplierDb, secConfig);
|
|||
|
|
|||
|
secConfig.setForeignKeyDatabase(partDb);
|
|||
|
secConfig.setForeignKeyDeleteAction(ForeignKeyDeleteAction.CASCADE);
|
|||
|
secConfig.setKeyCreator(new ShipmentByPartKeyCreator(javaCatalog,
|
|||
|
ShipmentData.class));
|
|||
|
shipmentByPartDb = env.openSecondaryDatabase(null, SHIPMENT_PART_INDEX,
|
|||
|
shipmentDb, secConfig);
|
|||
|
|
|||
|
secConfig.setForeignKeyDatabase(supplierDb);
|
|||
|
secConfig.setForeignKeyDeleteAction(ForeignKeyDeleteAction.CASCADE);
|
|||
|
secConfig.setKeyCreator(new ShipmentBySupplierKeyCreator(javaCatalog,
|
|||
|
ShipmentData.class));
|
|||
|
shipmentBySupplierDb = env.openSecondaryDatabase(null,
|
|||
|
SHIPMENT_SUPPLIER_INDEX,
|
|||
|
shipmentDb, secConfig);
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the storage environment for the database.
|
|||
|
*/
|
|||
|
public final Environment getEnvironment() {
|
|||
|
|
|||
|
return env;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the class catalog.
|
|||
|
*/
|
|||
|
public final StoredClassCatalog getClassCatalog() {
|
|||
|
|
|||
|
return javaCatalog;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the part storage container.
|
|||
|
*/
|
|||
|
public final Database getPartDatabase() {
|
|||
|
|
|||
|
return partDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the supplier storage container.
|
|||
|
*/
|
|||
|
public final Database getSupplierDatabase() {
|
|||
|
|
|||
|
return supplierDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the shipment storage container.
|
|||
|
*/
|
|||
|
public final Database getShipmentDatabase() {
|
|||
|
|
|||
|
return shipmentDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the shipment-by-part index.
|
|||
|
*/
|
|||
|
public final SecondaryDatabase getShipmentByPartDatabase() {
|
|||
|
|
|||
|
return shipmentByPartDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the shipment-by-supplier index.
|
|||
|
*/
|
|||
|
public final SecondaryDatabase getShipmentBySupplierDatabase() {
|
|||
|
|
|||
|
return shipmentBySupplierDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Return the supplier-by-city index.
|
|||
|
*/
|
|||
|
public final SecondaryDatabase getSupplierByCityDatabase() {
|
|||
|
|
|||
|
return supplierByCityDb;
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Close all stores (closing a store automatically closes its indices).
|
|||
|
*/
|
|||
|
public void close()
|
|||
|
throws DatabaseException {
|
|||
|
|
|||
|
// Close secondary databases, then primary databases.
|
|||
|
supplierByCityDb.close();
|
|||
|
shipmentByPartDb.close();
|
|||
|
shipmentBySupplierDb.close();
|
|||
|
partDb.close();
|
|||
|
supplierDb.close();
|
|||
|
shipmentDb.close();
|
|||
|
// And don't forget to close the catalog and the environment.
|
|||
|
javaCatalog.close();
|
|||
|
env.close();
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* The SecondaryKeyCreator for the SupplierByCity index. This is an
|
|||
|
* extension of the abstract class TupleSerialKeyCreator, which implements
|
|||
|
* SecondaryKeyCreator for the case where the data keys are of the format
|
|||
|
* TupleFormat and the data values are of the format SerialFormat.
|
|||
|
*/
|
|||
|
private static class SupplierByCityKeyCreator
|
|||
|
extends TupleSerialKeyCreator {
|
|||
|
|
|||
|
/**
|
|||
|
* Construct the city key extractor.
|
|||
|
* @param catalog is the class catalog.
|
|||
|
* @param valueClass is the supplier value class.
|
|||
|
*/
|
|||
|
private SupplierByCityKeyCreator(ClassCatalog catalog,
|
|||
|
Class valueClass) {
|
|||
|
|
|||
|
super(catalog, valueClass);
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Extract the city key from a supplier key/value pair. The city key
|
|||
|
* is stored in the supplier value, so the supplier key is not used.
|
|||
|
*/
|
|||
|
public boolean createSecondaryKey(TupleInput primaryKeyInput,
|
|||
|
Object valueInput,
|
|||
|
TupleOutput indexKeyOutput) {
|
|||
|
|
|||
|
SupplierData supplierData = (SupplierData) valueInput;
|
|||
|
String city = supplierData.getCity();
|
|||
|
if (city != null) {
|
|||
|
indexKeyOutput.writeString(supplierData.getCity());
|
|||
|
return true;
|
|||
|
} else {
|
|||
|
return false;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* The SecondaryKeyCreator for the ShipmentByPart index. This is an
|
|||
|
* extension of the abstract class TupleSerialKeyCreator, which implements
|
|||
|
* SecondaryKeyCreator for the case where the data keys are of the format
|
|||
|
* TupleFormat and the data values are of the format SerialFormat.
|
|||
|
*/
|
|||
|
private static class ShipmentByPartKeyCreator
|
|||
|
extends TupleSerialKeyCreator {
|
|||
|
|
|||
|
/**
|
|||
|
* Construct the part key extractor.
|
|||
|
* @param catalog is the class catalog.
|
|||
|
* @param valueClass is the shipment value class.
|
|||
|
*/
|
|||
|
private ShipmentByPartKeyCreator(ClassCatalog catalog,
|
|||
|
Class valueClass) {
|
|||
|
super(catalog, valueClass);
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Extract the part key from a shipment key/value pair. The part key
|
|||
|
* is stored in the shipment key, so the shipment value is not used.
|
|||
|
*/
|
|||
|
public boolean createSecondaryKey(TupleInput primaryKeyInput,
|
|||
|
Object valueInput,
|
|||
|
TupleOutput indexKeyOutput) {
|
|||
|
|
|||
|
String partNumber = primaryKeyInput.readString();
|
|||
|
// don't bother reading the supplierNumber
|
|||
|
indexKeyOutput.writeString(partNumber);
|
|||
|
return true;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* The SecondaryKeyCreator for the ShipmentBySupplier index. This is an
|
|||
|
* extension of the abstract class TupleSerialKeyCreator, which implements
|
|||
|
* SecondaryKeyCreator for the case where the data keys are of the format
|
|||
|
* TupleFormat and the data values are of the format SerialFormat.
|
|||
|
*/
|
|||
|
private static class ShipmentBySupplierKeyCreator
|
|||
|
extends TupleSerialKeyCreator {
|
|||
|
|
|||
|
/**
|
|||
|
* Construct the supplier key extractor.
|
|||
|
* @param catalog is the class catalog.
|
|||
|
* @param valueClass is the shipment value class.
|
|||
|
*/
|
|||
|
private ShipmentBySupplierKeyCreator(ClassCatalog catalog,
|
|||
|
Class valueClass) {
|
|||
|
super(catalog, valueClass);
|
|||
|
}
|
|||
|
|
|||
|
/**
|
|||
|
* Extract the supplier key from a shipment key/value pair. The
|
|||
|
* supplier key is stored in the shipment key, so the shipment value is
|
|||
|
* not used.
|
|||
|
*/
|
|||
|
public boolean createSecondaryKey(TupleInput primaryKeyInput,
|
|||
|
Object valueInput,
|
|||
|
TupleOutput indexKeyOutput) {
|
|||
|
|
|||
|
primaryKeyInput.readString(); // skip the partNumber
|
|||
|
String supplierNumber = primaryKeyInput.readString();
|
|||
|
indexKeyOutput.writeString(supplierNumber);
|
|||
|
return true;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|