io.github.repir.Repository.Pig.PigRepo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of RepIR Show documentation
The newest version!
package io.github.repir.Repository.Pig;

import io.github.repir.Repository.EntityStoredFeature;
import static io.github.repir.Repository.Feature.canonicalName;
import io.github.repir.Repository.Pig.PigRepo.File;
import io.github.repir.Repository.Pig.PigRepo.Tuple;
import io.github.repir.Repository.Repository;
import io.github.repir.Repository.StoredFeature;
import io.github.repir.tools.io.Datafile;
import io.github.repir.tools.io.struct.StructuredTextPig;
import io.github.repir.tools.io.struct.StructuredTextPigTuple;
import io.github.repir.tools.lib.Log;

/**
 * Can store one literal String per Document, e.g. collection ID, title, url.
 * @see EntityStoredFeature
 * @author jer
 */
public class PigRepo extends PigFeature  {

   public static Log log = new Log(PigRepo.class);

   private PigRepo(Repository repository) {
      super(repository);
   }

   public static PigRepo get(Repository repository) {
       String label = canonicalName(PigRepo.class);
       PigRepo pigrepo = (PigRepo) StoredFeature.getStoredFeature(repository, label);
       if (pigrepo == null) {
          pigrepo = new PigRepo(repository);
          StoredFeature.storeFeature(repository, label, pigrepo);
       }
       return pigrepo;
   }
   
   @Override
   public File createFile(Datafile datafile) {
      return new File(datafile);
   }

   @Override
   public Tuple getValue() {
      if (getFile().nextRecord()) {
         return new Tuple(file);
      } else {
         return null;  
      }
   }

   @Override
   public void write(Tuple value) {
      file.write(value);
   }
   
   public static class File extends StructuredTextPig {
      public StringField key = this.addString("key");
      public StringField value = this.addString("value");

      public File(Datafile df) {
         super(df);
      }
   }
   
   public static class Tuple extends StructuredTextPigTuple {
      public String key;
      public String value;
      
      public Tuple() {}
      
      protected Tuple(File file) {
         key = file.key.get();
         value = file.value.get();
      }
      
      @Override
      public void write(File file) {
         file.key.set(key);
         file.value.set(value);
         file.write();
      }
   }
}