1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * The contents of this file are subject to the terms of the Liferay Enterprise
5    * Subscription License ("License"). You may not use this file except in
6    * compliance with the License. You can obtain a copy of the License by
7    * contacting Liferay, Inc. See the License for the specific language governing
8    * permissions and limitations under the License, including but not limited to
9    * distribution rights of the Software.
10   *
11   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
12   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
13   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
14   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
15   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
16   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
17   * SOFTWARE.
18   */
19  
20  package com.liferay.portal.upgrade.util;
21  
22  import com.liferay.counter.service.CounterLocalServiceUtil;
23  
24  import java.sql.Types;
25  
26  /**
27   * <a href="PKUpgradeColumnImpl.java.html"><b><i>View Source</i></b></a>
28   *
29   * @author Brian Wing Shun Chan
30   * @author Alexander Chow
31   *
32   */
33  public class PKUpgradeColumnImpl extends BaseUpgradeColumnImpl {
34  
35      public PKUpgradeColumnImpl(String name, boolean trackValues) {
36          this(name, null, trackValues);
37      }
38  
39      public PKUpgradeColumnImpl(String name, Integer oldColumnType,
40                                 boolean trackValues) {
41  
42          super(name, oldColumnType);
43  
44          _newColumnType = new Integer(Types.BIGINT);
45          _trackValues = trackValues;
46  
47          if (_trackValues) {
48              _valueMapper = ValueMapperFactory.getValueMapper();
49          }
50      }
51  
52      public Integer getNewColumnType(Integer defaultType) {
53          return _newColumnType;
54      }
55  
56      public Object getNewValue(Object oldValue) throws Exception {
57          Long newValue = new Long(CounterLocalServiceUtil.increment());
58  
59          if (_trackValues) {
60              _valueMapper.mapValue(oldValue, newValue);
61          }
62  
63          return newValue;
64      }
65  
66      public boolean isTrackValues() {
67          return _trackValues;
68      }
69  
70      public ValueMapper getValueMapper() {
71          return _valueMapper;
72      }
73  
74      private Integer _newColumnType;
75      private boolean _trackValues;
76      private ValueMapper _valueMapper;
77  
78  }