diff --git a/internal/entity/country.go b/internal/entity/country.go index 63744a074..26c280461 100644 --- a/internal/entity/country.go +++ b/internal/entity/country.go @@ -69,7 +69,8 @@ func (m *Country) FirstOrCreate() *Country { // AfterCreate sets the New column used for database callback func (m *Country) AfterCreate(scope *gorm.Scope) error { - return scope.SetColumn("New", true) + m.New = true + return nil } // Code returns country code diff --git a/internal/entity/label.go b/internal/entity/label.go index af522be33..a528896ac 100644 --- a/internal/entity/label.go +++ b/internal/entity/label.go @@ -72,7 +72,8 @@ func (m *Label) FirstOrCreate() *Label { // AfterCreate sets the New column used for database callback func (m *Label) AfterCreate(scope *gorm.Scope) error { - return scope.SetColumn("New", true) + m.New = true + return nil } // SetName changes the label name. diff --git a/internal/entity/label_test.go b/internal/entity/label_test.go index e68e049ef..1a2afa5b9 100644 --- a/internal/entity/label_test.go +++ b/internal/entity/label_test.go @@ -70,6 +70,7 @@ func TestLabel_Update(t *testing.T) { assert.Equal(t, "label", Label.LabelName) err := Label.Update(*classifyLabel) + if err != nil { t.Fatal(err) } diff --git a/internal/entity/place.go b/internal/entity/place.go index 6ec3c4084..61487ea10 100644 --- a/internal/entity/place.go +++ b/internal/entity/place.go @@ -44,7 +44,8 @@ func CreateUnknownPlace() { // AfterCreate sets the New column used for database callback func (m *Place) AfterCreate(scope *gorm.Scope) error { - return scope.SetColumn("New", true) + m.New = true + return nil } // FindPlaceByLabel returns a place from an id or a label