· 8 years ago · Nov 04, 2017, 06:42 AM
1public class DBUtil {
2 String JDBC_URL = "jdbc:sqlite:d:/sqlite/reg.db";
3 Connection con;
4 PreparedStatement ps;
5
6 public void insertRecord(String tableName, String user, String pass) {
7 String sql = "insert into " + tableName + " values(?,?);";
8 try {
9 ps = con.prepareStatement(sql);
10 ps.setString(1, user);
11 ps.setString(2, pass);
12 ps.executeUpdate();
13 } catch (SQLException e) {
14 // TODO Auto-generated catch block
15 e.printStackTrace();
16 }
17 }
18
19 public void createTable(String tableName, String fname, String pass) {
20 String sql = "create table if not exists " + tableName + "(" + fname +","+ pass+ " text);";
21 try {
22 ps = con.prepareStatement(sql);
23 ps.executeUpdate();
24 } catch (SQLException e) {
25 // TODO Auto-generated catch block
26 e.printStackTrace();
27 }
28
29 }
30
31 public boolean validate(String user, String pass) {
32 String sql = "select * from login where user = ? and pass = ?";
33 try {
34 ps = con.prepareStatement(sql);
35 ps.setString(1, user);
36 ps.setString(2, pass);
37 ResultSet rst = ps.executeQuery();
38 if (rst.next())
39 return true;
40 else
41 return false;
42 } catch (SQLException e) {
43 // TODO Auto-generated catch block
44 e.printStackTrace();
45 }
46 return true;
47 }
48
49 public void init() {
50 new org.sqlite.JDBC();
51 try {
52 con = DriverManager.getConnection(JDBC_URL);
53 } catch (SQLException e) {
54 // TODO Auto-generated catch block
55 e.printStackTrace();
56 }
57 }
58
59 public static void main(String[] args) {
60 // TODO Auto-generated method stub
61 DBUtil util = new DBUtil();
62 util.init();
63
64 util.createTable("login", "user","pass");
65 util.insertRecord("login", "Nestea", "asd");
66 util.insertRecord("login", "Lipton", "asd");
67 util.insertRecord("login", "Milo", "asd");
68
69
70 /*System.out.println("valid? " + util.validate("Nestea"));
71 System.out.println("valid? " + util.validate("jilo"));*/
72
73 try {
74 util.con.close();
75
76 } catch (SQLException e) {
77 // TODO Auto-generated catch block
78 e.printStackTrace();
79 }
80 System.out.println("End");
81
82 }
83
84}