package model

import "github.com/jinzhu/gorm"

type DayData struct {
	gorm.Model
	DateStr      string `json:"dateStr"`
	WarnCount         int64 `json:"warnCount"`
}

func (obj *DayData) CreateOrSave() (err error) {
	err = DB.Model(&DayData{}).Save(&obj).Error
	if err != nil {
		return err
	}
	return nil
}

func (obj *DayData) GetByDate() (count int64, err error) {
	err = DB.Model(&DayData{}).Where("date_str = ?", obj.DateStr).First(&obj).Error
	if err != nil {
		return 0, err
	}
	return obj.WarnCount ,nil
}