diff --git a/scraper/tweet_trove.go b/scraper/tweet_trove.go index dada79d..3d2f390 100644 --- a/scraper/tweet_trove.go +++ b/scraper/tweet_trove.go @@ -27,5 +27,24 @@ func (trove TweetTrove) Transform() (tweets []Tweet, retweets []Retweet, users [ for _, val := range trove.Retweets { retweets = append(retweets, val) } + return } // TODO: refactor until this function isn't needed anymore + + +/** + * Combine two troves into one + */ +func (t1 *TweetTrove) MergeWith(t2 TweetTrove) { + for id, val := range t2.Tweets { + t1.Tweets[id] = val + } + for id, val := range t2.Users { + t1.Users[id] = val + } + for id, val := range t2.Retweets { + t1.Retweets[id] = val + } + + t1.TombstoneUsers = append(t1.TombstoneUsers, t2.TombstoneUsers...) +}