package entities import ( "fmt" "time" "github.com/google/uuid" "gorm.io/gorm" ) // EnterpriseInfo 企业信息实体 // 存储用户在认证过程中验证后的企业信息,认证完成后不可修改 // 与用户是一对一关系,每个用户最多对应一个企业信息 type EnterpriseInfo struct { // 基础标识 ID string `gorm:"primaryKey;type:varchar(36)" json:"id" comment:"企业信息唯一标识"` UserID string `gorm:"type:varchar(36);not null;uniqueIndex" json:"user_id" comment:"关联用户ID"` // 企业四要素 - 企业认证的核心信息 CompanyName string `gorm:"type:varchar(255);not null" json:"company_name" comment:"企业名称"` UnifiedSocialCode string `gorm:"type:varchar(50);not null;index" json:"unified_social_code" comment:"统一社会信用代码"` LegalPersonName string `gorm:"type:varchar(100);not null" json:"legal_person_name" comment:"法定代表人姓名"` LegalPersonID string `gorm:"type:varchar(50);not null" json:"legal_person_id" comment:"法定代表人身份证号"` // 认证状态 - 各环节的验证结果 IsOCRVerified bool `gorm:"default:false" json:"is_ocr_verified" comment:"OCR验证是否通过"` IsFaceVerified bool `gorm:"default:false" json:"is_face_verified" comment:"人脸识别是否通过"` IsCertified bool `gorm:"default:false" json:"is_certified" comment:"是否已完成认证"` VerificationData string `gorm:"type:text" json:"verification_data,omitempty" comment:"验证数据(JSON格式)"` // OCR识别结果 - 从营业执照中自动识别的信息 OCRRawData string `gorm:"type:text" json:"ocr_raw_data,omitempty" comment:"OCR原始返回数据(JSON格式)"` OCRConfidence float64 `gorm:"type:decimal(5,2)" json:"ocr_confidence,omitempty" comment:"OCR识别置信度(0-1)"` // 认证完成时间 CertifiedAt *time.Time `json:"certified_at,omitempty" comment:"认证完成时间"` // 时间戳字段 CreatedAt time.Time `gorm:"autoCreateTime" json:"created_at" comment:"创建时间"` UpdatedAt time.Time `gorm:"autoUpdateTime" json:"updated_at" comment:"更新时间"` DeletedAt gorm.DeletedAt `gorm:"index" json:"-" comment:"软删除时间"` // 关联关系 User *User `gorm:"foreignKey:UserID" json:"user,omitempty" comment:"关联的用户信息"` } // TableName 指定数据库表名 func (EnterpriseInfo) TableName() string { return "enterprise_infos" } // IsComplete 检查企业四要素是否完整 // 验证企业名称、统一社会信用代码、法定代表人姓名、身份证号是否都已填写 func (e *EnterpriseInfo) IsComplete() bool { return e.CompanyName != "" && e.UnifiedSocialCode != "" && e.LegalPersonName != "" && e.LegalPersonID != "" } // Validate 验证企业信息是否有效 // 这里可以添加企业信息的业务验证逻辑 // 比如统一社会信用代码格式验证、身份证号格式验证等 func (e *EnterpriseInfo) Validate() error { if !e.IsComplete() { return fmt.Errorf("企业信息不完整") } // 这里可以添加企业信息的业务验证逻辑 // 比如统一社会信用代码格式验证、身份证号格式验证等 return nil } // IsFullyVerified 检查是否已完成所有验证 func (e *EnterpriseInfo) IsFullyVerified() bool { return e.IsOCRVerified && e.IsFaceVerified && e.IsCertified } // UpdateOCRVerification 更新OCR验证状态 func (e *EnterpriseInfo) UpdateOCRVerification(isVerified bool, rawData string, confidence float64) { e.IsOCRVerified = isVerified e.OCRRawData = rawData e.OCRConfidence = confidence } // UpdateFaceVerification 更新人脸识别验证状态 func (e *EnterpriseInfo) UpdateFaceVerification(isVerified bool) { e.IsFaceVerified = isVerified } // CompleteCertification 完成认证 func (e *EnterpriseInfo) CompleteCertification() { e.IsCertified = true now := time.Now() e.CertifiedAt = &now } // IsReadOnly 检查企业信息是否只读(认证完成后不可修改) func (e *EnterpriseInfo) IsReadOnly() bool { return e.IsCertified } // BeforeCreate GORM钩子:创建前自动生成UUID func (e *EnterpriseInfo) BeforeCreate(tx *gorm.DB) error { if e.ID == "" { e.ID = uuid.New().String() } return nil }