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

import com.alibaba.fastjson.JSONObject;
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.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;

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

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

    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());
        if(CommonUtils.isNull(sessionUser)){
        	return "redirect:/users/login";
        }
        Integer spid = sessionUser.getSpid();
        if (!sessionUser.isMaster() && spid != 0) {
        	setmodel_site(model);
    		return "error.html";
        }
        if (spid == 0) spid = null;
        if (CommonUtils.isNull(page)) page = 1;
        if (CommonUtils.isNull(pageSize)) pageSize = 10;
        PageInfo<ShySites> sitePage = sitesService.getPage(page, pageSize, orderByClause, keyword, spid);
        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("pageSize", pageSize);
        model.addAttribute("business", business);
        return "page/site.html";
    }


    @PostMapping(value = "/write")
    public String write(Model model,Integer siteId, String[] write_business, HttpSession session, String redirect_url) {
        ShySites site = sitesService.get(siteId);
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        if(CommonUtils.isNull(sessionUser)){
        	return "redirect:/users/login";
        }
        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 "redirect:" + redirect_url;
        } else {
        	setmodel_result(model);
    		return "error.html";
        }
    }

}