count_peer() support custom field

This commit is contained in:
xiaomlove
2022-04-20 02:44:28 +08:00
parent 41ea4fd2a1
commit 7a79673669
3 changed files with 7 additions and 7 deletions

View File

@@ -705,20 +705,20 @@ function isIPV6 ($ip)
return filter_var($ip,FILTER_VALIDATE_IP, FILTER_FLAG_IPV6);
}
function count_peer($whereStr)
function count_peer($whereStr, $field = 'peer_id')
{
if (empty($whereStr)) {
throw new \InvalidArgumentException("require whereStr");
}
if (IN_NEXUS) {
$sql = "select count(distinct(peer_id)) as counts from peers where $whereStr";
$sql = "select count(distinct($field)) as counts from peers where $whereStr";
$res = sql_query($sql);
$count = mysql_fetch_assoc($res);
return $count['counts'];
} else {
$res = \Illuminate\Support\Facades\DB::table('peers')
->whereRaw($whereStr)
->selectRaw("count(distinct(peer_id)) as counts")
->selectRaw("count(distinct($field)) as counts")
->first();
return $res->counts;
}