95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
package idvalidator
|
|
|
|
import (
|
|
"errors"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// 检查ID参数
|
|
func CheckIDArgument(id string) bool {
|
|
_, err := GenerateCode(id)
|
|
|
|
return err == nil
|
|
}
|
|
|
|
// 生成数据
|
|
func GenerateCode(id string) (map[string]string, error) {
|
|
length := len(id)
|
|
if length == 15 {
|
|
return GenerateShortCode(id)
|
|
}
|
|
|
|
if length == 18 {
|
|
return GenerateLongCode(id)
|
|
}
|
|
|
|
return map[string]string{}, errors.New("Invalid ID card number length.")
|
|
}
|
|
|
|
// 生成短数据
|
|
func GenerateShortCode(id string) (map[string]string, error) {
|
|
if len(id) != 15 {
|
|
return map[string]string{}, errors.New("Invalid ID card number length.")
|
|
}
|
|
|
|
mustCompile := regexp.MustCompile("(.{6})(.{6})(.{3})")
|
|
subMatch := mustCompile.FindStringSubmatch(strings.ToLower(id))
|
|
|
|
return map[string]string{
|
|
"body": subMatch[0],
|
|
"addressCode": subMatch[1],
|
|
"birthdayCode": "19" + subMatch[2],
|
|
"order": subMatch[3],
|
|
"checkBit": "",
|
|
"type": "15",
|
|
}, nil
|
|
}
|
|
|
|
// 生成长数据
|
|
func GenerateLongCode(id string) (map[string]string, error) {
|
|
if len(id) != 18 {
|
|
return map[string]string{}, errors.New("Invalid ID card number length.")
|
|
}
|
|
mustCompile := regexp.MustCompile("((.{6})(.{8})(.{3}))(.)")
|
|
subMatch := mustCompile.FindStringSubmatch(strings.ToLower(id))
|
|
|
|
return map[string]string{
|
|
"body": subMatch[1],
|
|
"addressCode": subMatch[2],
|
|
"birthdayCode": subMatch[3],
|
|
"order": subMatch[4],
|
|
"checkBit": subMatch[5],
|
|
"type": "18",
|
|
}, nil
|
|
}
|
|
|
|
// 检查地址码
|
|
func CheckAddressCode(addressCode string, birthdayCode string) bool {
|
|
return GetAddressInfo(addressCode, birthdayCode)["province"] != ""
|
|
}
|
|
|
|
// 检查出生日期码
|
|
func CheckBirthdayCode(birthdayCode string) bool {
|
|
year, _ := strconv.Atoi(Substr(birthdayCode, 0, 4))
|
|
if year < 1800 {
|
|
return false
|
|
}
|
|
|
|
nowYear := time.Now().Year()
|
|
if year > nowYear {
|
|
return false
|
|
}
|
|
|
|
_, err := time.Parse("20060102", birthdayCode)
|
|
|
|
return err == nil
|
|
}
|
|
|
|
// 检查顺序码
|
|
func CheckOrderCode(orderCode string) bool {
|
|
return len(orderCode) == 3
|
|
}
|