cherry.sqlman.tool.load.SqlLoadControllerImpl.java Source code

Introduction

Here is the source code for cherry.sqlman.tool.load.SqlLoadControllerImpl.java

Source

/*
 * Copyright 2014,2015 agwlvssainokuni
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package cherry.sqlman.tool.load;

import static cherry.sqlman.ParamDef.REQ_ID;
import static cherry.sqlman.ParamDef.REQ_REF;
import static cherry.sqlman.util.ModelAndViewBuilder.redirect;
import static cherry.sqlman.util.ModelAndViewBuilder.withViewname;
import static org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder.fromMethodCall;
import static org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder.on;

import java.util.Locale;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.mobile.device.site.SitePreference;
import org.springframework.security.core.Authentication;
import org.springframework.stereotype.Controller;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.support.SessionStatus;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import org.springframework.web.util.UriComponents;
import org.springframework.web.util.UriComponentsBuilder;

import cherry.foundation.async.FileProcessResult;
import cherry.foundation.logicalerror.LogicalErrorUtil;
import cherry.sqlman.LogicError;
import cherry.sqlman.tool.metadata.MetadataService;
import cherry.sqlman.tool.metadata.SqlMetadataForm;
import cherry.sqlman.tool.shared.DataSourceDef;
import cherry.sqlman.tool.util.Util;
import cherry.sqlman.util.ViewNameUtil;

@Controller
public class SqlLoadControllerImpl extends SqlLoadSupport implements SqlLoadController {

    @Autowired
    private MetadataService metadataService;

    @Autowired
    private LoadService loadService;

    @Autowired
    private DataSourceDef dataSourceDef;

    private final String viewnameOfStart = ViewNameUtil
            .fromMethodCall(on(SqlLoadController.class).start(null, null, null, null, null, null, null));

    @Override
    public ModelAndView init(String redirTo, Integer ref, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request, SessionStatus status) {
        status.setComplete();
        return redirect(redirectOnInit(redirTo, ref)).build();
    }

    @Override
    public ModelAndView start(Integer ref, SqlLoadForm form, BindingResult binding, Authentication auth,
            Locale locale, SitePreference sitePref, NativeWebRequest request) {
        initializeForm(form, ref, auth);
        return withViewname(viewnameOfStart).build();
    }

    @Override
    public ModelAndView execute(SqlLoadForm form, BindingResult binding, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request, RedirectAttributes redirAttr) {

        if (hasErrors(form, binding)) {
            return withViewname(viewnameOfStart).build();
        }

        try {

            FileProcessResult result = handleFile(form.getFile(), form.getDatabaseName(), form.getSql());
            redirAttr.addFlashAttribute(result);
            form.setFile(null);

            return redirect(redirectOnExecute()).build();
        } catch (BadSqlGrammarException ex) {
            LogicalErrorUtil.reject(binding, LogicError.BadSqlGrammer, Util.getRootCause(ex).getMessage());
            return withViewname(viewnameOfStart).build();
        }
    }

    @Override
    public ModelAndView create(SqlLoadForm form, BindingResult binding, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request) {

        if (hasErrors(form, binding)) {
            return withViewname(viewnameOfStart).build();
        }

        int id = loadService.create(form, auth.getName());
        form.setFile(null);

        return redirect(redirectOnInit(redirectOnCreate(id).toUriString(), null)).build();
    }

    private UriComponents redirectOnInit(String redirTo, Integer ref) {
        if (StringUtils.isNotEmpty(redirTo)) {
            return UriComponentsBuilder.fromPath(redirTo).build();
        } else {
            UriComponentsBuilder ucb = fromMethodCall(
                    on(SqlLoadController.class).start(null, null, null, null, null, null, null));
            if (ref == null) {
                return ucb.replaceQuery(StringUtils.EMPTY).build();
            } else {
                return ucb.replaceQueryParam(REQ_REF, ref).build();
            }
        }
    }

    private UriComponents redirectOnExecute() {
        return fromMethodCall(on(SqlLoadController.class).start(null, null, null, null, null, null, null)).build();
    }

    private UriComponents redirectOnCreate(int id) {
        return fromMethodCall(on(SqlLoadIdController.class).init(null, 0, null, null, null, null, null))
                .replaceQueryParam(REQ_ID, id).build();
    }

    private void initializeForm(SqlLoadForm form, Integer ref, Authentication auth) {
        if (ref != null) {
            SqlMetadataForm mdForm = metadataService.findById(ref, auth.getName());
            if (mdForm != null) {
                SqlLoadForm f = loadService.findById(ref);
                if (f != null) {
                    form.setDatabaseName(f.getDatabaseName());
                    form.setSql(f.getSql());
                    form.setLockVersion(f.getLockVersion());
                    return;
                }
            }
        }
        form.setDatabaseName(dataSourceDef.getDefaultName());
    }

    private boolean hasErrors(SqlLoadForm form, BindingResult binding) {

        // ??
        if (binding.hasErrors()) {
            return true;
        }

        // ?

        // ??

        return false;
    }

}