| 1 |
|
package org.kuali.student.common.util.spring; |
| 2 |
|
|
| 3 |
|
import java.lang.reflect.Field; |
| 4 |
|
import java.util.Map; |
| 5 |
|
|
| 6 |
|
import javax.persistence.spi.PersistenceUnitTransactionType; |
| 7 |
|
|
| 8 |
|
import org.springframework.orm.jpa.persistenceunit.DefaultPersistenceUnitManager; |
| 9 |
|
import org.springframework.orm.jpa.persistenceunit.MutablePersistenceUnitInfo; |
| 10 |
|
|
|
|
|
| 0% |
Uncovered Elements: 19 (19) |
Complexity: 5 |
Complexity Density: 0.36 |
|
| 11 |
|
public class MergingPersistenceUnitManager extends |
| 12 |
|
DefaultPersistenceUnitManager { |
| 13 |
|
|
| 14 |
|
private String mergedPersistenceUnitName; |
| 15 |
|
private PersistenceUnitTransactionType transactionTypeOverride = PersistenceUnitTransactionType.RESOURCE_LOCAL; |
| 16 |
|
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
| 17 |
0
|
public void setMergedPersistenceUnitName(String mergedPersistenceUnitName) {... |
| 18 |
0
|
this.mergedPersistenceUnitName = mergedPersistenceUnitName; |
| 19 |
|
} |
| 20 |
|
|
|
|
|
| 0% |
Uncovered Elements: 14 (14) |
Complexity: 3 |
Complexity Density: 0.25 |
|
| 21 |
0
|
@Override... |
| 22 |
|
protected void postProcessPersistenceUnitInfo(MutablePersistenceUnitInfo pui) { |
| 23 |
0
|
try { |
| 24 |
0
|
Field persistenceUnitInfosField = DefaultPersistenceUnitManager.class.getDeclaredField("persistenceUnitInfos"); |
| 25 |
0
|
persistenceUnitInfosField.setAccessible(true); |
| 26 |
0
|
Map<String, MutablePersistenceUnitInfo> persistenceUnitInfos = (Map<String, MutablePersistenceUnitInfo>) persistenceUnitInfosField.get(this); |
| 27 |
|
|
| 28 |
0
|
if(persistenceUnitInfos.isEmpty()){ |
| 29 |
0
|
pui.setPersistenceUnitName(mergedPersistenceUnitName); |
| 30 |
0
|
pui.setTransactionType(transactionTypeOverride); |
| 31 |
|
}else{ |
| 32 |
0
|
MutablePersistenceUnitInfo currentlyMergedPui = persistenceUnitInfos.get(mergedPersistenceUnitName); |
| 33 |
0
|
for(String className:pui.getManagedClassNames()){ |
| 34 |
0
|
currentlyMergedPui.getManagedClassNames().add(className); |
| 35 |
|
} |
| 36 |
|
} |
| 37 |
|
|
| 38 |
|
} catch (Exception e) { |
| 39 |
0
|
throw new RuntimeException("Error while merging persistence units.",e); |
| 40 |
|
} |
| 41 |
|
|
| 42 |
0
|
super.postProcessPersistenceUnitInfo(pui); |
| 43 |
|
} |
| 44 |
|
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
| 45 |
0
|
public void setTransactionTypeOverride(PersistenceUnitTransactionType transactionTypeOverride) {... |
| 46 |
0
|
this.transactionTypeOverride = transactionTypeOverride; |
| 47 |
|
} |
| 48 |
|
|
| 49 |
|
|
| 50 |
|
} |