SitesController.java 3.42 KB
package com.cnlive.shenhe.controller;

import com.alibaba.fastjson.JSONObject;
import com.cnlive.shenhe.bean.ErrorEnum;
import com.cnlive.shenhe.bean.ResponseBean;
import com.cnlive.shenhe.bean.SessionUser;
import com.cnlive.shenhe.entity.ShySites;
import com.cnlive.shenhe.serviceImpl.SitesServiceImpl;
import com.cnlive.shenhe.utils.CommonConst;
import com.cnlive.shenhe.utils.CommonUtils;
import com.github.pagehelper.PageInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpSession;
import java.util.List;

/**
 * @Auther: 周伟鹏
 * @Date: 2018/11/30 17:28
 * @Description:
 */
@Controller
@RequestMapping("/sites")
public class SitesController {

    private static Logger logger = LoggerFactory.getLogger(SitesController.class);

    @Autowired
    SitesServiceImpl sitesService;

    /**
     * 站点白名单分页列表
     *
     * @param page
     * @param pageSize
     * @param orderByClause
     * @param keyword
     * @return
     */
    @GetMapping(value = "/page")
    public String page(Model model, Integer page, Integer pageSize, String orderByClause, String keyword, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        Integer spid = sessionUser.getSpid();
        if (spid != 0){
            //return "page/error.html";
        }
        if (CommonUtils.isNull(page)) page = 1;
        if (CommonUtils.isNull(pageSize)) pageSize = 10;
        PageInfo<ShySites> sitePage = sitesService.getPage(page, pageSize, orderByClause, keyword, null);
        List<ShySites> list = sitePage.getList();
        JSONObject business = CommonConst.BUSINESS;
        if(!list.isEmpty()){
            for (ShySites site:list) {
                String write_business_name = site.getWrite_business();
                if(CommonUtils.isNotEmpty(write_business_name)){
                    for (String bkey:business.keySet()) {
                        write_business_name = write_business_name.replace(bkey, business.getString(bkey));
                    }
                }else{
                    write_business_name = "无";
                }
                site.setWrite_business_name(write_business_name);
            }
        }
        model.addAttribute("sitePage", sitePage);
        model.addAttribute("business", business);
        return "page/site.html";
    }


    @PostMapping(value = "/write")
    @ResponseBody
    public ResponseBean write(Integer siteId, String[] write_business, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        Integer spid = sessionUser.getSpid();
        if (spid != 0){
            //return "page/error.html";
        }
        ShySites site = sitesService.get(siteId);
        String write = "";
        if(CommonUtils.isNotNull(write_business)&&write_business.length>0){
            for (String w:write_business) {
                write = write+","+w;
            }
            write = write.substring(1);
        }
        site.setWrite_business(write);
        boolean success = sitesService.update(site);
        if(success){
            return new ResponseBean();
        }else{
            return new ResponseBean(ErrorEnum.ERROR);
        }
    }

}