package models import ( "database/sql" "github.com/blockloop/scan" "time" ) type User struct { UserID int64 Username *string SendKey string ReadKey string AdminKey string TimestampCreated time.Time TimestampLastRead *time.Time TimestampLastSent *time.Time MessagesSent int QuotaToday int QuotaDay *string IsPro bool ProToken *string } func (u User) JSON() UserJSON { return UserJSON{ UserID: u.UserID, Username: u.Username, ReadKey: u.ReadKey, SendKey: u.SendKey, AdminKey: u.AdminKey, TimestampCreated: u.TimestampCreated.Format(time.RFC3339Nano), TimestampLastRead: timeOptFmt(u.TimestampLastRead, time.RFC3339Nano), TimestampLastSent: timeOptFmt(u.TimestampLastSent, time.RFC3339Nano), MessagesSent: u.MessagesSent, QuotaToday: u.QuotaToday, QuotaDay: u.QuotaDay, IsPro: u.IsPro, } } type UserJSON struct { UserID int64 `json:"user_id"` Username *string `json:"username"` ReadKey string `json:"read_key"` SendKey string `json:"send_key"` AdminKey string `json:"admin_key"` TimestampCreated string `json:"timestamp_created"` TimestampLastRead *string `json:"timestamp_last_read"` TimestampLastSent *string `json:"timestamp_last_sent"` MessagesSent int `json:"messages_sent"` QuotaToday int `json:"quota_today"` QuotaDay *string `json:"quota_day"` IsPro bool `json:"is_pro"` } type UserDB struct { UserID int64 `db:"user_id"` Username *string `db:"username"` SendKey string `db:"send_key"` ReadKey string `db:"read_key"` AdminKey string `db:"admin_key"` TimestampCreated int64 `db:"timestamp_created"` TimestampLastRead *int64 `db:"timestamp_lastread"` TimestampLastSent *int64 `db:"timestamp_lastsent"` MessagesSent int `db:"messages_sent"` QuotaToday int `db:"quota_today"` QuotaDay *string `db:"quota_day"` IsPro bool `db:"is_pro"` ProToken *string `db:"pro_token"` } func (u UserDB) Model() User { return User{ UserID: u.UserID, Username: u.Username, SendKey: u.SendKey, ReadKey: u.ReadKey, AdminKey: u.AdminKey, TimestampCreated: time.UnixMilli(u.TimestampCreated), TimestampLastRead: timeOptFromMilli(u.TimestampLastRead), TimestampLastSent: timeOptFromMilli(u.TimestampLastSent), MessagesSent: u.MessagesSent, QuotaToday: u.QuotaToday, QuotaDay: u.QuotaDay, IsPro: u.IsPro, } } func DecodeUser(r *sql.Rows) (User, error) { var udb UserDB err := scan.RowStrict(&udb, r) if err != nil { return User{}, err } return udb.Model(), nil }