kurz/storage/info.go

package storage

import (
	"net"
	"time"
)

type EventInfo struct {
	Id     int64
	Source string
	Ip     net.IP
	ts     time.Time
}

type EventInfoStorage struct {
}

func (eis EventInfoStorage) Table() string {
	return "eventinfo"
}

var EventInfoService EventInfoStorage = EventInfoStorage{}

func EventInfoLoadFromId(id int64, storage Storage) (EventInfo, error) {
	var ei EventInfo
	var source string
	var ip string
	var ts int64

	sql := `
SELECT source, ip, ts
FROM eventinfo
WHERE id = ?
	`

	err := storage.DB.QueryRow(sql, id).Scan(&source, &ip, &ts)
	if err != nil {
		ei = EventInfo{
			Id:     id,
			Source: source,
			Ip:     net.ParseIP(ip),
			ts:     time.Unix(ts, 0),
		}
	}

	return ei, err
}

/**
Save() persists an EventInfo to the storage.

  - if it already has an non-zero ID, Save() will attempt to update it, and set the Id in the event object
  - otherwise it will attempt to insert it

*/
func (ei *EventInfo) Save(s Storage) error {
	var err error

	ip := ei.Ip.String()
	ts := ei.ts.Unix()

	if ei.Id == 0 {
		sql := `
INSERT INTO eventinfo(source, ip, ts)
VALUES (?, ?, ?)
		`
		result, err := s.DB.Exec(sql, ei.Source, ip, ts)
		if err == nil {
			ei.Id, err = result.LastInsertId()
		}
	} else {
		sql := `
UPDATE eventinfo
SET source = ?, ip = ?, ts = ?
WHERE id = ?
		`
		_, err = s.DB.Exec(sql, ei.Source, ip, ts, ei.Id)
	}

	return err
}