Merge pull request #35 from dipdup-io/fix/update-state

Fix: update state

Author: aopoltorzhicky

database/db_test.go

@@ -18,7 +18,7 @@ const (
 	testIndex = "test_index"
 )
 
-func newDatabase(ctx context.Context, typ string, cfg config.Database) (Database, error) {
+func newDatabase(_ context.Context, typ string, _ config.Database) (Database, error) {
 	switch typ {
 	case "bun":
 		return NewBun(), nil

database/state.go

@@ -22,17 +22,18 @@ type State struct {
 	CreatedAt int       `gorm:"autoCreateTime" comment:"Created timestamp"`
 }
 
-// BeforeInsert -
-func (s *State) BeforeInsert(ctx context.Context) (context.Context, error) {
-	s.UpdatedAt = int(time.Now().Unix())
-	s.CreatedAt = s.UpdatedAt
-	return ctx, nil
-}
-
-// BeforeUpdate -
-func (s *State) BeforeUpdate(ctx context.Context) (context.Context, error) {
-	s.UpdatedAt = int(time.Now().Unix())
-	return ctx, nil
+var _ bun.BeforeAppendModelHook = (*State)(nil)
+
+func (s *State) BeforeAppendModel(ctx context.Context, query bun.Query) error {
+	switch query.(type) {
+	case *bun.InsertQuery:
+		s.UpdatedAt = int(time.Now().Unix())
+		s.CreatedAt = s.UpdatedAt
+
+	case *bun.UpdateQuery:
+		s.UpdatedAt = int(time.Now().Unix())
+	}
+	return nil
 }
 
 // TableName -