fix update

charlie67 · Sep 21, 2022 · 9a62f37 · 9a62f37
1 parent 37fe7e1
commit 9a62f37
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/src/main/java/to/charlie/spotifyplayhistory/domain/service/SpotifyApiService.java b/src/main/java/to/charlie/spotifyplayhistory/domain/service/SpotifyApiService.java
@@ -42,10 +42,12 @@
 import to.charlie.spotifyplayhistory.domain.entity.ArtistEntity;
 import to.charlie.spotifyplayhistory.domain.entity.PlayEntity;
 import to.charlie.spotifyplayhistory.domain.entity.Token;
+import to.charlie.spotifyplayhistory.domain.entity.TrackEntity;
 import to.charlie.spotifyplayhistory.domain.repository.ArtistRepository;
 import to.charlie.spotifyplayhistory.domain.repository.MigrationRepository;
 import to.charlie.spotifyplayhistory.domain.repository.PlayRepository;
 import to.charlie.spotifyplayhistory.domain.repository.TokenRepository;
+import to.charlie.spotifyplayhistory.domain.repository.TrackRepository;
 
 
 @Service
@@ -63,17 +65,21 @@ public class SpotifyApiService
 
   private final MigrationRepository migrationRepository;
 
+  private final TrackRepository trackRepository;
+
   public SpotifyApiService(PlayRepository playRepository,
                            ArtistRepository artistRepository,
                            SpotifyProperties spotifyProperties,
                            TokenRepository tokenRepository,
                            FlywayMigrator flywayMigrator,
-                           MigrationRepository migrationRepository) throws URISyntaxException
+                           MigrationRepository migrationRepository,
+                           TrackRepository trackRepository) throws URISyntaxException
   {
     this.playRepository = playRepository;
     this.artistRepository = artistRepository;
     this.flywayMigrator = flywayMigrator;
     this.migrationRepository = migrationRepository;
+    this.trackRepository = trackRepository;
 
     Optional<Token> token = tokenRepository.findById(1);
     token.ifPresent(value -> {
@@ -223,6 +229,24 @@ private void savePlayHistory(PagingCursorbased<PlayHistory> history)
         }
       }
 
+      Optional<TrackEntity> optionalTrack = trackRepository.findById(trackId);
+      TrackEntity trackEntity;
+      if (optionalTrack.isPresent())
+      {
+        trackEntity = optionalTrack.get();
+      }
+      else
+      {
+        trackEntity = TrackEntity.builder()
+            .id(trackId)
+            .trackName(track.getName())
+            .popularity(-1)
+            .songLength(track.getDurationMs())
+            .artists(artistEntities)
+            .build();
+        trackRepository.save(trackEntity);
+      }
+
       PlayEntity playEntity = PlayEntity.builder().id(timePlayed)
           .trackId(trackId)
           .build();