Thinking--validatorjs中isIP的巧妙实现

Posted 2022-07-30 奋飛

Thinking系列，旨在利用10分钟的时间传达一种可落地的编程思想。

isIP 是 validatorjs 中经常被使用到的验证方法，关于该验证方法具体是如何实现的？validatorjs 中又有哪些好的点可以借鉴？

方法	说明
isIP(str [, version])	check if the string is an IP (version 4 or 6)

version 作为可选项，如果不传递任何值，需要进行 v4 或 v6 的判断，任何一个通过即可。

模拟实现

1. 编写两个冗长的正则 IPv4RegExp、IPv6RegExp
2. 对 version 进行判断

   if (version === '4')  return IPv4RegExp.test(str)
   else if (version === '6')  return IPv6RegExp.test(str)
   else  return IPv4RegExp.test(str) || IPv6RegExp.test(str)
   

上述方式有两个问题：

（1）IPv4RegExp、IPv6RegExp 正则的编写可能比较冗长，且兼容性、效率可能有待商榷；
（2）代码整洁度不够，特别是 if...else if...else 的结构；

isIP(str [, version])

检查字符串是否为 IP（版本 4 或 6）。源码地址

export default function isIP(str, version = '') 
  version = String(version);
  if (!version) 
    return isIP(str, 4) || isIP(str, 6);
  
  if (version === '4') 
    if (!IPv4AddressRegExp.test(str)) 
      return false;
    
    const parts = str.split('.').sort((a, b) => a - b);
    return parts[3] <= 255;
  
  if (version === '6') 
    return !!IPv6AddressRegExp.test(str);
  
  return false;

关于 IPv4AddressRegExp 的实现：

const IPv4SegmentFormat = '(?:[0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5])';
const IPv4AddressFormat = `($IPv4SegmentFormat[.])3$IPv4SegmentFormat`;
const IPv4AddressRegExp = new RegExp(`^$IPv4AddressFormat$`)

version 为空的情况，v4、v6任一通过即可

if (IPv4AddressRegExp.test(str) || IPv6AddressRegExp.test(str)) 

不做特殊分支，而是依附于当前分支情况

return isIP(str, 4) || isIP(str, 6)