Launching all entities struck by lightning into the air

Yeah, this is pretty specific, but I really wanna figure it out.

So like it says, I want any entities that get struck by lightning to get launched into the air. Here's my code:

Code:java

@EventHandler
public void onStruckByLightning(LightningStrikeEvent e)
{
double x = e.getLightning().getLocation().getX();
double y = e.getLightning().getLocation().getY();
double z = e.getLightning().getLocation().getZ();
List<Entity> entity = e.getLightning().getNearbyEntities(x, y, z);
Entity[] ent = entity.toArray(new Entity[entity.size()]);
for (int i = 0; i < ent.length; i++)
{
Entity currentEnt = ent[i]; 
   currentEnt.setVelocity(currentEnt.getVelocity().add(new Vector(0, 1.25, 0)));
}
}}[/i]

This doesn't do anything though. I think my problem is that I can't figure out how to get each individual entity hit by lightning; I wasn't really sure what I was doing with the list and array things. So if this is possible and someone knows how to do this, please tell me how.

Thank you in advance

I don't know why I didn't think of this but this worked:

Code:java

@EventHandler
public void onEntityDamage(EntityDamageEvent event)
    {
if (!(event.getEntity() instanceof Player))
{
if (doLaunch)
{
if (event.getCause() == EntityDamageEvent.DamageCause.LIGHTNING && event.getCause() != null)
{
event.getEntity().setVelocity(event.getEntity().getVelocity().add(new Vector(0, 1.25, 0)));
doLaunch = false;
}
}
}

Neeevermind ._.

Actually that still doesn't work. It only affects the nearest mob to the lightning bolt.

If anyone feels like looking into this go ahead, but I don't feel like getting this to work anymore.

EDIT by Moderator: merged posts, please use the edit button instead of double posting.

 

So, you mean something like this?

Code:

LightningStrikeEvent:
  for(Entity e : event.getLightning().getNearbyEntities(3, 3, 3)) {
    e.setVelocity(new Vector(0, 2, 0));
  }

 

Yeah, that works. Thank you. I haven't used those types of loops before, so that never really occurred to me.