package ware;

import java.sql.Connection;
import java.sql.Statement;
import java.sql.ResultSet;
import java.sql.DriverManager;
import java.sql.*;
/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2007</p>
 *
 * <p>Company: </p>
 *
 * @author not attributable
 * @version 1.0
 */
public class DBCon {
    Connection conn = null;
    Statement stat = null;
    ResultSet rs = null;
    String sql = null;
    public DBCon() {
        try {
            Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");
        } catch (ClassNotFoundException ex) {
            ex.printStackTrace();
        }
    }
    public int getNumber(String str , int num){
        sql = "select wareamount from waretable where warename='"+str+"'";
        try {
            conn = DriverManager.getConnection("jdbc:odbc:test", "sa", "");
            stat = conn.createStatement();
            rs = stat.executeQuery(sql);
            if(rs.next()){
                sql = "update waretable set wareamount=wareamount+"+num+" where warename='"+str+"'";
                System.out.println(sql);
                stat.executeUpdate(sql);
            }else{
                sql = "insert into waretable(warename, wareamount) values('"+str+"',"+num+")";
                System.out.println(sql);
                stat.executeUpdate(sql);
            }
            sql = "select wareamount from waretable where warename='"+str+"'";
            rs = stat.executeQuery(sql);
            if(rs.next()){
                num = rs.getInt(1);
            }
        } catch (SQLException ex) {
            ex.printStackTrace();
         }
        return num;
    }
}