ThinkAdmin/app/data/service/UserService.php

<?php

namespace app\data\service;

use think\admin\Service;

/**
 * 用户数据接口服务
 * Class UserService
 * @package app\store\service
 */
class UserService extends Service
{
    /**
     * 绑定数据表
     * @var string
     */
    protected $table = 'DataUser';

    /**
     * 获取用户资料
     * @param mixed $map 查询条件
     * @param boolean $force 刷新令牌
     * @return array
     * @throws \think\Exception
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function get($map, bool $force = false): array
    {
        if (is_numeric($map)) {
            $map = ['id' => $map];
        } elseif (is_string($map)) {
            $map = ['token|openid1|openid2|unionid' => $map];
        }
        $user = $this->save($map, [], $force);
        if (empty($user)) throw new \think\Exception('登录授权失败');
        // if ($member['tokenv'] !== $this->buildTokenVerify()) {
        //     throw new \think\Exception('请重新登录授权');
        // }
        return $user;
    }

    /**
     * 更新用户用户参数
     * @param array $map 查询条件
     * @param array $data 更新数据
     * @param boolean $force 强刷令牌
     * @return array
     * @throws \think\db\exception\DataNotFoundException
     * @throws \think\db\exception\DbException
     * @throws \think\db\exception\ModelNotFoundException
     */
    public function save(array $map, array $data = [], bool $force = false): array
    {
        $query = $this->app->db->name($this->table)->where($map);
        $member = $query->withoutField('deleted,password')->where(['deleted' => 0])->find() ?: [];
        unset($data['id'], $data['token'], $data['tokenv'], $data['deleted'], $data['create_at']);
        if (empty($data['phone']) && empty($data['unionid']) && empty($data['openid1']) && empty($data['openid2'])) {
            return $member;
        }
        if ($force) $data = array_merge($data, $this->_buildUserToken());
        if (isset($member['id']) && $member['id'] > 0) {
            $map = ['id' => $member['id'], 'deleted' => 0];
            $this->app->db->name($this->table)->strict(false)->where($map)->update($data);
        } else {
            $member['id'] = $this->app->db->name($this->table)->strict(false)->insertGetId($data);
        }
        $map = ['id' => $member['id'], 'deleted' => 0];
        $query = $this->app->db->name($this->table)->where($map);
        return $query->withoutField('deleted,password')->find() ?: [];
    }

    /**
     * 获取用户数据统计
     * @param int $mid 用户UID
     * @return array
     */
    public function total(int $mid): array
    {
        $query = $this->app->db->name($this->table);
        return ['my_invite' => $query->where(['from' => $mid])->count()];
    }

    /**
     * 生成新的用户令牌
     * @return array
     */
    private function _buildUserToken(): array
    {
        do $map = ['token' => md5(uniqid('', true) . rand(100, 999))];
        while ($this->app->db->name($this->table)->where($map)->count() > 0);
        return ['token' => $map['token'], 'tokenv' => $this->_buildTokenVerify()];
    }

    /**
     * 获取令牌的认证值
     * @return string
     */
    private function _buildTokenVerify(): string
    {
        return md5($this->app->request->server('HTTP_USER_AGENT', '-'));
    }

}