/*
 * Copyright (c) 2017, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.s1asdev.jdbc41.ejb;

import jakarta.ejb.*;
import javax.naming.*;
import javax.sql.*;
import java.rmi.*;
import java.util.*;
import java.sql.*;
import jakarta.transaction.UserTransaction;
import java.util.concurrent.*;
import java.lang.reflect.Method;

public class SimpleSessionBean implements SessionBean
{

    private SessionContext ctxt_;
    private InitialContext ic_;
    private DataSource ds;
    PrintTask task1 = new PrintTask("thread1");
    PrintTask task2 = new PrintTask("thread2");
    ExecutorService threadExecutor = Executors.newCachedThreadPool(new ThreadFactory() {
        public Thread newThread(Runnable r) {
            Thread th = new Thread(r);
            th.setDaemon(true);
            return th;
        }
    });

    class PrintTask implements Runnable {
        private String threadName;

        public PrintTask(String name) {
            threadName = name;
        }

        public void run() {
            System.out.println("Thread : " + threadName + " executed");
        }
    }

    public void setSessionContext(SessionContext context) {
        ctxt_ = context;
        try {
            ic_ = new InitialContext();
            ds = (DataSource) ic_.lookup("java:comp/env/DataSource");
        } catch( Exception ne ) {
            ne.printStackTrace();
        }
    }

    public void ejbCreate() throws CreateException {
    }

    /**
     * Testing connection object
     */
    public boolean test1() throws Exception {
        Connection con = null;
        Connection[] conns = new Connection[32];
        Statement stmt = null;
        boolean passed = true;
        try {
            con = ds.getConnection("scott", "tiger");
            stmt = con.createStatement();

            //Try getting schema from con object
            System.out.println(">>> con.getSchema()=" + con.getSchema());

            //try getting network timeout from con object
            try {
                System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
            } catch(Exception ex) {
                passed = true;
                //Expected as this is not implemented in derby 10.8 jdbc41 driver
            }

            //Invoke abort on con object
            threadExecutor.submit(task1);
            threadExecutor.submit(task2);
            con.abort(threadExecutor);
            System.out.println("Marked connection for aborting...");

            //Try to use the aborted connection to create a statement. should fail
            try {
                con.createStatement();
            } catch(SQLException ex) {
                passed = true;
                System.out.println("Creating statement from con after tx abort : Failed : Expected");
            }
        } catch(Exception ex) {
            passed = false;
        } finally {
            if ( stmt != null ) {
                try { stmt.close(); } catch( Exception e1) {}
            }
            if ( con != null ) {
                try { con.close(); } catch( Exception e1) {}
            }
            for(int i=0; i<32; i++) {
                if(conns[i] != null) {
                    try { conns[i].close(); } catch(Exception ex) {}
                }
            }
        }
        return passed;
    }

    /**
     * Testing connection object in a User transaction
     */
    public boolean test2() throws Exception {
        Connection con = null;
        Connection[] conns = new Connection[32];
        Connection[] conns1 = new Connection[32];
        Statement[] stmts1 = new Statement[32];
        Statement stmt = null;
        boolean passed = true;
        try {
            UserTransaction usertx = (UserTransaction) ctxt_.getUserTransaction();
            usertx.begin();
            con = ds.getConnection("scott", "tiger");
            //System.out.println("!!!! TEST2 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(con)).hashCode());
            stmt = con.createStatement();

            //Invoke abort on con object
            threadExecutor.submit(task1);
            threadExecutor.submit(task2);
            con.abort(threadExecutor);
            System.out.println("Marked connection for aborting...");

            //Operations on con object after marking for abort in a tx
            //before tx commit, all operations on con pass
            //Try getting schema from con object
            System.out.println(">>> con.getSchema()=" + con.getSchema());

            //try getting network timeout from con object
            try {
                System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
            } catch(Exception ex) {
                passed = true;
                //Expected as this is not implemented in derby 10.8 jdbc41 driver
            }

            usertx.commit();

            //Try to use the aborted connection to create a statement. should fail
            try {
                con.createStatement();
            } catch(SQLException ex) {
                passed = true;
                System.out.println("Creating statement from con after tx abort : Failed : Expected");
            }
        } catch(Exception ex) {
            passed = false;
        } finally {
            if ( stmt != null ) {
                try { stmt.close(); } catch( Exception e1) {}
            }
            if ( con != null ) {
                try { con.close(); } catch( Exception e1) {}
            }
            //Test to get (max pool size-1) no. of connections after a previous abort.
            //should pass as the aborted connection is destroyed
            for(int i=0; i<32; i++) {
                if(i == 31) {
                    //Last connection should not be got
                    try {
                        conns1[i] = ds.getConnection("scott", "tiger");
                    } catch(Exception ex) {
                        passed = true;
                    }
                } else {
                    conns1[i] = ds.getConnection("scott", "tiger");
                    stmts1[i] = conns1[i].createStatement();
                }
            }
            for(int i=0; i<32; i++) {
                if(stmts1[i] != null) {
                    try { stmts1[i].close(); } catch(Exception ex) {}
                }
            }
            for(int i=0; i<32; i++) {
                if(conns1[i] != null) {
                    try { conns1[i].close(); } catch(Exception ex) {}
                }
            }
        }
        return passed;
    }

    //Tests DatabaseMetaData
    public boolean test3() throws Exception {
        Connection con = null;
        DatabaseMetaData dmd = null;
        boolean passed = true;
        try {
            con = ds.getConnection("scott", "tiger");
            dmd = con.getMetaData();
            //Testing DatabaseMetaData
            System.out.println(">>> dmd.generatedKeyAlwaysReturned()=" + dmd.generatedKeyAlwaysReturned());
        } catch(Exception ex) {
            passed = false;
        } finally {
            if(con!= null) {
                try { con.close(); } catch(Exception ex) {}
            }
        }
        return passed;
    }

    /**
     * Testing Statement object within a transaction
     */
    public boolean test4() throws Exception {
        Connection conn1 = null;
        Statement stmt1 = null;
        ResultSet rs1 = null;
        boolean passed = true;

        //Now try getting a connection again, but within a transaction
        int size = 0;
        try {
            createTable();
            UserTransaction tx = (UserTransaction) ctxt_.getUserTransaction();
            tx.begin();
            conn1 = ds.getConnection("scott", "tiger");
            System.out.println("!!!! TEST4 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(conn1)).hashCode());
            stmt1 = conn1.createStatement();
            insertRow(stmt1);

            threadExecutor.submit(task1);
            threadExecutor.submit(task2);

            //Testing Statement object
            stmt1.closeOnCompletion();
            System.out.println(">>> stmt1.isCloseOnCompletion()=" + stmt1.isCloseOnCompletion());

            conn1.abort(threadExecutor);

            System.out.println("Marked connection for aborting...");
            insertRow(stmt1);

            //After con abort (within the tx)
            size = getCount(stmt1);

            if(size != 2) {
                passed = false;
                System.out.println("FAILED as size!= 2");
            }

            tx.commit();

            try {
                //After tx commit, insert data. Should fail because of no current connection
                insertRow(stmt1);
            } catch(SQLException ex) {
                passed = true;
                System.out.println("Trying to insert data after a tx commit (prev. conn abort): Failed with : " + ex + " Expected.");
            }
        } catch (Exception e) {
           passed = false;
        } finally {
            if (rs1 != null ) {
                try { rs1.close(); } catch( Exception e1 ) {}
            }
            if ( stmt1 != null ) {
                try { stmt1.close(); } catch( Exception e1) {}
            }
            if ( conn1 != null ) {
                try { conn1.close(); } catch( Exception e1) {}
            }
        }
        return passed;
    }

    private void createTable() throws SQLException {
        String tableName = "JDBC41";

        Connection con = null;
        Statement stmt = null;
        con = ds.getConnection("scott", "tiger");
        stmt = con.createStatement();

        try{
            stmt.executeUpdate("drop table " + tableName);
        }catch(Exception ex){
        }
        stmt.executeUpdate("create table " + tableName + " (qty integer, name varchar(32))");
        stmt.close();
        con.close();
    }

    private void insertRow(Statement stmt) throws SQLException {
        String tableName = "JDBC41";
        stmt.executeUpdate("INSERT INTO " + tableName + " values (100, 'COFFEE')");
    }

    private int getCount(Statement stmt) throws SQLException {
        String tableName = "JDBC41";
        ResultSet rs = null;
        rs = stmt.executeQuery( "SELECT count(*) FROM " + tableName);
        rs.next();
        int count = rs.getInt(1);
        rs.close();
        return count;
    }

    public void ejbLoad() {}
    public void ejbStore() {}
    public void ejbRemove() {}
    public void ejbActivate() {}
    public void ejbPassivate() {}
    public void unsetEntityContext() {}
    public void ejbPostCreate() {}
}