PHP-数组黑名单白名单

php 里经常通过 join 查询得到的数据, 会有这种情况:

$data = [{id=>1,name=>a,age=>a},
    {id=>1,name=>b,age=>b},
    {id=>2,name=>c,age=>c},
    {id=>3,name=>f,age=>f},
    {id=>2,name=>d,age=>d},
];

这种格式的数据其实很常见, 实际上我们想要的数据效果是:

$data = [[id=>1,user_info=>[[name=>a,age=>a],[name=>b,age=>b]]],
    [id=>2,user_info=>[[name=>c,age=>c],[name=>d,age=>d]]],
    [id=>3,user_info=>[[name=>f,age=>f]]],
];

于是我们需要各种 foreach 处理其中函数;
这里记录一下, 两个函数
数组的黑名单:

/**
 * 数组获取黑名单内的数据
 * @param array $arr 源数组
 * @param array $list 白名单
 * @return array
 * @example
 * $arr = ['a'=>1,'b'=>2,'c'=>3,'d'=>4]
 * $list = ['d','c'];
 * $return = ['a'=>1,'b'=>2];
 */
public function array_black_list(Array $arr, Array $list)
{return array_diff_key($arr, array_flip($list));
}

数组的白名单

/**
 * 数组获取白名单内的数据
 * @param array $arr 源数组
 * @param array $list 白名单
 * @return array
 * @example
 * $arr = ['a'=>1,'b'=>2,'c'=>3,'d'=>4]
 * $list = ['d','c'];
 * $return = ['c'=>3,'d'=>4];
 */
public function array_white_list(Array $arr, Array $list)
{return array_intersect_key($arr, array_flip($list));
}

利用黑名单和白名单, 可以很容易实现上述数据格式转换:

/**
 * repeat 数据处理(一般是 leftJoin 数据结构完善)
 * leftJoin 查询出来的多组数据, 根据 primary_id 进行数据结构修改, 保存为  字段 =>[{},{},{}]类似结构
 * 效果:$data = [{id=>1,name=>a,age=>a},{id=>1,name=>b,age=>b},{id=>2,name=>c,age=>c},{id=>3,name=>null,age=>null}],$fieldArr = ['name','age']
 * return [{id=>1,repeat_info=>[{name=>a,age=>a},{name=>b,age=>b}]},{id=>2,repeat_info=>[{name=>c,age=>c}]},{id=>3,repeat_info=>[]}]
 * @param $data    array 原数据;
 * @param $fieldArr array 需要的字段,
 * @param string $primary_key 重复的数据 凭证;(一般用主键 id)
 * @param string $repeat_name 重复的数据要保存数组的 字段名
 * @return array
 */
public function dealRepeatData(Array $data, Array $fieldArr, $primary_key = 'id', $repeat_name = 'repeat_info')
{$res = [];
    // 通过黑名单获取不重复的值, 归并到按 id 分类的数组里;
    foreach ($data as $nre) {$res[$nre[$primary_key]] = $res[$nre[$primary_key]] ?? $this->array_black_list($nre, $fieldArr);
    }
    // 通过白名单获取重复的值, 归并到黑名单的 $repeat_name 字段里;
    foreach ($data as $re) {$res[$re[$primary_key]][$repeat_name][] = $this->array_white_list($re, $fieldArr);
    }

    return array_values($res);
}