public void setUpgrade(UpgradeEntity upgrade) {
    this.upgrade = upgrade;
  }
Sets the associated upgrade entity.

@param upgrade
ambari-server/src/main/java/org/apache/ambari/server/orm/entities/UpgradeHistoryEntity.java