package models

import "gorm.io/gorm"

type Device struct {
	Id         string `gorm:"column:id;primary_key;" json:"id"`
	Name       string `gorm:"column:name" json:"name"`
	Addr       string `gorm:"column:addr" json:"addr"`
	Pos        string `gorm:"column:pos" json:"pos"`
	Ext        string `gorm:"column:ext" json:"ext"`
	Ip         string `gorm:"column:ip" json:"ip"`
	CreateTime int64  `gorm:"column:create_time;autoCreateTime;" json:"create_time"`
	UpdateTime int64  `gorm:"column:update_time;autoUpdateTime" json:"update_time"`
	DeleteTime int64  `gorm:"column:delete_time" json:"delete_time"`
}

func (d *Device) TableName() string {
	return "devices"
}

func (d *Device) FindById(id string) error {
	return db.Table(d.TableName()).First(&d, "id = ?", id).Error
}

func (d *Device) FindAll() ([]Device, error) {
	var devices []Device
	if err := db.Table(d.TableName()).Find(&devices).Error; err != nil {
		return nil, err
	}

	return devices, nil
}

func (d *Device) Upsert() error {
	var existingDevice Device
	if err := db.Table(d.TableName()).Where("id = ?", d.Id).First(&existingDevice).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			// 记录不存在，创建新记录
			db.Create(&d)
		} else {
			// 其他错误，你可以进行适当的处理
			return err
		}
	} else {
		// 记录存在，更新现有记录
		if d.Pos == "" || d.Ip == "" {
			return nil
		}

		existingDevice.Pos = d.Pos
		existingDevice.Ip = d.Ip

		db.Save(&existingDevice)
	}

	return nil
}