/**
* 按下面方式存数据 map: appidA : userIds(TreeSet) appidB : userIds(TreeSet)
*
* @param map
* @param appUser
*/
private void put(TreeMap<Integer, TreeSet<String>> map, AppUser appUser) {
Integer appId = appUser.getApp().getId();
String userId = appUser.getUserId();
TreeSet<String> set = map.get(appId);
if (set == null) {
set = new TreeSet<String>();
set.add(userId);
map.put(appId, set);
} else {
set.add(userId);
}
}
@Override
public void saveFriend(ArrayList<AppUser> appUserList) throws InnerException {
try {
// 循环取appUser
for (AppUser appUser : appUserList) {
App app = appDao.findById(appUser.getApp().getId());
// 跳过app不存在的appUser
if (app == null) continue;
Set<AppUser> friends = appUser.getFriends();
Iterator<AppUser> iterator = friends.iterator();
List<AppUser> friendsPersisted = new ArrayList<AppUser>();
// 循环取朋友名
while (iterator.hasNext()) {
AppUser friend = iterator.next();
// 查找朋友是否已存在数据库
List<AppUser> list = appUserDao.findByUserIdAndApp(friend.getUserId(), app);
// 保存不存在的朋友，若存在，删除列表中的脱管对象，插入持久态对象
if (list.size() == 0) {
friend.setCreateTime(System.currentTimeMillis());
appUserDao.save(friend);
} else {
friendsPersisted.add(list.get(0));
iterator.remove();
}
}
friends.addAll(friendsPersisted);
// 如果appUser不存在，保存；若存在，更新朋友列表
List<AppUser> list = appUserDao.findByUserIdAndApp(appUser.getUserId(), app);
if (list.size() == 0) {
appUser.setCreateTime(System.currentTimeMillis());
appUserDao.save(appUser);
} else {
AppUser appUserPersisted = list.get(0);
appUserPersisted.setFriends(appUser.getFriends());
appUserDao.update(appUserPersisted);
}
}
} catch (RuntimeException re) {
log.error("error saving friend: " + re.toString());
throw new InnerException();
}
}
@Override
public void saveGroup(HashSet<AppUser> appUserSet) throws InnerException {
Set<AppUser> contain = new HashSet<AppUser>();
TreeMap<Integer, TreeSet<String>> map = new TreeMap<Integer, TreeSet<String>>();
try {
// 循环取appUser，插入不存在的appUser，然后将组插入TreeMap（为了排序），
// TreeMap按key(appid)排序，value(appUserIds:可能一个应用同时等两个号)按自然排序
for (AppUser appUser : appUserSet) {
App app = appDao.findById(appUser.getApp().getId());
if (app != null) {
List<AppUser> list = appUserDao.findByUserIdAndApp(appUser.getUserId(), app);
if (list.size() == 0) {
appUser.setCreateTime(System.currentTimeMillis());
appUserDao.save(appUser);
put(map, appUser);
contain.add(appUser);
} else {
put(map, appUser);
contain.add(list.get(0));
}
}
}
// 如果这个组大于1个用户，通过containString查看之前有无相同的组，没有就存入
if (contain.size() > 1) {
String containString = map.toString();
List<AppUserGroup> list = appUserGroupDao.findByProperty("containString", containString);
if (list.size() == 0) {
AppUserGroup appUserGroup = new AppUserGroup();
appUserGroup.setContainString(containString);
appUserGroup.setCreateTime(System.currentTimeMillis());
appUserGroup.setContains(contain);
appUserGroupDao.save(appUserGroup);
}
}
} catch (RuntimeException re) {
log.error("error saving group: " + re.toString());
throw new InnerException();
}
}