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