Message.php 7.1 KB
<?php

namespace app\admin\controller;

use app\common\controller\Backend;
use app\admin\model\User;
use think\Db;

/**
 * 系统消息管理
 *
 * @icon fa fa-circle-o
 */
class Message extends Backend
{
    
    /**
     * Message模型对象
     * @var \app\admin\model\Message
     */
    protected $model = null;

    public function _initialize()
    {
        parent::_initialize();
        $this->request->filter(['strip_tags']);
        $this->model = new \app\admin\model\Message;

    }
    
    /**
     * 默认生成的控制器所继承的父类中有index/add/edit/del/multi五个基础方法、destroy/restore/recyclebin三个回收站方法
     * 因此在当前控制器中可不用编写增删改查的代码,除非需要自己控制这部分逻辑
     * 需要将application/admin/library/traits/Backend.php中对应的方法复制到当前控制器,然后进行修改
     */


    /**
     * 添加
     */
    public function add()
    {
        if ($this->request->isPost()) {
            $params = $this->request->post("row/a");
            $user_ids = $this->request->post("user_ids");
            if ($params) {
                if ($this->dataLimit && $this->dataLimitFieldAutoFill) {
                    $params[$this->dataLimitField] = $this->auth->id;
                }
                try {
                    //是否采用模型验证
                    if ($this->modelValidate) {
                        $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                        $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.add' : $name) : $this->modelValidate;
                        $this->model->validate($validate);
                    }
                    $result = $this->model->allowField(true)->save($params);
                    //系统推送消息
                    $id = $this->model->id;
                    $userModel = new User();
                    if(!empty($user_ids)){
                        $users = explode(',',$user_ids);
                    }else{
                        $users = $userModel->where(['status'=>'normal'])->column('id');
                    }
                    $logList = [];
                    foreach ($users as $value){
                        $log['message_id'] = $id;
                        $log['createtime'] = time();
                        $log['user_id'] = $value;
                        array_push($logList,$log);
                    }
                    $row = false;
                    if(!empty($logList)){
                        $row = Db::name('message_log')->insertAll($logList);
                    }

                    if ($result !== false && $row) {
                        $this->success();
                    } else {
                        $this->error($this->model->getError());
                    }
                } catch (\think\exception\PDOException $e) {
                    $this->error($e->getMessage());
                } catch (\think\Exception $e) {
                    $this->error($e->getMessage());
                }
            }
            $this->error(__('Parameter %s can not be empty', ''));
        }
        return $this->view->fetch();
    }

    /**
     * 编辑
     */
    public function edit($ids = NULL)
    {
        $row = $this->model->get($ids);
        $userIds=Db::name('message_log')->where(['message_id'=>$ids])->column('user_id');
        if (!$row)
            $this->error(__('No Results were found'));
        $adminIds = $this->getDataLimitAdminIds();
        if (is_array($adminIds)) {
            if (!in_array($row[$this->dataLimitField], $adminIds)) {
                $this->error(__('You have no permission'));
            }
        }
        if ($this->request->isPost()) {
            $params = $this->request->post("row/a");
            $user_ids = $this->request->post("user_ids");
            if ($params) {
                try {
                    //是否采用模型验证
                    if ($this->modelValidate) {
                        $name = str_replace("\\model\\", "\\validate\\", get_class($this->model));
                        $validate = is_bool($this->modelValidate) ? ($this->modelSceneValidate ? $name . '.edit' : $name) : $this->modelValidate;
                        $row->validate($validate);
                    }
                    $userModel = new User();
                    if(!empty($user_ids)){
                        $users = explode(',',$user_ids);
                    }else{
                        $users = $userModel->where(['status'=>'normal'])->column('id');
                    }
                    $result = $row->allowField(true)->save($params);
                    $sameUsers = array_intersect($userIds,$users);
                    $needDeleteUsersIds = array_diff($userIds, $sameUsers);
                    $newUsersIds        = array_diff($users, $sameUsers);
                    $logList = [];
                    foreach ($newUsersIds as $value){
                        $log['message_id'] = $ids;
                        $log['createtime'] = time();
                        $log['user_id'] = $value;
                        array_push($logList,$log);
                    }
                    //删除旧数据
                    $row2 = true;
                    if(!empty($needDeleteUsersIds)){
                        $row2 = Db::name('message_log')->where(['user_id'=>['in',$needDeleteUsersIds],'message_id'=>$ids])->delete();
                    }
                    //新增新数据
                    $row = true;
                    if(!empty($logList)){
                        $row = Db::name('message_log')->insertAll($logList);
                    }

                    if ($result !== false && $row && $row2) {
                        $this->success();
                    } else {
                        $this->error($row->getError());
                    }
                } catch (\think\exception\PDOException $e) {
                    $this->error($e->getMessage());
                } catch (\think\Exception $e) {
                    $this->error($e->getMessage());
                }
            }
            $this->error(__('Parameter %s can not be empty', ''));
        }
        $this->view->assign("userIds", implode(',',$userIds));
        $this->view->assign("row", $row);
        return $this->view->fetch();
    }

    /**
     * 删除
     */
    public function del($ids = "")
    {
        if ($ids) {
            $pk = $this->model->getPk();
            $adminIds = $this->getDataLimitAdminIds();
            if (is_array($adminIds)) {
                $count = $this->model->where($this->dataLimitField, 'in', $adminIds);
            }
            $list = $this->model->where($pk, 'in', $ids)->select();
            $count = 0;
            foreach ($list as $k => $v) {
                $count += $v->delete();
                Db::name('message_log')->where(['message_id'=>$v['id']])->delete();
            }
            if ($count) {
                $this->success();
            } else {
                $this->error(__('No rows were deleted'));
            }
        }
        $this->error(__('Parameter %s can not be empty', 'ids'));
    }

}