// SaveUser creates or updates a user. Upon creation, users are assigned a unique ID.
func (db *DynamoDB) SaveUser(u *models.User) error {
if u.ID == "" {
id, err := shortid.ID(64)
if err != nil {
return err
}
u.ID = id
}
item, err := dynamodbattribute.MarshalMap(u)
if err != nil {
return err
}
_, err = db.DB.PutItem(&dynamodb.PutItemInput{
TableName: aws.String("users"),
Item: item,
})
if err != nil {
return err
}
return nil
}
// SaveUser save or updates a user object in the database.
func (db UserDB) SaveUser(u *models.User) error {
if u.ID == "" {
u.ID = strconv.Itoa(len(db.ids) + 1)
}
db.ids[u.ID] = u
db.emails[u.Email] = u
return nil
}
