Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implements a pluggable DML interface #19

Merged
merged 1 commit into from
Aug 27, 2014
Merged

Implements a pluggable DML interface #19

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 33 additions & 4 deletions fflib/src/classes/fflib_SObjectUnitOfWork.cls
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,8 @@ public virtual class fflib_SObjectUnitOfWork

private SendEmailWork m_emailWork = new SendEmailWork();

private IDML m_dml;

/**
* Interface describes work to be performed during the commitWork method
**/
Expand All @@ -75,12 +77,37 @@ public virtual class fflib_SObjectUnitOfWork
void doWork();
}

public interface IDML
{
void dmlInsert(List<SObject> objList);
void dmlUpdate(List<SObject> objList);
void dmlDelete(List<SObject> objList);
}

public class SimpleDML implements IDML
{
public void dmlInsert(List<SObject> objList){
insert objList;
}
public void dmlUpdate(List<SObject> objList){
update objList;
}
public void dmlDelete(List<SObject> objList){
delete objList;
}
}
/**
* Constructs a new UnitOfWork to support work against the given object list
*
* @param sObjectList A list of objects given in dependency order (least dependent first)
*/
public fflib_SObjectUnitOfWork(List<Schema.SObjectType> sObjectTypes)
{
this(sObjectTypes,new SimpleDML());
}


public fflib_SObjectUnitOfWork(List<Schema.SObjectType> sObjectTypes, IDML dml)
{
m_sObjectTypes = sObjectTypes.clone();

Expand All @@ -93,6 +120,8 @@ public virtual class fflib_SObjectUnitOfWork
}

m_workList.add(m_emailWork);

m_dml = dml;
}

/**
Expand Down Expand Up @@ -171,7 +200,7 @@ public virtual class fflib_SObjectUnitOfWork
throw new UnitOfWorkException(String.format('SObject type {0} is not supported by this unit of work', new String[] { sObjectType }));
m_dirtyListByType.get(sObjectType).add(record);
}

/**
* Register an existing record to be deleted during the commitWork method
*
Expand Down Expand Up @@ -200,15 +229,15 @@ public virtual class fflib_SObjectUnitOfWork
for(Schema.SObjectType sObjectType : m_sObjectTypes)
{
m_relationships.get(sObjectType.getDescribe().getName()).resolve();
insert m_newListByType.get(sObjectType.getDescribe().getName());
m_dml.dmlInsert(m_newListByType.get(sObjectType.getDescribe().getName()));
}
// Update by type
for(Schema.SObjectType sObjectType : m_sObjectTypes)
update m_dirtyListByType.get(sObjectType.getDescribe().getName());
m_dml.dmlUpdate(m_dirtyListByType.get(sObjectType.getDescribe().getName()));
// Delete by type (in reverse dependency order)
Integer objectIdx = m_sObjectTypes.size() - 1;
while(objectIdx>=0)
delete m_deletedListByType.get(m_sObjectTypes[objectIdx--].getDescribe().getName());
m_dml.dmlDelete(m_deletedListByType.get(m_sObjectTypes[objectIdx--].getDescribe().getName()));
// Generic work
for(IDoWork work : m_workList)
work.doWork();
Expand Down