RelationMigration.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.contacts.relation;
 
import java.util.stream.Stream;
import org.turro.string.Strings;
import org.turro.contacts.BusinessRelation;
import org.turro.contacts.Contact;
import org.turro.contacts.ContactType;
import org.turro.contacts.Relation;
import org.turro.contacts.db.ContactsPU;
import org.turro.jpa.Dao;
import org.turro.jpa.DaoUtil;
 
public class RelationMigration {
  
  public static void migrate() {
    RelationMigration migration = new RelationMigration();
    if(migration.isNecessary()) {
      migration.start();
    }
    qualifyCompanies();
  }
 
  private boolean isNecessary() {
    Dao dao = new ContactsPU();
    return DaoUtil.isEmpty(dao, "BusinessRelation") && !DaoUtil.isEmpty(dao, "Relation");
  }
 
  private void start() {
    Dao dao = new ContactsPU();
    try(Stream<Relation> relations = new ContactsPU().stream(Relation.class, "select r from Relation r")) {
      relations.forEach((relation) -> {
        if(Strings.isBlank(relation.getRelationType()) || !relation.getRelationType().startsWith("$INT_")) {
          BusinessRelation br = new BusinessRelation();
          br.setId(relation.getId());
          br.setContact(relation.getContact1());
          br.setBusiness(relation.getContact2());
          br.setRelationType(relation.getRelationType());
          br.setDescription(relation.getDescription());
          br.setPreferential(relation.isPreferential());
          br.setValidated(true);
          dao.saveObject(br);
        }
      });
    }
  }
 
  private static void qualifyCompanies() {
    Dao dao = new ContactsPU();
    try(Stream<BusinessRelation> relations = new ContactsPU().stream(BusinessRelation.class, 
            "select r from BusinessRelation r")) {
      relations.forEach((relation) -> {
        Contact business = relation.getBusiness();
        if(ContactType.CONTACT_USER.equals(business.getType())) {
          business.setType(ContactType.CONTACT_COMPANY);
          dao.saveObject(business);
        }
      });
    }
  }
 
}