52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/tribalwarshelp/shared/models"
|
|
)
|
|
|
|
type lineParser struct {
|
|
location *time.Location
|
|
}
|
|
|
|
func newLineParser(timezone string) (*lineParser, error) {
|
|
loc, err := time.LoadLocation(timezone)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Invalid timezone: %s", timezone)
|
|
}
|
|
return &lineParser{
|
|
location: loc,
|
|
}, nil
|
|
}
|
|
|
|
func (parser *lineParser) parse(line []string) (*models.Ennoblement, error) {
|
|
if len(line) != 4 {
|
|
return nil, fmt.Errorf("Invalid line format (should be village_id,timestamp,new_owner,old_owner)")
|
|
}
|
|
var err error
|
|
e := &models.Ennoblement{}
|
|
e.VillageID, err = strconv.Atoi(line[0])
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "*models.Ennoblement.VillageID")
|
|
}
|
|
timestamp, err := strconv.Atoi(line[1])
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "timestamp")
|
|
}
|
|
e.EnnobledAt = time.Unix(int64(timestamp), 0).In(parser.location)
|
|
e.NewOwnerID, err = strconv.Atoi(line[2])
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "*models.Ennoblement.NewOwnerID")
|
|
}
|
|
e.OldOwnerID, err = strconv.Atoi(line[3])
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "*models.Ennoblement.OldOwnerID")
|
|
}
|
|
|
|
return e, nil
|
|
}
|