/**
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   *     http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.apache.hadoop.hbase.procedure2.store.wal;
  
  import java.io.FileNotFoundException;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.OutputStream;
  import java.util.Arrays;
  import java.util.Comparator;
  import java.util.HashSet;
  import java.util.Set;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.apache.hadoop.fs.FileSystem;
  import org.apache.hadoop.fs.FileStatus;
  import org.apache.hadoop.fs.Path;
  import org.apache.hadoop.hbase.HBaseCommonTestingUtility;
  import org.apache.hadoop.hbase.procedure2.Procedure;
  import org.apache.hadoop.hbase.procedure2.ProcedureTestingUtility;
  import org.apache.hadoop.hbase.procedure2.ProcedureTestingUtility.LoadCounter;
  import org.apache.hadoop.hbase.procedure2.ProcedureTestingUtility.TestProcedure;
  import org.apache.hadoop.hbase.procedure2.SequentialProcedure;
  import org.apache.hadoop.hbase.procedure2.store.ProcedureStore;
  import org.apache.hadoop.hbase.procedure2.store.ProcedureStore.ProcedureIterator;
  import org.apache.hadoop.hbase.testclassification.SmallTests;
  import org.apache.hadoop.hbase.util.Bytes;
  import org.apache.hadoop.io.IOUtils;
  
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  import org.junit.experimental.categories.Category;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertFalse;
  import static org.junit.Assert.assertTrue;
  import static org.junit.Assert.fail;
  
  @Category(SmallTests.class)
  public class TestWALProcedureStore {
    private static final Log LOG = LogFactory.getLog(TestWALProcedureStore.class);
  
    private static final int PROCEDURE_STORE_SLOTS = 1;
  
    private WALProcedureStore procStore;
  
    private HBaseCommonTestingUtility htu;
    private FileSystem fs;
    private Path testDir;
    private Path logDir;
  
    @Before
    public void setUp() throws IOException {
      htu = new HBaseCommonTestingUtility();
      testDir = htu.getDataTestDir();
      fs = testDir.getFileSystem(htu.getConfiguration());
      assertTrue(testDir.depth() > 1);
  
      logDir = new Path(testDir, "proc-logs");
      procStore = ProcedureTestingUtility.createWalStore(htu.getConfiguration(), logDir);
      procStore.start(PROCEDURE_STORE_SLOTS);
      procStore.recoverLease();
      procStore.load(new LoadCounter());
    }
  
    @After
    public void tearDown() throws IOException {
      procStore.stop(false);
      fs.delete(logDir, true);
    }
  
    private void storeRestart(ProcedureStore.ProcedureLoader loader) throws Exception {
      ProcedureTestingUtility.storeRestart(procStore, loader);
    }
  
    @Test
    public void testEmptyRoll() throws Exception {
      for (int i = 0; i < 10; ++i) {
        procStore.periodicRollForTesting();
      }
      FileStatus[] status = fs.listStatus(logDir);
     assertEquals(1, status.length);
   }
 
   @Test
   public void testEmptyLogLoad() throws Exception {
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(0, loader.getMaxProcId());
     assertEquals(0, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
   }
 
   @Test
   public void testLoad() throws Exception {
     Set<Long> procIds = new HashSet<>();
 
     // Insert something in the log
     Procedure proc1 = new TestSequentialProcedure();
     procIds.add(proc1.getProcId());
     procStore.insert(proc1, null);
 
     Procedure proc2 = new TestSequentialProcedure();
     Procedure[] child2 = new Procedure[2];
     child2[0] = new TestSequentialProcedure();
     child2[1] = new TestSequentialProcedure();
 
     procIds.add(proc2.getProcId());
     procIds.add(child2[0].getProcId());
     procIds.add(child2[1].getProcId());
     procStore.insert(proc2, child2);
 
     // Verify that everything is there
     verifyProcIdsOnRestart(procIds);
 
     // Update and delete something
     procStore.update(proc1);
     procStore.update(child2[1]);
     procStore.delete(child2[1].getProcId());
     procIds.remove(child2[1].getProcId());
 
     // Verify that everything is there
     verifyProcIdsOnRestart(procIds);
 
     // Remove 4 byte from the trailers
     procStore.stop(false);
     FileStatus[] logs = fs.listStatus(logDir);
     assertEquals(3, logs.length);
     for (int i = 0; i < logs.length; ++i) {
       corruptLog(logs[i], 4);
     }
     verifyProcIdsOnRestart(procIds);
   }
 
   @Test
   public void testNoTrailerDoubleRestart() throws Exception {
     // log-0001: proc 0, 1 and 2 are inserted
     Procedure proc0 = new TestSequentialProcedure();
     procStore.insert(proc0, null);
     Procedure proc1 = new TestSequentialProcedure();
     procStore.insert(proc1, null);
     Procedure proc2 = new TestSequentialProcedure();
     procStore.insert(proc2, null);
     procStore.rollWriterForTesting();
 
     // log-0002: proc 1 deleted
     procStore.delete(proc1.getProcId());
     procStore.rollWriterForTesting();
 
     // log-0003: proc 2 is update
     procStore.update(proc2);
     procStore.rollWriterForTesting();
 
     // log-0004: proc 2 deleted
     procStore.delete(proc2.getProcId());
 
     // stop the store and remove the trailer
     procStore.stop(false);
     FileStatus[] logs = fs.listStatus(logDir);
     assertEquals(4, logs.length);
     for (int i = 0; i < logs.length; ++i) {
       corruptLog(logs[i], 4);
     }
 
     // Test Load 1
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(1, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
 
     // Test Load 2
     assertEquals(5, fs.listStatus(logDir).length);
     loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(1, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
 
     // remove proc-0
     procStore.delete(proc0.getProcId());
     procStore.periodicRollForTesting();
     assertEquals(1, fs.listStatus(logDir).length);
     storeRestart(loader);
   }
 
   @Test
   public void testCorruptedTrailer() throws Exception {
     // Insert something
     for (int i = 0; i < 100; ++i) {
       procStore.insert(new TestSequentialProcedure(), null);
     }
 
     // Stop the store
     procStore.stop(false);
 
     // Remove 4 byte from the trailer
     FileStatus[] logs = fs.listStatus(logDir);
     assertEquals(1, logs.length);
     corruptLog(logs[0], 4);
 
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(100, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
   }
 
   @Test
   public void testCorruptedEntries() throws Exception {
     // Insert something
     for (int i = 0; i < 100; ++i) {
       procStore.insert(new TestSequentialProcedure(), null);
     }
 
     // Stop the store
     procStore.stop(false);
 
     // Remove some byte from the log
     // (enough to cut the trailer and corrupt some entries)
     FileStatus[] logs = fs.listStatus(logDir);
     assertEquals(1, logs.length);
     corruptLog(logs[0], 1823);
 
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertTrue(procStore.getCorruptedLogs() != null);
     assertEquals(1, procStore.getCorruptedLogs().size());
     assertEquals(85, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
   }
 
   @Test
   public void testCorruptedProcedures() throws Exception {
     // Insert root-procedures
     TestProcedure[] rootProcs = new TestProcedure[10];
     for (int i = 1; i <= rootProcs.length; i++) {
       rootProcs[i-1] = new TestProcedure(i, 0);
       procStore.insert(rootProcs[i-1], null);
       rootProcs[i-1].addStackId(0);
       procStore.update(rootProcs[i-1]);
     }
     // insert root-child txn
     procStore.rollWriterForTesting();
     for (int i = 1; i <= rootProcs.length; i++) {
       TestProcedure b = new TestProcedure(rootProcs.length + i, i);
       rootProcs[i-1].addStackId(1);
       procStore.insert(rootProcs[i-1], new Procedure[] { b });
     }
     // insert child updates
     procStore.rollWriterForTesting();
     for (int i = 1; i <= rootProcs.length; i++) {
       procStore.update(new TestProcedure(rootProcs.length + i, i));
     }
 
     // Stop the store
     procStore.stop(false);
 
     // the first log was removed,
     // we have insert-txn and updates in the others so everything is fine
     FileStatus[] logs = fs.listStatus(logDir);
     assertEquals(Arrays.toString(logs), 2, logs.length);
     Arrays.sort(logs, new Comparator<FileStatus>() {
       @Override
       public int compare(FileStatus o1, FileStatus o2) {
         return o1.getPath().getName().compareTo(o2.getPath().getName());
       }
     });
 
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(rootProcs.length * 2, loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
 
     // Remove the second log, we have lost all the root/parent references
     fs.delete(logs[0].getPath(), false);
     loader.reset();
     storeRestart(loader);
     assertEquals(0, loader.getLoadedCount());
     assertEquals(rootProcs.length, loader.getCorruptedCount());
     for (Procedure proc: loader.getCorrupted()) {
       assertTrue(proc.toString(), proc.getParentProcId() <= rootProcs.length);
       assertTrue(proc.toString(),
                   proc.getProcId() > rootProcs.length &&
                   proc.getProcId() <= (rootProcs.length * 2));
     }
   }
 
   @Test(timeout=60000)
   public void testWalReplayOrder_AB_A() throws Exception {
     /*
      * | A B | -> | A |
      */
     TestProcedure a = new TestProcedure(1, 0);
     TestProcedure b = new TestProcedure(2, 1);
 
     procStore.insert(a, null);
     a.addStackId(0);
     procStore.update(a);
 
     procStore.insert(a, new Procedure[] { b });
     b.addStackId(1);
     procStore.update(b);
 
     procStore.rollWriterForTesting();
 
     a.addStackId(2);
     procStore.update(a);
 
     storeRestart(new ProcedureStore.ProcedureLoader() {
       @Override
       public void setMaxProcId(long maxProcId) {
         assertEquals(2, maxProcId);
       }
 
       @Override
       public void load(ProcedureIterator procIter) throws IOException {
         assertTrue(procIter.hasNext());
         assertEquals(1, procIter.nextAsProcedureInfo().getProcId());
         assertTrue(procIter.hasNext());
         assertEquals(2, procIter.nextAsProcedureInfo().getProcId());
         assertFalse(procIter.hasNext());
       }
 
       @Override
       public void handleCorrupted(ProcedureIterator procIter) throws IOException {
         assertFalse(procIter.hasNext());
       }
     });
   }
 
   @Test(timeout=60000)
   public void testWalReplayOrder_ABC_BAD() throws Exception {
     /*
      * | A B C | -> | B A D |
      */
     TestProcedure a = new TestProcedure(1, 0);
     TestProcedure b = new TestProcedure(2, 1);
     TestProcedure c = new TestProcedure(3, 2);
     TestProcedure d = new TestProcedure(4, 0);
 
     procStore.insert(a, null);
     a.addStackId(0);
     procStore.update(a);
 
     procStore.insert(a, new Procedure[] { b });
     b.addStackId(1);
     procStore.update(b);
 
     procStore.insert(b, new Procedure[] { c });
     b.addStackId(2);
     procStore.update(b);
 
     procStore.rollWriterForTesting();
 
     b.addStackId(3);
     procStore.update(b);
 
     a.addStackId(4);
     procStore.update(a);
 
     procStore.insert(d, null);
     d.addStackId(0);
     procStore.update(d);
 
     storeRestart(new ProcedureStore.ProcedureLoader() {
       @Override
       public void setMaxProcId(long maxProcId) {
         assertEquals(4, maxProcId);
       }
 
       @Override
       public void load(ProcedureIterator procIter) throws IOException {
         assertTrue(procIter.hasNext());
         assertEquals(4, procIter.nextAsProcedureInfo().getProcId());
         // TODO: This will be multiple call once we do fast-start
         //assertFalse(procIter.hasNext());
 
         assertTrue(procIter.hasNext());
         assertEquals(1, procIter.nextAsProcedureInfo().getProcId());
         assertTrue(procIter.hasNext());
         assertEquals(2, procIter.nextAsProcedureInfo().getProcId());
         assertTrue(procIter.hasNext());
         assertEquals(3, procIter.nextAsProcedureInfo().getProcId());
         assertFalse(procIter.hasNext());
       }
 
       @Override
       public void handleCorrupted(ProcedureIterator procIter) throws IOException {
         assertFalse(procIter.hasNext());
       }
     });
   }
 
   @Test
   public void testRollAndRemove() throws IOException {
     // Insert something in the log
     Procedure proc1 = new TestSequentialProcedure();
     procStore.insert(proc1, null);
 
     Procedure proc2 = new TestSequentialProcedure();
     procStore.insert(proc2, null);
 
     // roll the log, now we have 2
     procStore.rollWriterForTesting();
     assertEquals(2, procStore.getActiveLogs().size());
 
     // everything will be up to date in the second log
     // so we can remove the first one
     procStore.update(proc1);
     procStore.update(proc2);
     assertEquals(1, procStore.getActiveLogs().size());
 
     // roll the log, now we have 2
     procStore.rollWriterForTesting();
     assertEquals(2, procStore.getActiveLogs().size());
 
     // remove everything active
     // so we can remove all the logs
     procStore.delete(proc1.getProcId());
     procStore.delete(proc2.getProcId());
     assertEquals(1, procStore.getActiveLogs().size());
   }
 
   @Test
   public void testFileNotFoundDuringLeaseRecovery() throws IOException {
     TestProcedure[] procs = new TestProcedure[3];
     for (int i = 0; i < procs.length; ++i) {
       procs[i] = new TestProcedure(i + 1, 0);
       procStore.insert(procs[i], null);
     }
     procStore.rollWriterForTesting();
     for (int i = 0; i < procs.length; ++i) {
       procStore.update(procs[i]);
       procStore.rollWriterForTesting();
     }
     procStore.stop(false);
 
     FileStatus[] status = fs.listStatus(logDir);
     assertEquals(procs.length + 2, status.length);
 
     // simulate another active master removing the wals
     procStore = new WALProcedureStore(htu.getConfiguration(), logDir,
         new WALProcedureStore.LeaseRecovery() {
       private int count = 0;
 
       @Override
       public void recoverFileLease(FileSystem fs, Path path) throws IOException {
         if (++count <= 2) {
           fs.delete(path, false);
           LOG.debug("Simulate FileNotFound at count=" + count + " for " + path);
           throw new FileNotFoundException("test file not found " + path);
         }
         LOG.debug("Simulate recoverFileLease() at count=" + count + " for " + path);
       }
     });
 
     final LoadCounter loader = new LoadCounter();
     procStore.start(PROCEDURE_STORE_SLOTS);
     procStore.recoverLease();
     procStore.load(loader);
     assertEquals(procs.length, loader.getMaxProcId());
     assertEquals(procs.length - 1, loader.getRunnableCount());
     assertEquals(0, loader.getCompletedCount());
     assertEquals(0, loader.getCorruptedCount());
   }
 
   @Test
   public void testLoadChildren() throws Exception {
     TestProcedure a = new TestProcedure(1, 0);
     TestProcedure b = new TestProcedure(2, 1);
     TestProcedure c = new TestProcedure(3, 1);
 
     // INIT
     procStore.insert(a, null);
 
     // Run A first step
     a.addStackId(0);
     procStore.update(a);
 
     // Run A second step
     a.addStackId(1);
     procStore.insert(a, new Procedure[] { b, c });
 
     // Run B first step
     b.addStackId(2);
     procStore.update(b);
 
     // Run C first and last step
     c.addStackId(3);
     procStore.update(c);
 
     // Run B second setp
     b.addStackId(4);
     procStore.update(b);
 
     // back to A
     a.addStackId(5);
     a.setFinishedState();
     procStore.delete(a, new long[] { b.getProcId(), c.getProcId() });
     restartAndAssert(3, 0, 1, 0);
   }
 
   private void restartAndAssert(long maxProcId, long runnableCount,
       int completedCount, int corruptedCount) throws Exception {
     ProcedureTestingUtility.storeRestartAndAssert(procStore, maxProcId,
       runnableCount, completedCount, corruptedCount);
   }
 
   private void corruptLog(final FileStatus logFile, final long dropBytes)
       throws IOException {
     assertTrue(logFile.getLen() > dropBytes);
     LOG.debug("corrupt log " + logFile.getPath() +
               " size=" + logFile.getLen() + " drop=" + dropBytes);
     Path tmpPath = new Path(testDir, "corrupted.log");
     InputStream in = fs.open(logFile.getPath());
     OutputStream out =  fs.create(tmpPath);
     IOUtils.copyBytes(in, out, logFile.getLen() - dropBytes, true);
     if (!fs.rename(tmpPath, logFile.getPath())) {
       throw new IOException("Unable to rename");
     }
   }
 
   private void verifyProcIdsOnRestart(final Set<Long> procIds) throws Exception {
     LOG.debug("expected: " + procIds);
     LoadCounter loader = new LoadCounter();
     storeRestart(loader);
     assertEquals(procIds.size(), loader.getLoadedCount());
     assertEquals(0, loader.getCorruptedCount());
   }
 
   private void assertEmptyLogDir() {
     try {
       FileStatus[] status = fs.listStatus(logDir);
       assertTrue("expected empty state-log dir", status == null || status.length == 0);
     } catch (FileNotFoundException e) {
       fail("expected the state-log dir to be present: " + logDir);
     } catch (IOException e) {
       fail("got en exception on state-log dir list: " + e.getMessage());
     }
   }
 
   public static class TestSequentialProcedure extends SequentialProcedure<Void> {
     private static long seqid = 0;
 
     public TestSequentialProcedure() {
       setProcId(++seqid);
     }
 
     @Override
     protected Procedure[] execute(Void env) { return null; }
 
     @Override
     protected void rollback(Void env) { }
 
     @Override
     protected boolean abort(Void env) { return false; }
 
     @Override
     protected void serializeStateData(final OutputStream stream) throws IOException {
       long procId = getProcId();
       if (procId % 2 == 0) {
         stream.write(Bytes.toBytes(procId));
       }
     }
 
     @Override
     protected void deserializeStateData(InputStream stream) throws IOException {
       long procId = getProcId();
       if (procId % 2 == 0) {
         byte[] bProcId = new byte[8];
         assertEquals(8, stream.read(bProcId));
         assertEquals(procId, Bytes.toLong(bProcId));
       } else {
         assertEquals(0, stream.available());
       }
     }
   }
 }