1
22
23 package com.liferay.portal.upgrade.v5_1_0.util;
24
25 import com.liferay.portal.kernel.util.Validator;
26 import com.liferay.portal.upgrade.util.BaseUpgradeColumnImpl;
27 import com.liferay.portal.upgrade.util.UpgradeColumn;
28 import com.liferay.portlet.blogs.service.BlogsEntryLocalServiceUtil;
29
30 import java.util.HashSet;
31 import java.util.Set;
32
33
40 public class BlogsEntryUrlTitleUpgradeColumnImpl extends BaseUpgradeColumnImpl {
41
42 public BlogsEntryUrlTitleUpgradeColumnImpl(
43 UpgradeColumn entryIdColumn, UpgradeColumn titleColumn) {
44
45 super("urlTitle");
46
47 _entryIdColumn = entryIdColumn;
48 _titleColumn = titleColumn;
49 _urlTitles = new HashSet<String>();
50 }
51
52 public Object getNewValue(Object oldValue) throws Exception {
53 String oldUrlTitle = (String)oldValue;
54
55 String newUrlTitle = oldUrlTitle;
56
57 if (Validator.isNull(oldUrlTitle)) {
58 long entryId = ((Long)_entryIdColumn.getOldValue()).longValue();
59
60 String title = (String)_titleColumn.getOldValue();
61
62 newUrlTitle = getUrlTitle(entryId, title);
63
64 _urlTitles.add(newUrlTitle);
65 }
66
67 return newUrlTitle;
68 }
69
70 protected String getUrlTitle(long entryId, String title) {
71 String urlTitle = BlogsEntryLocalServiceUtil.getUrlTitle(
72 entryId, title);
73
74 String newUrlTitle = new String(urlTitle);
75
76 for (int i = 1;; i++) {
77 if (!_urlTitles.contains(newUrlTitle)) {
78 break;
79 }
80
81 newUrlTitle = urlTitle + "_" + i;
82 }
83
84 return newUrlTitle;
85 }
86
87 private UpgradeColumn _entryIdColumn;
88 private UpgradeColumn _titleColumn;
89 private Set<String> _urlTitles;
90
91 }