|
|
@@ -0,0 +1,139 @@
|
|
|
+/**
|
|
|
+ * 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.zookeeper.server.persistence;
|
|
|
+
|
|
|
+import org.apache.jute.Record;
|
|
|
+import org.apache.zookeeper.server.DataTree;
|
|
|
+import org.apache.zookeeper.test.ClientBase;
|
|
|
+import org.apache.zookeeper.txn.TxnHeader;
|
|
|
+import org.junit.Assert;
|
|
|
+import org.junit.Test;
|
|
|
+
|
|
|
+import java.io.File;
|
|
|
+import java.io.IOException;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.concurrent.ConcurrentHashMap;
|
|
|
+
|
|
|
+public class FileTxnSnapLogTest {
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Test verifies the auto creation of data dir and data log dir.
|
|
|
+ * Sets "zookeeper.datadir.autocreate" to true.
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testWithAutoCreateDataLogDir() throws IOException {
|
|
|
+ File tmpDir = ClientBase.createEmptyTestDir();
|
|
|
+ File dataDir = new File(tmpDir, "data");
|
|
|
+ File snapDir = new File(tmpDir, "data_txnlog");
|
|
|
+ Assert.assertFalse("data directory already exists", dataDir.exists());
|
|
|
+ Assert.assertFalse("snapshot directory already exists", snapDir.exists());
|
|
|
+
|
|
|
+ String priorAutocreateDirValue = System.getProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE);
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE, "true");
|
|
|
+ FileTxnSnapLog fileTxnSnapLog;
|
|
|
+ try {
|
|
|
+ fileTxnSnapLog = new FileTxnSnapLog(dataDir, snapDir);
|
|
|
+ } finally {
|
|
|
+ if (priorAutocreateDirValue == null) {
|
|
|
+ System.clearProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE);
|
|
|
+ } else {
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE, priorAutocreateDirValue);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ Assert.assertTrue(dataDir.exists());
|
|
|
+ Assert.assertTrue(snapDir.exists());
|
|
|
+ Assert.assertTrue(fileTxnSnapLog.getDataDir().exists());
|
|
|
+ Assert.assertTrue(fileTxnSnapLog.getSnapDir().exists());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Test verifies server should fail when data dir or data log dir doesn't
|
|
|
+ * exists. Sets "zookeeper.datadir.autocreate" to false.
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testWithoutAutoCreateDataLogDir() throws Exception {
|
|
|
+ File tmpDir = ClientBase.createEmptyTestDir();
|
|
|
+ File dataDir = new File(tmpDir, "data");
|
|
|
+ File snapDir = new File(tmpDir, "data_txnlog");
|
|
|
+ Assert.assertFalse("data directory already exists", dataDir.exists());
|
|
|
+ Assert.assertFalse("snapshot directory already exists", snapDir.exists());
|
|
|
+
|
|
|
+ String priorAutocreateDirValue = System.getProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE);
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE, "false");
|
|
|
+ try {
|
|
|
+ FileTxnSnapLog fileTxnSnapLog = new FileTxnSnapLog(dataDir, snapDir);
|
|
|
+ } catch (FileTxnSnapLog.DatadirException e) {
|
|
|
+ Assert.assertFalse(dataDir.exists());
|
|
|
+ Assert.assertFalse(snapDir.exists());
|
|
|
+ return;
|
|
|
+ } finally {
|
|
|
+ if (priorAutocreateDirValue == null) {
|
|
|
+ System.clearProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE);
|
|
|
+ } else {
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DATADIR_AUTOCREATE, priorAutocreateDirValue);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ Assert.fail("Expected exception from FileTxnSnapLog");
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testAutoCreateDb() throws IOException {
|
|
|
+ File tmpDir = ClientBase.createEmptyTestDir();
|
|
|
+ File dataDir = new File(tmpDir, "data");
|
|
|
+ File snapDir = new File(tmpDir, "data_txnlog");
|
|
|
+ Assert.assertTrue("cannot create data directory", dataDir.mkdir());
|
|
|
+ Assert.assertTrue("cannot create snapshot directory", snapDir.mkdir());
|
|
|
+ File initFile = new File(dataDir, "initialize");
|
|
|
+ Assert.assertFalse("initialize file already exists", initFile.exists());
|
|
|
+
|
|
|
+ String priorAutocreateDbValue = System.getProperty(FileTxnSnapLog.ZOOKEEPER_DB_AUTOCREATE);
|
|
|
+ Map<Long, Integer> sessions = new ConcurrentHashMap<>();
|
|
|
+
|
|
|
+ attemptAutoCreateDb(dataDir, snapDir, sessions, priorAutocreateDbValue, "false", -1L);
|
|
|
+
|
|
|
+ attemptAutoCreateDb(dataDir, snapDir, sessions, priorAutocreateDbValue, "true", 0L);
|
|
|
+
|
|
|
+ Assert.assertTrue("cannot create initialize file", initFile.createNewFile());
|
|
|
+ attemptAutoCreateDb(dataDir, snapDir, sessions, priorAutocreateDbValue, "false", 0L);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void attemptAutoCreateDb(File dataDir, File snapDir, Map<Long, Integer> sessions,
|
|
|
+ String priorAutocreateDbValue, String autoCreateValue,
|
|
|
+ long expectedValue) throws IOException {
|
|
|
+ sessions.clear();
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DB_AUTOCREATE, autoCreateValue);
|
|
|
+ FileTxnSnapLog fileTxnSnapLog = new FileTxnSnapLog(dataDir, snapDir);
|
|
|
+
|
|
|
+ try {
|
|
|
+ long zxid = fileTxnSnapLog.restore(new DataTree(), sessions, new FileTxnSnapLog.PlayBackListener() {
|
|
|
+ @Override
|
|
|
+ public void onTxnLoaded(TxnHeader hdr, Record rec) {
|
|
|
+ // empty by default
|
|
|
+ }
|
|
|
+ });
|
|
|
+ Assert.assertEquals("unexpected zxid", expectedValue, zxid);
|
|
|
+ } finally {
|
|
|
+ if (priorAutocreateDbValue == null) {
|
|
|
+ System.clearProperty(FileTxnSnapLog.ZOOKEEPER_DB_AUTOCREATE);
|
|
|
+ } else {
|
|
|
+ System.setProperty(FileTxnSnapLog.ZOOKEEPER_DB_AUTOCREATE, priorAutocreateDbValue);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
Brian Nixon