public HostEntity getHostEntity() {
    return hostEntity;
  }
Sets the associated {@link StageEntity} for this command. If the
{@link StageEntity} has been persisted, then this will also set the
commands stage and request ID fields.

@param stage
ambari-server/src/main/java/org/apache/ambari/server/orm/entities/HostRoleCommandEntity.java