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