Examples with Version org.apache.hadoop.yarn.server.records.Version used on opensource projects

Example 11 with Version

use of org.apache.hadoop.yarn.server.records.Version in project hadoop by apache.

the class TestFSRMStateStore method testCheckMajorVersionChange.

@Test(timeout = 60000)
public void testCheckMajorVersionChange() throws Exception {
    HdfsConfiguration conf = new HdfsConfiguration();
    MiniDFSCluster cluster = new MiniDFSCluster.Builder(conf).numDataNodes(1).build();
    try {
        fsTester = new TestFSRMStateStoreTester(cluster, false) {

            Version VERSION_INFO = Version.newInstance(Integer.MAX_VALUE, 0);

            @Override
            public Version getCurrentVersion() throws Exception {
                return VERSION_INFO;
            }

            @Override
            public RMStateStore getRMStateStore() throws Exception {
                YarnConfiguration conf = new YarnConfiguration();
                conf.set(YarnConfiguration.FS_RM_STATE_STORE_URI, workingDirPathURI.toString());
                this.store = new TestFileSystemRMStore(conf) {

                    Version storedVersion = null;

                    @Override
                    public Version getCurrentVersion() {
                        return VERSION_INFO;
                    }

                    @Override
                    protected synchronized Version loadVersion() throws Exception {
                        return storedVersion;
                    }

                    @Override
                    protected synchronized void storeVersion() throws Exception {
                        storedVersion = VERSION_INFO;
                    }
                };
                return store;
            }
        };
        // default version
        RMStateStore store = fsTester.getRMStateStore();
        Version defaultVersion = fsTester.getCurrentVersion();
        store.checkVersion();
        Assert.assertEquals(defaultVersion, store.loadVersion());
    } finally {
        cluster.shutdown();
    }
}

Example 12 with Version

use of org.apache.hadoop.yarn.server.records.Version in project hadoop by apache.

the class LeveldbTimelineStateStore method loadVersion.

@VisibleForTesting
Version loadVersion() throws IOException {
    try {
        byte[] data = db.get(TIMELINE_STATE_STORE_VERSION_KEY);
        // if version is not stored previously, treat it as CURRENT_VERSION_INFO.
        if (data == null || data.length == 0) {
            return getCurrentVersion();
        }
        Version version = new VersionPBImpl(YarnServerCommonProtos.VersionProto.parseFrom(data));
        return version;
    } catch (DBException e) {
        throw new IOException(e);
    }
}

Example 13 with Version

use of org.apache.hadoop.yarn.server.records.Version in project hadoop by apache.

the class TestLeveldbTimelineStateStore method testCheckVersion.

@Test
public void testCheckVersion() throws IOException {
    LeveldbTimelineStateStore store = initAndStartTimelineServiceStateStoreService();
    // default version
    Version defaultVersion = store.getCurrentVersion();
    Assert.assertEquals(defaultVersion, store.loadVersion());
    // compatible version
    Version compatibleVersion = Version.newInstance(defaultVersion.getMajorVersion(), defaultVersion.getMinorVersion() + 2);
    store.storeVersion(compatibleVersion);
    Assert.assertEquals(compatibleVersion, store.loadVersion());
    store.stop();
    // overwrite the compatible version
    store = initAndStartTimelineServiceStateStoreService();
    Assert.assertEquals(defaultVersion, store.loadVersion());
    // incompatible version
    Version incompatibleVersion = Version.newInstance(defaultVersion.getMajorVersion() + 1, defaultVersion.getMinorVersion());
    store.storeVersion(incompatibleVersion);
    store.stop();
    try {
        initAndStartTimelineServiceStateStoreService();
        Assert.fail("Incompatible version, should expect fail here.");
    } catch (ServiceStateException e) {
        Assert.assertTrue("Exception message mismatch", e.getMessage().contains("Incompatible version for timeline state store"));
    }
}

Example 14 with Version

use of org.apache.hadoop.yarn.server.records.Version in project hadoop by apache.

the class TestLeveldbTimelineStore method testCheckVersion.

@Test
public void testCheckVersion() throws IOException {
    LeveldbTimelineStore dbStore = (LeveldbTimelineStore) store;
    // default version
    Version defaultVersion = dbStore.getCurrentVersion();
    Assert.assertEquals(defaultVersion, dbStore.loadVersion());
    // compatible version
    Version compatibleVersion = Version.newInstance(defaultVersion.getMajorVersion(), defaultVersion.getMinorVersion() + 2);
    dbStore.storeVersion(compatibleVersion);
    Assert.assertEquals(compatibleVersion, dbStore.loadVersion());
    restartTimelineStore();
    dbStore = (LeveldbTimelineStore) store;
    // overwrite the compatible version
    Assert.assertEquals(defaultVersion, dbStore.loadVersion());
    // incompatible version
    Version incompatibleVersion = Version.newInstance(defaultVersion.getMajorVersion() + 1, defaultVersion.getMinorVersion());
    dbStore.storeVersion(incompatibleVersion);
    try {
        restartTimelineStore();
        Assert.fail("Incompatible version, should expect fail here.");
    } catch (ServiceStateException e) {
        Assert.assertTrue("Exception message mismatch", e.getMessage().contains("Incompatible version for timeline store"));
    }
}

Example 15 with Version

use of org.apache.hadoop.yarn.server.records.Version in project hadoop by apache.

the class TestRollingLevelDBTimelineStore method testCheckVersion.

@Test
public void testCheckVersion() throws IOException {
    RollingLevelDBTimelineStore dbStore = (RollingLevelDBTimelineStore) store;
    // default version
    Version defaultVersion = dbStore.getCurrentVersion();
    Assert.assertEquals(defaultVersion, dbStore.loadVersion());
    // compatible version
    Version compatibleVersion = Version.newInstance(defaultVersion.getMajorVersion(), defaultVersion.getMinorVersion() + 2);
    dbStore.storeVersion(compatibleVersion);
    Assert.assertEquals(compatibleVersion, dbStore.loadVersion());
    restartTimelineStore();
    dbStore = (RollingLevelDBTimelineStore) store;
    // overwrite the compatible version
    Assert.assertEquals(defaultVersion, dbStore.loadVersion());
    // incompatible version
    Version incompatibleVersion = Version.newInstance(defaultVersion.getMajorVersion() + 1, defaultVersion.getMinorVersion());
    dbStore.storeVersion(incompatibleVersion);
    try {
        restartTimelineStore();
        Assert.fail("Incompatible version, should expect fail here.");
    } catch (ServiceStateException e) {
        Assert.assertTrue("Exception message mismatch", e.getMessage().contains("Incompatible version for timeline store"));
    }
}