Prevent some rare cases of NPE and Deadlocks, better error handling on yaml load

This commit is contained in:
snowleo 2011-12-06 14:39:52 +01:00
parent 019b49ef11
commit 51390a9698
8 changed files with 145 additions and 105 deletions

View file

@ -8,6 +8,7 @@ import java.io.IOException;
import java.util.logging.Level; import java.util.logging.Level;
import org.bukkit.Bukkit; import org.bukkit.Bukkit;
import org.bukkit.plugin.Plugin; import org.bukkit.plugin.Plugin;
import org.yaml.snakeyaml.error.YAMLException;
public abstract class AbstractDelayedYamlFileReader<T extends StorageObject> implements Runnable public abstract class AbstractDelayedYamlFileReader<T extends StorageObject> implements Runnable
@ -33,9 +34,17 @@ public abstract class AbstractDelayedYamlFileReader<T extends StorageObject> imp
try try
{ {
onStart(); onStart();
try
{
reader = new FileReader(file); reader = new FileReader(file);
final T object = new YamlStorageReader(reader, plugin).load(clazz); T object = new YamlStorageReader(reader, plugin).load(clazz);
onFinish(object); onSuccess(object);
}
catch (ObjectLoadException ex)
{
onException();
Bukkit.getLogger().log(Level.SEVERE, "File broken: " + file.toString(), ex.getCause());
}
} }
catch (FileNotFoundException ex) catch (FileNotFoundException ex)
{ {
@ -57,5 +66,7 @@ public abstract class AbstractDelayedYamlFileReader<T extends StorageObject> imp
} }
} }
public abstract void onFinish(T object); public abstract void onSuccess(T object);
public abstract void onException();
} }

View file

@ -3,5 +3,5 @@ package com.earth2me.essentials.storage;
public interface IStorageReader public interface IStorageReader
{ {
<T extends StorageObject> T load(final Class<? extends T> clazz); <T extends StorageObject> T load(final Class<? extends T> clazz) throws ObjectLoadException;
} }

View file

@ -0,0 +1,10 @@
package com.earth2me.essentials.storage;
public class ObjectLoadException extends Exception
{
public ObjectLoadException(Throwable thrwbl)
{
super(thrwbl);
}
}

View file

@ -4,8 +4,6 @@ import java.io.Reader;
import java.lang.reflect.Field; import java.lang.reflect.Field;
import java.util.*; import java.util.*;
import java.util.concurrent.locks.ReentrantLock; import java.util.concurrent.locks.ReentrantLock;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.bukkit.plugin.Plugin; import org.bukkit.plugin.Plugin;
import org.yaml.snakeyaml.TypeDescription; import org.yaml.snakeyaml.TypeDescription;
import org.yaml.snakeyaml.Yaml; import org.yaml.snakeyaml.Yaml;
@ -14,8 +12,8 @@ import org.yaml.snakeyaml.constructor.Constructor;
public class YamlStorageReader implements IStorageReader public class YamlStorageReader implements IStorageReader
{ {
private transient static Map<Class, Yaml> preparedYamls = Collections.synchronizedMap(new HashMap<Class, Yaml>()); private transient static final Map<Class, Yaml> PREPARED_YAMLS = Collections.synchronizedMap(new HashMap<Class, Yaml>());
private transient static Map<Class, ReentrantLock> locks = new HashMap<Class, ReentrantLock>(); private transient static final Map<Class, ReentrantLock> LOCKS = new HashMap<Class, ReentrantLock>();
private transient final Reader reader; private transient final Reader reader;
private transient final Plugin plugin; private transient final Plugin plugin;
@ -26,49 +24,40 @@ public class YamlStorageReader implements IStorageReader
} }
@Override @Override
public <T extends StorageObject> T load(final Class<? extends T> clazz) public <T extends StorageObject> T load(final Class<? extends T> clazz) throws ObjectLoadException
{ {
Yaml yaml = preparedYamls.get(clazz); Yaml yaml = PREPARED_YAMLS.get(clazz);
if (yaml == null) if (yaml == null)
{ {
yaml = new Yaml(prepareConstructor(clazz)); yaml = new Yaml(prepareConstructor(clazz));
preparedYamls.put(clazz, yaml); PREPARED_YAMLS.put(clazz, yaml);
} }
ReentrantLock lock; ReentrantLock lock;
synchronized (locks) synchronized (LOCKS)
{ {
lock = locks.get(clazz); lock = LOCKS.get(clazz);
if (lock == null) if (lock == null)
{ {
lock = new ReentrantLock(); lock = new ReentrantLock();
} }
} }
T ret;
lock.lock(); lock.lock();
try try
{ {
ret = (T)yaml.load(reader); T object = (T)yaml.load(reader);
if (object == null) {
object = clazz.newInstance();
}
return object;
}
catch (Exception ex)
{
throw new ObjectLoadException(ex);
} }
finally finally
{ {
lock.unlock(); lock.unlock();
} }
if (ret == null)
{
try
{
ret = (T)clazz.newInstance();
}
catch (InstantiationException ex)
{
Logger.getLogger(StorageObject.class.getName()).log(Level.SEVERE, null, ex);
}
catch (IllegalAccessException ex)
{
Logger.getLogger(StorageObject.class.getName()).log(Level.SEVERE, null, ex);
}
}
return ret;
} }
private Constructor prepareConstructor(final Class<?> clazz) private Constructor prepareConstructor(final Class<?> clazz)

View file

