dpfmanager.shell.modules.database.DatabaseController Maven / Gradle / Ivy

Go to download
/**
 * DatabaseController.java
  This program is free software: you can redistribute it
 * and/or modify it under the terms of the GNU General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any later version; or,
 * at your choice, under the terms of the Mozilla Public License, v. 2.0. SPDX GPL-3.0+ or MPL-2.0+.
 * 
  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 General Public License and the Mozilla Public License for more details. 
 *  You should have received a copy of the GNU General Public License and the Mozilla Public
 * License along with this program. If not, see http://www.gnu.org/licenses/
 * and at http://mozilla.org/MPL/2.0 . 
  NB: for the
 * © statement, include Easy Innova SL or other company/Person contributing the code. 
  ©
 * 2015 Easy Innova, SL 
 *
 * @author Adria Llorens
 * @version 1.0
 * @since 23/7/2015
 */

package dpfmanager.shell.modules.database;

import dpfmanager.shell.core.adapter.DpfSpringController;
import dpfmanager.shell.core.config.BasicConfig;
import dpfmanager.shell.core.context.ConsoleContext;
import dpfmanager.shell.core.messages.DpfMessage;
import dpfmanager.shell.modules.database.core.DatabaseService;
import dpfmanager.shell.modules.database.messages.JobsMessage;
import dpfmanager.shell.modules.server.messages.PostMessage;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Controller;

import javax.annotation.PostConstruct;

/**
 * Created by Adria Llorens on 20/04/2016.
 */
@Controller(BasicConfig.MODULE_DATABASE)
public class DatabaseController extends DpfSpringController {

  @Autowired
  private DatabaseService service;

  @Autowired
  private ApplicationContext appContext;

  @Override
  synchronized public void handleMessage(DpfMessage dpfMessage) {
    if (dpfMessage.isTypeOf(JobsMessage.class)){
      service.handleJobsMessage(dpfMessage.getTypedMessage(JobsMessage.class));
    }
  }

  @Override
  public Object handleMessageWithResponse(DpfMessage dpfMessage) {
    if (dpfMessage.isTypeOf(PostMessage.class)){
      PostMessage pm = dpfMessage.getTypedMessage(PostMessage.class);
      if (pm.isAsk() && pm.getHash() == null){
        return service.getJobStatus(pm.getId());
      } else if (pm.isAsk()){
        return service.getJobStatusByHash(pm.getHash());
      }
    }
    return null;
  }

  @PostConstruct
  public void init() {
    service.setContext(new ConsoleContext(appContext));
  }
}