admin
2022-04-16 04f09e52ffd4681bdfd85e51acd3da0d1280c3d3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package com.hxh.spring.test;
 
import com.yeshi.buwan.dao.user.LoginUserDao;
import com.yeshi.buwan.dao.user.LoginUserExtraDao;
import com.yeshi.buwan.domain.user.LoginUser;
import com.yeshi.buwan.domain.user.LoginUserExtra;
import com.yeshi.buwan.util.StringUtil;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;
 
import javax.annotation.Resource;
import java.util.List;
 
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath:spring.xml"})
@WebAppConfiguration
public class UserTest {
 
    @Resource
    private LoginUserExtraDao loginUserExtraDao;
 
    @Resource
    private LoginUserDao loginUserDao;
 
    @Test
    public void test() {
        Query query = new Query();
        List<LoginUserExtra> extraList = loginUserExtraDao.findList(query);
        for (LoginUserExtra extra : extraList) {
            LoginUser user = loginUserDao.find(LoginUser.class, extra.getId());
            if (user != null && StringUtil.isNullOrEmpty(user.getPortrait())) {
                if (!StringUtil.isNullOrEmpty(extra.getWxPortrait())) {
                    user.setPortrait(extra.getWxPortrait());
                    loginUserDao.update(user);
                } else if (!StringUtil.isNullOrEmpty(extra.getQqPortrait())) {
                    user.setPortrait(extra.getQqPortrait());
                    loginUserDao.update(user);
                }
            }
        }
    }
 
 
}