From 6f4ec9c4cc5cffdd830ab0536c82c6aa1eed927b Mon Sep 17 00:00:00 2001
From: fullwall <fullwall25@gmail.com>
Date: Tue, 20 Aug 2019 14:35:41 +0800
Subject: [PATCH] Implement new method

---
 .../java/net/citizensnpcs/npc/CitizensTraitFactory.java     | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java b/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java
index 497b45d0e..07155da12 100644
--- a/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java
+++ b/main/src/main/java/net/citizensnpcs/npc/CitizensTraitFactory.java
@@ -1,5 +1,6 @@
 package net.citizensnpcs.npc;
 
+import java.util.Collection;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -132,6 +133,11 @@ public class CitizensTraitFactory implements TraitFactory {
         registered.remove(info.getTraitName());
     }
 
+    @Override
+    public Collection<TraitInfo> getRegisteredTraits() {
+        return registered.values();
+    }
+
     @Override
     public <T extends Trait> T getTrait(Class<T> clazz) {
         for (TraitInfo entry : registered.values()) {