1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * Permission is hereby granted, free of charge, to any person obtaining a copy
5    * of this software and associated documentation files (the "Software"), to deal
6    * in the Software without restriction, including without limitation the rights
7    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8    * copies of the Software, and to permit persons to whom the Software is
9    * furnished to do so, subject to the following conditions:
10   *
11   * The above copyright notice and this permission notice shall be included in
12   * all copies or substantial portions of the Software.
13   *
14   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20   * SOFTWARE.
21   */
22  
23  package com.liferay.portal.upgrade.util;
24  
25  import com.liferay.counter.service.CounterLocalServiceUtil;
26  import com.liferay.portal.kernel.upgrade.util.BaseUpgradeColumnImpl;
27  import com.liferay.portal.kernel.upgrade.util.ValueMapper;
28  import com.liferay.portal.kernel.upgrade.util.ValueMapperFactoryUtil;
29  
30  import java.sql.Types;
31  
32  /**
33   * <a href="PKUpgradeColumnImpl.java.html"><b><i>View Source</i></b></a>
34   *
35   * @author Brian Wing Shun Chan
36   * @author Alexander Chow
37   */
38  public class PKUpgradeColumnImpl extends BaseUpgradeColumnImpl {
39  
40      public PKUpgradeColumnImpl(String name, boolean trackValues) {
41          this(name, null, trackValues);
42      }
43  
44      public PKUpgradeColumnImpl(
45          String name, Integer oldColumnType, boolean trackValues) {
46  
47          super(name, oldColumnType);
48  
49          _newColumnType = new Integer(Types.BIGINT);
50          _trackValues = trackValues;
51  
52          if (_trackValues) {
53              _valueMapper = ValueMapperFactoryUtil.getValueMapper();
54          }
55      }
56  
57      public Integer getNewColumnType(Integer defaultType) {
58          return _newColumnType;
59      }
60  
61      public Object getNewValue(Object oldValue) throws Exception {
62          Long newValue = new Long(CounterLocalServiceUtil.increment());
63  
64          if (_trackValues) {
65              _valueMapper.mapValue(oldValue, newValue);
66          }
67  
68          return newValue;
69      }
70  
71      public boolean isTrackValues() {
72          return _trackValues;
73      }
74  
75      public ValueMapper getValueMapper() {
76          return _valueMapper;
77      }
78  
79      private Integer _newColumnType;
80      private boolean _trackValues;
81      private ValueMapper _valueMapper;
82  
83  }