ProcedureMysql.java 1.85 KB
Newer Older
Paolo committed
1 2
package web.multitask.app.mysql;

3

Paolo committed
4 5 6 7 8 9 10 11 12
import org.json.JSONObject;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;

@Service
public class ProcedureMysql {
13
    
Paolo committed
14
    private final JdbcTemplate jdbcTemplate;
15

Paolo committed
16 17 18 19
    public ProcedureMysql(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

20
    public JSONObject ProcedureExecution(String procedure ,String database, Object[] params) {
Paolo committed
21
        try {
22 23 24 25 26 27 28 29 30 31 32 33 34
            StringBuilder query = new StringBuilder("CALL " + database + "." + procedure);

            if (params.length > 0) {
                query.append("(");
                for (int i = 0; i < params.length; i++) {
                    query.append("?");
                    if (i < params.length - 1) {
                        query.append(",");
                    }
                }
                query.append(")");
            }

35
            String checkProcedure = "SELECT COUNT(*) FROM information_schema.routines WHERE routine_schema = '" + database + "' AND routine_name = '" + procedure;
36 37 38 39 40 41 42
            List<Map<String, Object>> countProcedure = jdbcTemplate.queryForList(checkProcedure);
            
            if (countProcedure.get(0).get("COUNT(*)").toString().equals("0")) {
                return new JSONObject().put("message", "Procedure not found").put("status", false);
            }

            List<Map<String, Object>> list = jdbcTemplate.queryForList(query.toString(), params);
Paolo committed
43 44
            JSONObject result = new JSONObject();
            result.put("data", list);
45 46
            result.put("message", "Success");
            result.put("status", true);
Paolo committed
47 48
            return result;
        } catch (Exception e) {
49
            return new JSONObject().put("data", new JSONObject()).put("message", e.getMessage()).put("status", false);
Paolo committed
50 51 52
        }
    }
}