1 files changed, 15 insertions, 0 deletions
diff --git a/bundle-plugin/src/main/scala/com/yahoo/container/plugin/classanalysis/AttributeVisitorTrait.scala b/bundle-plugin/src/main/scala/com/yahoo/container/plugin/classanalysis/AttributeVisitorTrait.scala
new file mode 100644
index 00000000000..0603cebf5af
--- /dev/null
+++ b/bundle-plugin/src/main/scala/com/yahoo/container/plugin/classanalysis/AttributeVisitorTrait.scala
@@ -0,0 +1,15 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.plugin.classanalysis
+
+import org.objectweb.asm.{Type, Attribute}
+
+/**
+ * @author  tonytv
+ */
+private trait AttributeVisitorTrait {
+  protected val imports: ImportsSet
+
+  def visitAttribute(attribute: Attribute) {
+    imports ++= getClassName(Type.getObjectType(attribute.`type`)).toList
+  }
+}