17.发布动态

发布动态

一、MovementsController

package com.tanhua.server.controller;
 
import com.tanhua.model.mongo.Movement;
import com.tanhua.server.service.MovementsService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
 
import java.io.IOException;
 
/**
 * 圈子
 */
@RestController
@RequestMapping("/movements")
public class MovementsController {
 
 
 
    @Autowired
    private MovementsService movementsService;
 
 
    /**
     * 发布动态:
     * 请求路径：/movements
     * 请求方式：post
     * 请求参数：动态图片imageContent，还有Movenment
     * 响应数据：null
     */
 
    @PostMapping
    public ResponseEntity publishMoments(Movement movement, MultipartFile[] imageContent) throws IOException {
 
        //1.接收参数，调用api完成动态发布
        movementsService.publishMoments(movement,imageContent);
        //2.响应数据
 
        return ResponseEntity.ok(null);
    }
 
 
 
}
 

二、MovementsService

package com.tanhua.server.service;
 
import com.tanhua.autoconfig.template.OssTemplate;
import com.tanhua.dubbo.api.MovementsApi;
import com.tanhua.model.mongo.Movement;
import com.tanhua.model.vo.ErrorResult;
import com.tanhua.server.exception.BusinessException;
import com.tanhua.server.interceptor.ThreadLocalUntils;
import org.apache.dubbo.config.annotation.DubboReference;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;
 
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
 
@Service
public class MovementsService {
 
 
    @DubboReference
    private MovementsApi movementsApi;
 
    @Autowired
    private OssTemplate ossTemplate;
 
    //发布动态
    public void publishMoments(Movement movement, MultipartFile[] imageContent) throws IOException {
 
        //1.判断用户发布的动态的文字内容是否为空,为空则抛出异常
        if(StringUtils.isEmpty(movement.getTextContent())){
            throw new BusinessException(ErrorResult.contentError());
        }
 
        //2.获取当前登录的用户id
        Long userId = ThreadLocalUntils.getUserId();
 
        //3.遍历发布的图片数组，上传到阿里云oss，得到图片访问路径
 
        List<String> urls = new ArrayList<>();
        for (MultipartFile multipartFile : imageContent) {
            String imageUrl = ossTemplate.upload(multipartFile.getOriginalFilename(), multipartFile.getInputStream());
            urls.add(imageUrl);
        }
 
        //4.把用户id和图片的访问路径集合封装进动态详情对象
        movement.setUserId(userId);
        movement.setMedias(urls);
 
        //5.调用api完成发布动态
        movementsApi.publishMoments(movement);
 
    }
}
 

三、MovementsApiImpl

package com.tanhua.dubbo.api;
 
import com.tanhua.dubbo.utils.IdWorker;
import com.tanhua.model.mongo.Friend;
import com.tanhua.model.mongo.Movement;
import com.tanhua.model.mongo.MovementTimeLine;
import org.apache.dubbo.config.annotation.DubboService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
 
import java.util.List;
 
@DubboService
public class MovementsApiImpl implements MovementsApi {
 
    @Autowired
    private MongoTemplate mongoTemplate;
 
 
    @Autowired
    private IdWorker idWorker;
    /**
     *发布动态
     */
    public void publishMoments(Movement movement) {
 
        //1.保存用户发布的动态到动态详情表,保存之间先封装齐movement数据
        // 先设置pid，再设置发布时间
        movement.setPid(idWorker.getNextId("movement"));
        movement.setCreated(System.currentTimeMillis());
        mongoTemplate.save(movement);
 
        //2.根据当前登录用户的id查询好友
        Criteria criteria = Criteria.where("userId").is(movement.getUserId());
        Query query = Query.query(criteria);
        List<Friend> friends = mongoTemplate.find(query, Friend.class);
 
        //3.遍历好友集合,为每个好友设置时间线
        for (Friend friend : friends) {
            MovementTimeLine mtl = new MovementTimeLine();
            mtl.setMovementId(movement.getId());
            mtl.setUserId(friend.getUserId());
            mtl.setFriendId(friend.getFriendId());
            mtl.setCreated(System.currentTimeMillis());
        }
 
 
    }
}
 

四、代码优化

1.启动类上添加@@EnableAsync注解

2.新建一个类，异步处理工具类,SaveTimeLIne；

把它注入为spring管理的bean,在方法上使用@Async注解，把查询好友跟添加好友时间线的代码抽取出来

package com.tanhua.dubbo.utils;
 
import com.tanhua.model.mongo.Friend;
import com.tanhua.model.mongo.MovementTimeLine;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
 
import java.util.List;
 
@Component
public class SaveTimeLIne {
 
    @Autowired
    private MongoTemplate mongoTemplate;
 
 
    @Async
    public void saveTimeLine(Long userId, ObjectId movementId){
 
 
        //2.根据当前登录用户的id查询好友
        Criteria criteria = Criteria.where("userId").is(userId);
        Query query = Query.query(criteria);
        List<Friend> friends = mongoTemplate.find(query, Friend.class);
 
        //3.遍历好友集合,为每个好友设置时间线
        for (Friend friend : friends) {
            MovementTimeLine mtl = new MovementTimeLine();
            mtl.setMovementId(movementId);
            mtl.setUserId(friend.getUserId());
            mtl.setFriendId(friend.getFriendId());
            mtl.setCreated(System.currentTimeMillis());
        }
 
    }
}
 

三、api实现类上直接调用方法

package com.tanhua.dubbo.api;
 
import com.tanhua.dubbo.utils.IdWorker;
import com.tanhua.dubbo.utils.SaveTimeLIne;
import com.tanhua.model.mongo.Movement;
import org.apache.dubbo.config.annotation.DubboService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
 
@DubboService
public class MovementsApiImpl implements MovementsApi {
 
    @Autowired
    private MongoTemplate mongoTemplate;
 
 
    @Autowired
    private IdWorker idWorker;
 
 
 
    @Autowired
    private SaveTimeLIne saveTimeLIne;
 
 
    /**
     *发布动态
     */
    public void publishMoments(Movement movement) {
 
        //1.保存用户发布的动态到动态详情表,保存之间先封装齐movement数据
        // 先设置pid，再设置发布时间
        movement.setPid(idWorker.getNextId("movement"));
        movement.setCreated(System.currentTimeMillis());
        mongoTemplate.save(movement);
 
        saveTimeLIne.saveTimeLine(movement.getUserId(), movement.getId());
 
    }
}