MalformedFileService.java

package com.hubspot.blazar.data.service;

import com.hubspot.blazar.base.GitInfo;
import com.hubspot.blazar.base.MalformedFile;
import com.hubspot.blazar.data.dao.MalformedFileDao;

import javax.inject.Inject;
import javax.inject.Singleton;
import javax.transaction.Transactional;
import java.util.Set;

@Singleton
public class MalformedFileService {
  private final MalformedFileDao malformedFileDao;

  @Inject
  public MalformedFileService(MalformedFileDao malformedFileDao) {
    this.malformedFileDao = malformedFileDao;
  }

  public Set<MalformedFile> getMalformedFiles(int branchId) {
    return malformedFileDao.getMalformedFiles(branchId);
  }

  @Transactional
  public void setMalformedFiles(GitInfo gitInfo, Set<MalformedFile> malformedFiles) {
    malformedFileDao.clearMalformedFiles(gitInfo.getId().get());
    if (!malformedFiles.isEmpty()) {
      malformedFileDao.insertMalformedFile(malformedFiles);
    }
  }
}