org.cristalise.storage.jooqdb.lookup.JooqItemHandler Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cristalise-jooqdb Show documentation
Show all versions of cristalise-jooqdb Show documentation
CRISTAL-iSE JOOQ Storage Module
/**
* This file is part of the CRISTAL-iSE jOOQ Cluster Storage Module.
* Copyright (c) 2001-2017 The CRISTAL Consortium. All rights reserved.
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation; either version 3 of the License, or (at
* your option) any later version.
*
* This library is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; with out even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
*
* http://www.fsf.org/licensing/licenses/lgpl.html
*/
package org.cristalise.storage.jooqdb.lookup;
import static org.jooq.impl.DSL.constraint;
import static org.jooq.impl.DSL.field;
import static org.jooq.impl.DSL.name;
import static org.jooq.impl.DSL.select;
import static org.jooq.impl.DSL.table;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import org.cristalise.kernel.common.PersistencyException;
import org.cristalise.kernel.lookup.AgentPath;
import org.cristalise.kernel.lookup.ItemPath;
import org.cristalise.kernel.lookup.Path;
import org.cristalise.kernel.process.Gateway;
import org.cristalise.kernel.property.BuiltInItemProperties;
import org.cristalise.kernel.property.Property;
import org.cristalise.storage.jooqdb.JooqHandler;
import org.cristalise.storage.jooqdb.clusterStore.JooqItemPropertyHandler;
import org.jooq.DSLContext;
import org.jooq.Field;
import org.jooq.InsertSetMoreStep;
import org.jooq.Record;
import org.jooq.Result;
import org.jooq.Table;
import org.jooq.UpdateSetMoreStep;
import org.jooq.impl.SQLDataType;
/**
* Table to store data of ItemPath and AgentPath
*/
public class JooqItemHandler {
static final Table ITEM_TABLE = table(name("ITEM"));
static final Field UUID = field(name("UUID"), UUID.class);
static final Field IOR = field(name("IOR"), String.class);
static final Field IS_AGENT = field(name("IS_AGENT"), Boolean.class);
static final Field IS_PASSWORD_TEMPORARY = field(name("IS_PASSWORD_TEMPORARY"), Boolean.class);
static final Field PASSWORD = field(name("PASSWORD"), String.class);
public void createTables(DSLContext context) throws PersistencyException {
context.createTableIfNotExists(ITEM_TABLE)
.column(UUID, JooqHandler.UUID_TYPE .nullable(false))
.column(IOR, JooqHandler.IOR_TYPE .nullable(true))
.column(IS_AGENT, SQLDataType.BOOLEAN .nullable(false))
.column(IS_PASSWORD_TEMPORARY, SQLDataType.BOOLEAN .nullable(true))
.column(PASSWORD, JooqHandler.PASSWORD_TYPE.nullable(true))
.constraints(
constraint("PK_"+ITEM_TABLE).primaryKey(UUID))
.execute();
}
public void dropTables(DSLContext context) throws PersistencyException {
context.dropTableIfExists(ITEM_TABLE).execute();
}
public int updatePassword(DSLContext context, AgentPath agent, String password, boolean temporary) throws PersistencyException {
UpdateSetMoreStep update = context
.update(ITEM_TABLE)
.set(PASSWORD, password);
if (Gateway.getProperties().getBoolean("JOOQ.TemporaryPwdFieldImplemented", true)) update.set(IS_PASSWORD_TEMPORARY, temporary);
return update.where(UUID.equal(agent.getUUID())).execute();
}
public int updateIOR(DSLContext context, ItemPath item, String ior) throws PersistencyException {
return context
.update(ITEM_TABLE)
.set(IOR, ior)
.where(UUID.equal(item.getUUID()))
.execute();
}
public int update(DSLContext context, ItemPath path) throws PersistencyException {
boolean isAgent = path instanceof AgentPath;
return context
.update(ITEM_TABLE)
.set(IOR, path.getIORString())
.set(IS_AGENT, isAgent)
.where(UUID.equal(path.getUUID()))
.execute();
}
public int delete(DSLContext context, UUID uuid) throws PersistencyException {
return context
.delete(ITEM_TABLE)
.where(UUID.equal(uuid))
.execute();
}
public int insert(DSLContext context, AgentPath agentPath, JooqItemPropertyHandler properties) throws PersistencyException {
InsertSetMoreStep insert = context.insertInto(ITEM_TABLE)
.set(UUID, agentPath.getUUID())
.set(IOR, agentPath.getIORString())
.set(IS_AGENT, true);
if (Gateway.getProperties().getBoolean("JOOQ.TemporaryPwdFieldImplemented", true))
insert.set(IS_PASSWORD_TEMPORARY, agentPath.isPasswordTemporary());
int rows = insert.execute();
if (rows != 1) throw new PersistencyException("Insert into ITEM table rows:"+rows);
Property name = new Property(BuiltInItemProperties.NAME, agentPath.getAgentName(), true);
rows = properties.insert(context, agentPath.getUUID(), name);
if (rows != 1) throw new PersistencyException("Insert into ITEM_PROPERTY table rows:"+rows);
return 1;
}
public int insert(DSLContext context, ItemPath path) throws PersistencyException {
return context
.insertInto(ITEM_TABLE)
.set(UUID, path.getUUID())
.set(IOR, path.getIORString())
.set(IS_AGENT, false)
.execute();
}
public boolean exists(DSLContext context, UUID uuid) throws PersistencyException {
return context.fetchExists( select().from(ITEM_TABLE).where(UUID.equal(uuid)) );
}
public ItemPath fetch(DSLContext context, UUID uuid, JooqItemPropertyHandler properties) throws PersistencyException {
Record result = context
.select().from(ITEM_TABLE)
.where(UUID.equal(uuid))
.fetchOne();
return getItemPath(context, properties, result);
}
public String fetchPassword(DSLContext context, UUID uuid) throws PersistencyException {
Record result = context
.select(PASSWORD).from(ITEM_TABLE)
.where(UUID.equal(uuid))
.fetchOne();
return result.get(PASSWORD);
}
public static ItemPath getItemPath(DSLContext context, JooqItemPropertyHandler properties, Record record) throws PersistencyException {
if(record != null) {
UUID uuid;
//Reading UUID is done this way because of a bug in jooq supporting MySQL: check issue #23
if (record.get(UUID.getName()) instanceof String) uuid = java.util.UUID.fromString(record.get(UUID.getName(), String.class));
else uuid = record.get(UUID);
//Reading IS_AGENT boolean is done this way because of a bug in jooq supporting MySQL: check issue #23
boolean isAgent = record.get(IS_AGENT.getName(), Boolean.class);
String ior = record.get(IOR);
Boolean isTempPwd = false;
try {
isTempPwd = record.get(IS_PASSWORD_TEMPORARY.getName(), Boolean.class);
}
catch (IllegalArgumentException e) {}
String nameProp = BuiltInItemProperties.NAME.toString();
if(isAgent) {
String name;
if (record.field(nameProp) != null) {
name = record.get(nameProp, String.class);
} else {
Property nameProperty = (Property) properties.fetch(context, uuid, nameProp);
if (nameProperty == null) {
return null;
}
name = nameProperty.getValue();
}
if (name == null) {
return null;
}
return new AgentPath(uuid, ior, name, isTempPwd != null ? isTempPwd : false);
}
else
return new ItemPath(uuid, ior);
}
return null;
}
public List fetchAll(DSLContext context, List uuids, JooqItemPropertyHandler properties)
throws PersistencyException
{
Result result = context
.select().from(ITEM_TABLE)
.where(UUID.in(uuids))
.fetch();
List foundPathes = new ArrayList<>();
for (Record record : result) foundPathes.add(getItemPath(context, properties, record));
return foundPathes;
}
}