package db
|
|
import (
|
"basic.com/valib/logger.git"
|
"errors"
|
"gorm.io/gorm"
|
)
|
|
// 查询小区表
|
func GetCommunityIDs() ([]string, error) {
|
// 查询数据
|
var db = DB
|
var communityIDs []string
|
result := db.Table("domain_unit").Where("domainType = ?", 1).Pluck("id", &communityIDs)
|
if result.Error != nil {
|
return nil, result.Error
|
}
|
|
return communityIDs, nil
|
}
|
|
// 查询设备表
|
func GetDeviceData() ([]Device, error) {
|
// 查询数据
|
var db = DB
|
var device []Device
|
result := db.Table("device").Where("communityID != ?", "").Find(&device)
|
if result.Error != nil {
|
return nil, result.Error
|
}
|
|
return device, nil
|
}
|
|
// 查询全部数据
|
func GetAllTaskData() ([]Task, error) {
|
var db = DB
|
var task []Task
|
if err := db.Find(&task).Error; err != nil {
|
return nil, err
|
}
|
|
return task, nil
|
}
|
|
// 查询全部数据
|
func GetAllData() ([]PersonnelStatusRule, error) {
|
var db = DB
|
var rules []PersonnelStatusRule
|
if err := db.Find(&rules).Error; err != nil {
|
return nil, err
|
}
|
|
return rules, nil
|
}
|
|
// 查询住户时间数据
|
func GetResidentData(status, communityID string) ([]Resident, error) {
|
var residents []Resident
|
//var db = DB.Debug()
|
var db = DB
|
// 执行查询
|
rows, err := db.Table("snapshot_count_summary").
|
Select("document_number", "community_id", "last_appearance_time", "created_at").
|
Where("status = ? AND community_id = ?", status, communityID).
|
//Where("snapshot_count_summary.created_at is not null").
|
Rows()
|
if err != nil {
|
return nil, err
|
}
|
defer rows.Close()
|
|
// 遍历查询结果
|
for rows.Next() {
|
var resident Resident
|
err := rows.Scan(&resident.DocumentNumber, &resident.CommunityId, &resident.LastAppearanceTime, &resident.CreateAt)
|
if err != nil {
|
logger.Error("err: ", err)
|
return nil, err
|
}
|
//fmt.Println("resident111: ", resident)
|
residents = append(residents, resident)
|
}
|
if err := rows.Err(); err != nil {
|
return nil, err
|
}
|
|
return residents, nil
|
}
|
|
// 查询人物属性
|
func GetDBPersonStatusData(id string) ([]PersonStatus, error) {
|
var db = DB
|
// 查询数据
|
var personStatusList []PersonStatus
|
if err := db.Table("person_status").
|
Select("documentNumber, status, frequentAddress").
|
Where("communityID = ?", id).
|
Find(&personStatusList).Error; err != nil {
|
return nil, err
|
}
|
|
return personStatusList, nil
|
}
|
|
// 查询人物年龄
|
func GetAgeById(id string) (int, error) {
|
var db = DB
|
// 查询数据
|
var age int
|
if err := db.Table("dbtablepersons").
|
Select("age").
|
Where("id = ?", id).
|
Find(&age).Error; err != nil {
|
return 0, err
|
}
|
|
return age, nil
|
}
|
|
//// 根据社区id和住户属性查询住户档案编号
|
//func GetDocNumberFromPersonStatus(id, status string) ([]string, error) {
|
// var db = DB
|
// // 查询数据
|
// var personStatusList []PersonStatus
|
// if err := db.Table("person_status").
|
// Select("documentNumber, status, frequentAddress").
|
// Where("communityID = ? AND status = ?", id, status).
|
// Find(&personStatusList).Error; err != nil {
|
// return nil, err
|
// }
|
//
|
// docNum := make([]string, 0)
|
// for _, ps := range personStatusList {
|
// docNum = append(docNum, ps.DocumentNumber)
|
// }
|
//
|
// return docNum, nil
|
//}
|
|
// 查询人物身份属性表
|
func GetLabelManageIdentity(IdentityType int) ([]LabelManage, error) {
|
var db = DB
|
// 查询数据
|
var labelManageIdentity []LabelManage
|
if err := db.Table("label_manage").
|
Select("id, name, valid_days").
|
Where("type = ?", IdentityType).
|
Find(&labelManageIdentity).Error; err != nil {
|
return nil, err
|
}
|
|
return labelManageIdentity, nil
|
}
|
|
// 根据dbtablepersons表id查询目标档案年龄
|
func QueryAgeById(id string) (int, error) {
|
var db = DB
|
var age int
|
err := db.Table("dbtablepersons").
|
Select("age").
|
Where("id = ?", id).
|
Scan(&age).Error
|
if err != nil {
|
return 0, err
|
}
|
return age, nil
|
}
|
|
// UpdatePersonInfo 更新或插入多个人员信息
|
func UpdateMoveInout(personsMoveInout []MoveInout) error {
|
var db = DB
|
// 遍历人员信息
|
for _, personMoveInout := range personsMoveInout {
|
|
// 检查记录是否存在
|
var existingPerson MoveInout
|
err := db.Where("document_number = ? AND community_id = ?", personMoveInout.DocumentNumber, personMoveInout.CommunityID).First(&existingPerson).Error
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
logger.Error("Query person error:", err, personMoveInout.DocumentNumber, personMoveInout.CommunityID)
|
//fmt.Println("asdasfasfasf")
|
continue
|
//return err
|
}
|
|
// 如果记录存在，则更新
|
if existingPerson.DocumentNumber != "" {
|
//fmt.Println("existingPerson.DocumentNumber: ", existingPerson.DocumentNumber)
|
if existingPerson.Status != "Verified" {
|
err := db.Model(&MoveInout{}).
|
Where("document_number = ? AND community_id = ?", personMoveInout.DocumentNumber, personMoveInout.CommunityID).
|
Updates(map[string]interface{}{
|
"status": personMoveInout.Status,
|
"move_out_date": personMoveInout.MoveOutDate,
|
}).Error
|
if err != nil {
|
return err
|
}
|
} else {
|
err := db.Model(&MoveInout{}).
|
Where("document_number = ? AND community_id = ?", personMoveInout.DocumentNumber, personMoveInout.CommunityID).
|
Updates(map[string]interface{}{
|
"move_out_date": personMoveInout.MoveOutDate,
|
}).Error
|
if err != nil {
|
return err
|
}
|
}
|
} else {
|
// 如果记录不存在，则插入新记录
|
//fmt.Println("插入记录失败")
|
//fmt.Println("data", &personMoveInout)
|
err := db.Create(&personMoveInout).Error
|
if err != nil {
|
return err
|
}
|
}
|
|
}
|
|
return nil
|
}
|
|
// UpdatePersonInfo 更新或插入多个人员身份信息
|
func UpdateDBPersonLabel(personsIdentity []Identity) error {
|
var db = DB
|
// 遍历人员信息
|
for _, personIdentity := range personsIdentity {
|
|
// 检查记录是否存在
|
var existingPerson Identity
|
err := db.Where("dbtablepersons_id = ? AND community_id = ? AND label_id = ?",
|
personIdentity.DocumentNumber,
|
personIdentity.CommunityID,
|
personIdentity.LabelId,
|
).First(&existingPerson).Error
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
logger.Error("Query person error:", err,
|
personIdentity.DocumentNumber,
|
personIdentity.CommunityID,
|
personIdentity.LabelId)
|
//fmt.Println("asdasfasfasf")
|
continue
|
//return err
|
}
|
|
// 如果记录存在，则更新
|
if existingPerson.DocumentNumber != "" {
|
err := db.Model(&Identity{}).
|
Where("dbtablepersons_id = ? AND community_id = ? AND label_id = ?",
|
personIdentity.DocumentNumber,
|
personIdentity.CommunityID,
|
personIdentity.LabelId,
|
).
|
Updates(map[string]interface{}{
|
"expire_time": personIdentity.ExpireTime,
|
}).Error
|
if err != nil {
|
return err
|
}
|
} else {
|
// 如果记录不存在，则插入新记录
|
err := db.Create(&personIdentity).Error
|
if err != nil {
|
return err
|
}
|
}
|
|
}
|
|
return nil
|
}
|
|
// UpdatePersonInfo 更新或插入多个人员信息
|
func UpdatePersonInfo(persons []PersonStatus, communityID string) error {
|
var db = DB
|
// 遍历人员信息
|
for _, person := range persons {
|
|
// 检查记录是否存在
|
var existingPerson PersonStatus
|
err := db.Where("document_number = ? AND community_id = ?", person.DocumentNumber, communityID).First(&existingPerson).Error
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
logger.Error("Query person error:", err, person.DocumentNumber, communityID)
|
//fmt.Println("asdasfasfasf")
|
continue
|
//return err
|
}
|
|
// 如果记录存在，则更新
|
if existingPerson.DocumentNumber != "" {
|
err := db.Model(&PersonStatus{}).
|
Where("document_number = ? AND community_id = ?", person.DocumentNumber, communityID).
|
Updates(map[string]interface{}{
|
"status": person.Status,
|
"frequent_address": person.FrequentAddress,
|
}).Error
|
if err != nil {
|
return err
|
}
|
}
|
//else {
|
// // 如果记录不存在，则插入新记录
|
// err := db.Create(&person).Error
|
// if err != nil {
|
// return err
|
// }
|
//}
|
}
|
|
return nil
|
}
|
