Generalize updating entry and feed with update method

feed.go

@@ -26,6 +26,15 @@ type Feed struct {
 }
 
 // NewFeed creates a new Feed. It returns a *Feed.
+// update sets the Updated time to time.Now.
+func (f *Feed) update() {
+	if f.Updated == nil {
+		f.Updated = NewDate(time.Now())
+	} else {
+		f.Updated.DateTime = DateTime(time.Now())
+	}
+}
+
 func NewFeed(title string) *Feed {
 	return &Feed{
 		CommonAttributes: NewCommonAttributes(),
@@ -38,13 +47,8 @@ func NewFeed(title string) *Feed {
 // AddAuthor adds the Person as an author to the Feed. It returns its index as
 // an int.
 func (f *Feed) AddAuthor(p *Person) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
 	return addToSlice(&f.Authors, p)
+	f.update()
 }
 
 // DeleteAuthor deletes the Person at index from the Feed. It return an error.
@@ -53,23 +57,13 @@ func (f *Feed) DeleteAuthor(index int) error {
 		return fmt.Errorf("error deleting author %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
 // AddCategory adds the Category to the Feed. It returns its index as an int.
 func (f *Feed) AddCategory(c *Category) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return addToSlice(&f.Categories, c)
 }
 
@@ -80,24 +74,14 @@ func (f *Feed) DeleteCategory(index int) error {
 		return fmt.Errorf("error deleting category %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
 // AddContributor adds the Person as a contributor to the Feed. It returns its
 // index as an int.
 func (f *Feed) AddContributor(c *Person) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return addToSlice(&f.Contributors, c)
 }
 
@@ -108,24 +92,14 @@ func (f *Feed) DeleteContributor(index int) error {
 		return fmt.Errorf("error deleting contributor %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
 // AddLink adds the Link to the Feed. There should be one Link with Rel "self".
 // It returns its index as an int.
 func (f *Feed) AddLink(l *Link) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return addToSlice(&f.Links, l)
 }
 
@@ -135,23 +109,13 @@ func (f *Feed) DeleteLink(index int) error {
 		return fmt.Errorf("error deleting link %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
 // AddExtension adds the Extension to the Feed. It returns its index as an int.
 func (f *Feed) AddExtension(e *ExtensionElement) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return addToSlice(&f.Extensions, e)
 }
 
@@ -162,23 +126,13 @@ func (f *Feed) DeleteExtension(index int) error {
 		return fmt.Errorf("error deleting extension %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
 // AddEntry adds the Entry to the Feed. It returns its index as an int.
 func (f *Feed) AddEntry(e *Entry) int {
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return addToSlice(&f.Entries, e)
 }
 
@@ -188,12 +142,7 @@ func (f *Feed) DeleteEntry(index int) error {
 		return fmt.Errorf("error deleting entry %v from entry %v: %v", index, f.ID.URI, err)
 	}
 
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
-
+	f.update()
 	return nil
 }
 
@@ -215,11 +164,7 @@ func (f *Feed) DeleteEntryByURI(uri string) error {
 	}
 
 	f.Entries = append(f.Entries[:index], f.Entries[index+1:]...)
-	if f.Updated == nil {
-		f.Updated = NewDate(time.Now())
-	} else {
-		f.Updated.DateTime = DateTime(time.Now())
-	}
+	f.update()
 
 	return nil
 }