@ -81,7 +81,6 @@ public class User extends UserBase implements IOfflineUser
new UserDataWriter(); new UserDataWriter();
} }
private class UserDataWriter extends AbstractDelayedYamlFileWriter private class UserDataWriter extends AbstractDelayedYamlFileWriter
{ {
public UserDataWriter() public UserDataWriter()

View file

@ -1,6 +1,7 @@
package com.earth2me.essentials; package com.earth2me.essentials;
import com.earth2me.essentials.settings.Settings; import com.earth2me.essentials.settings.Settings;
import com.earth2me.essentials.storage.ObjectLoadException;
import com.earth2me.essentials.storage.StorageObject; import com.earth2me.essentials.storage.StorageObject;
import com.earth2me.essentials.storage.YamlStorageReader; import com.earth2me.essentials.storage.YamlStorageReader;
import com.earth2me.essentials.storage.YamlStorageWriter; import com.earth2me.essentials.storage.YamlStorageWriter;
@ -11,7 +12,6 @@ import org.bukkit.World;
import org.bukkit.World.Environment; import org.bukkit.World.Environment;
import org.bukkit.plugin.InvalidDescriptionException; import org.bukkit.plugin.InvalidDescriptionException;
import org.junit.Test; import org.junit.Test;
import org.yaml.snakeyaml.Yaml;
public class StorageTest extends TestCase public class StorageTest extends TestCase
@ -41,6 +41,8 @@ public class StorageTest extends TestCase
@Test @Test
public void testSettings() public void testSettings()
{
try
{ {
assertTrue(StorageObject.class.isAssignableFrom(Settings.class)); assertTrue(StorageObject.class.isAssignableFrom(Settings.class));
ExecuteTimer ext = new ExecuteTimer(); ExecuteTimer ext = new ExecuteTimer();
@ -70,9 +72,16 @@ public class StorageTest extends TestCase
//assertEquals("Default and rewritten config should be equal", settings, settings2); //assertEquals("Default and rewritten config should be equal", settings, settings2);
//that assertion fails, because empty list and maps return as null //that assertion fails, because empty list and maps return as null
} }
catch (ObjectLoadException ex)
{
fail(ex.getMessage());
}
}
@Test @Test
public void testUserdata() public void testUserdata()
{
try
{ {
FakeServer server = new FakeServer(); FakeServer server = new FakeServer();
World world = server.createWorld("testWorld", Environment.NORMAL); World world = server.createWorld("testWorld", Environment.NORMAL);
@ -116,8 +125,11 @@ public class StorageTest extends TestCase
System.out.println(userdata.toString()); System.out.println(userdata.toString());
System.out.println(userdata2.toString()); System.out.println(userdata2.toString());
System.out.println(ext.end()); System.out.println(ext.end());
com.earth2me.essentials.user.User test = new com.earth2me.essentials.user.User(null, ess); }
test.example(); catch (ObjectLoadException ex)
{
fail(ex.getMessage());
}
} }

View file

@ -35,6 +35,7 @@ public class EssentialsSpawn extends JavaPlugin
} }
spawns = new SpawnStorage(ess); spawns = new SpawnStorage(ess);
ess.addReloadListener(spawns);
final EssentialsSpawnPlayerListener playerListener = new EssentialsSpawnPlayerListener(ess, spawns); final EssentialsSpawnPlayerListener playerListener = new EssentialsSpawnPlayerListener(ess, spawns);
pluginManager.registerEvent(Type.PLAYER_RESPAWN, playerListener, Priority.Low, this); pluginManager.registerEvent(Type.PLAYER_RESPAWN, playerListener, Priority.Low, this);

View file

@ -6,12 +6,14 @@ import com.earth2me.essentials.IEssentialsModule;
import com.earth2me.essentials.settings.Spawns; import com.earth2me.essentials.settings.Spawns;
import com.earth2me.essentials.storage.AbstractDelayedYamlFileReader; import com.earth2me.essentials.storage.AbstractDelayedYamlFileReader;
import com.earth2me.essentials.storage.AbstractDelayedYamlFileWriter; import com.earth2me.essentials.storage.AbstractDelayedYamlFileWriter;
import com.earth2me.essentials.storage.ObjectLoadException;
import com.earth2me.essentials.storage.StorageObject; import com.earth2me.essentials.storage.StorageObject;
import java.io.File; import java.io.File;
import java.util.HashMap; import java.util.HashMap;
import java.util.Locale; import java.util.Locale;
import java.util.Map; import java.util.Map;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock;
import org.bukkit.Bukkit;
import org.bukkit.Location; import org.bukkit.Location;
import org.bukkit.World; import org.bukkit.World;
@ -35,6 +37,10 @@ public class SpawnStorage implements IConf, IEssentialsModule
rwl.writeLock().lock(); rwl.writeLock().lock();
try try
{ {
if (spawns == null)
{
spawns = new Spawns();
}
if (spawns.getSpawns() == null) if (spawns.getSpawns() == null)
{ {
spawns.setSpawns(new HashMap<String, Location>()); spawns.setSpawns(new HashMap<String, Location>());
@ -136,9 +142,21 @@ public class SpawnStorage implements IConf, IEssentialsModule
} }
@Override @Override
public void onFinish(final Spawns object) public void onSuccess(final Spawns object)
{
if (object != null)
{ {
spawns = object; spawns = object;
}
rwl.writeLock().unlock();
}
@Override
public void onException()
{
if (spawns == null) {
spawns = new Spawns();
}
rwl.writeLock().unlock(); rwl.writeLock().unlock();
} }
} }