1
19
20 package com.liferay.portal.upgrade.util;
21
22 import com.liferay.portal.upgrade.StagnantRowException;
23
24 import java.util.Iterator;
25 import java.util.LinkedHashMap;
26 import java.util.LinkedHashSet;
27 import java.util.Map;
28 import java.util.Set;
29
30
37 public class MemoryValueMapper implements ValueMapper {
38
39 public MemoryValueMapper() {
40 this(new LinkedHashSet<Object>());
41 }
42
43 public MemoryValueMapper(Set<Object> exceptions) {
44 _map = new LinkedHashMap<Object, Object>();
45 _exceptions = exceptions;
46 }
47
48 public Object getNewValue(Object oldValue) throws Exception {
49 Object value = _map.get(oldValue);
50
51 if (value == null) {
52 if (_exceptions.contains(oldValue)) {
53 value = oldValue;
54 }
55 else {
56 throw new StagnantRowException(String.valueOf(oldValue));
57 }
58 }
59
60 return value;
61 }
62
63 public void mapValue(Object oldValue, Object newValue) throws Exception {
64 _map.put(oldValue, newValue);
65 }
66
67 public void appendException(Object exception) {
68 _exceptions.add(exception);
69 }
70
71 public Iterator<Object> iterator() throws Exception {
72 return _map.keySet().iterator();
73 }
74
75 public int size() throws Exception {
76 return _map.size();
77 }
78
79 public Map<Object, Object> getMap() {
80 return _map;
81 }
82
83 private Map<Object, Object> _map;
84 private Set<Object> _exceptions;
85
86 }