starter
/
starter-wsqlite
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

Browse Source
master
wangbing 5 years ago
parent 3d9e68778a 63b3669665
commit 14160a2cec
4 changed files with 85 additions and 111 deletions
Show Stats Download Patch File Download Diff File

72
src/main/java/xyz/wbsite/wsqlite/Client.java
Unescape View File

@ -2,6 +2,7 @@ package xyz.wbsite.wsqlite;
import java.io.File; import java.io.File;
import java.sql.*; import java.sql.*;
import java.sql.Date;
/** /**
* xyz.wbsite.wsqlite.Client * xyz.wbsite.wsqlite.Client
@ -11,7 +12,6 @@ import java.sql.*;
public class Client { public class Client {
Connection connection; Connection connection;
Statement statement;
ResultSet resultSet; ResultSet resultSet;
String dbFilePath; String dbFilePath;
@ -38,7 +38,7 @@ public class Client {
*/ */
public void execute(String sql) throws SQLException, ClassNotFoundException { public void execute(String sql) throws SQLException, ClassNotFoundException {
try { try {
getStatement().execute(sql); executeUpdate(sql);
} finally { } finally {
destroyed(); destroyed();
} }
@ -51,27 +51,61 @@ public class Client {
* @throws SQLException * @throws SQLException
* @throws ClassNotFoundException * @throws ClassNotFoundException
*/ */
public ResultSet executeQuery(String sql) throws SQLException, ClassNotFoundException { public ResultSet executeQuery(String sql, Object... args) throws SQLException, ClassNotFoundException {
return getStatement().executeQuery(sql); PreparedStatement preparedStatement = getConnection().prepareStatement(sql);
setArg(preparedStatement, args);
return preparedStatement.executeQuery();
} }
/** /**
* 执行sql更新语句 * 执行sql语句
* *
* @param sqls * @param sql
* @throws SQLException * @throws SQLException
* @throws ClassNotFoundException * @throws ClassNotFoundException
*/ */
public int executeUpdate(String... sqls) throws SQLException, ClassNotFoundException { public int executeUpdate(String sql, Object... args) throws SQLException, ClassNotFoundException {
int count = 0;
try { try {
for (String sql : sqls) { PreparedStatement preparedStatement = getConnection().prepareStatement(sql);
count += getStatement().executeUpdate(sql); setArg(preparedStatement, args);
} return preparedStatement.executeUpdate();
} finally { } finally {
destroyed(); destroyed();
} }
return count; }
private void setArg(PreparedStatement ps, Object... args) throws SQLException {
for (int i = 0; i < args.length; i++) {
Object arg = args[i];
if (arg == null) {
ps.setNull(i + 1, Types.NULL);
} else if (arg instanceof String) {
ps.setString(i + 1, (String) arg);
} else if (arg instanceof Date) {
ps.setDate(i + 1, (Date) arg);
} else if (arg instanceof Time) {
ps.setTime(i + 1, (Time) arg);
} else if (arg instanceof java.util.Date) {
ps.setLong(i + 1, ((java.util.Date) arg).getTime());
} else if (arg instanceof Boolean) {
ps.setBoolean(i + 1, (Boolean) arg);
} else if (arg instanceof Byte) {
ps.setByte(i + 1, (Byte) arg);
} else if (arg instanceof Short) {
ps.setShort(i + 1, (Short) arg);
} else if (arg instanceof Integer) {
ps.setInt(i + 1, (int) arg);
} else if (arg instanceof Long) {
ps.setLong(i + 1, (long) arg);
} else if (arg instanceof Float) {
ps.setFloat(i + 1, (float) arg);
} else if (arg instanceof Double) {
ps.setDouble(i + 1, (double) arg);
} else if (arg instanceof byte[]) {
ps.setBytes(i + 1, (byte[]) arg);
}
}
} }
/** /**
@ -89,16 +123,6 @@ public class Client {
return connection; return connection;
} }
/**
* 获取数据库Statement对象
*/
Statement getStatement() throws SQLException, ClassNotFoundException {
if (null == statement) {
statement = getConnection().createStatement();
}
return statement;
}
/** /**
* 数据库资源关闭和释放 * 数据库资源关闭和释放
*/ */
@ -109,10 +133,6 @@ public class Client {
resultSet = null; resultSet = null;
} }
if (null != statement) {
statement.close();
statement = null;
}
if (null != connection) { if (null != connection) {
connection.close(); connection.close();
connection = null; connection = null;

88
src/main/java/xyz/wbsite/wsqlite/ObjectClient.java
Unescape View File

@ -5,10 +5,7 @@ import xyz.wbsite.wsqlite.anonation.TableField;
import java.io.File; import java.io.File;
import java.lang.reflect.Field; import java.lang.reflect.Field;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.ArrayList; import java.util.*;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/** /**
* xyz.wbsite.wsqlite.Client * xyz.wbsite.wsqlite.Client
@ -67,6 +64,9 @@ public class ObjectClient extends Client {
} else if (f.getType() == Byte[].class || f.getType() == byte[].class) { } else if (f.getType() == Byte[].class || f.getType() == byte[].class) {
sql.append(f.getName().toUpperCase()); sql.append(f.getName().toUpperCase());
sql.append(" BLOB,"); sql.append(" BLOB,");
} else if (f.getType() == Date.class) {
sql.append(f.getName().toUpperCase());
sql.append(" TIMESTAMP,");
} }
} }
} }
@ -94,39 +94,14 @@ public class ObjectClient extends Client {
StringBuffer fieldsSql = new StringBuffer(); StringBuffer fieldsSql = new StringBuffer();
StringBuffer valueSql = new StringBuffer(); StringBuffer valueSql = new StringBuffer();
Object[] values = new Object[fs.size()];
for (int i = 0; i < fs.size(); i++) { for (int i = 0; i < fs.size(); i++) {
Field f = fs.get(i); Field f = fs.get(i);
f.setAccessible(true); f.setAccessible(true);
fieldsSql.append(f.getName().toUpperCase()); fieldsSql.append(f.getName().toUpperCase());
valueSql.append("?");
Object value = f.get(po); values[i] = f.get(po);
if (f.getType() == String.class) {
valueSql.append("'");
valueSql.append(value);
valueSql.append("'");
} else if (f.getType() == Boolean.class || f.getType() == boolean.class) {
valueSql.append("'");
valueSql.append(String.valueOf(value));
valueSql.append("'");
} else if (f.getType() == Byte.class || f.getType() == byte.class ||
f.getType() == Short.class || f.getType() == short.class ||
f.getType() == Character.class || f.getType() == char.class ||
f.getType() == Integer.class || f.getType() == int.class ||
f.getType() == Long.class || f.getType() == long.class) {
valueSql.append("'");
valueSql.append(value);
valueSql.append("'");
} else if (f.getType() == Float.class || f.getType() == float.class ||
f.getType() == Double.class || f.getType() == double.class) {
valueSql.append("'");
valueSql.append(value);
valueSql.append("'");
} else if (f.getType() == Byte[].class || f.getType() == byte[].class) {
valueSql.append("'");
valueSql.append(value);
valueSql.append("'");
}
if (i != fs.size() - 1) { if (i != fs.size() - 1) {
fieldsSql.append(","); fieldsSql.append(",");
valueSql.append(","); valueSql.append(",");
@ -138,7 +113,7 @@ public class ObjectClient extends Client {
sql.append(valueSql); sql.append(valueSql);
sql.append(")"); sql.append(")");
System.out.println("SQL ==> " + sql.toString()); System.out.println("SQL ==> " + sql.toString());
return executeUpdate(sql.toString()); return executeUpdate(sql.toString(), values);
} }
} catch (IllegalAccessException e) { } catch (IllegalAccessException e) {
e.printStackTrace(); e.printStackTrace();
@ -180,47 +155,21 @@ public class ObjectClient extends Client {
if (aClass == null) { if (aClass == null) {
System.err.println(poClass.getName() + " not found."); System.err.println(poClass.getName() + " not found.");
} else { } else {
//获取字段列表
List<Field> fs = getFields(poClass);
StringBuffer sql = new StringBuffer(); StringBuffer sql = new StringBuffer();
Object[] values = new Object[fs.size()];
sql.append("UPDATE "); sql.append("UPDATE ");
sql.append(aClass.getSimpleName()); sql.append(aClass.getSimpleName());
sql.append(" SET "); sql.append(" SET ");
//获取字段列表
List<Field> fs = getFields(poClass);
for (int i = 0; i < fs.size(); i++) { for (int i = 0; i < fs.size(); i++) {
Field f = fs.get(i); Field f = fs.get(i);
f.setAccessible(true); f.setAccessible(true);
sql.append(f.getName()); sql.append(f.getName());
sql.append(" = "); sql.append(" = ?");
values[i] = f.get(po);
Object value = f.get(po);
if (f.getType() == String.class) {
sql.append("'");
sql.append(value);
sql.append("'");
} else if (f.getType() == Boolean.class || f.getType() == boolean.class) {
sql.append("'");
sql.append(String.valueOf(value));
sql.append("'");
} else if (f.getType() == Byte.class || f.getType() == byte.class ||
f.getType() == Short.class || f.getType() == short.class ||
f.getType() == Character.class || f.getType() == char.class ||
f.getType() == Integer.class || f.getType() == int.class ||
f.getType() == Long.class || f.getType() == long.class) {
sql.append("'");
sql.append(value);
sql.append("'");
} else if (f.getType() == Float.class || f.getType() == float.class ||
f.getType() == Double.class || f.getType() == double.class) {
sql.append("'");
sql.append(value);
sql.append("'");
} else if (f.getType() == Byte[].class || f.getType() == byte[].class) {
sql.append("'");
sql.append(value);
sql.append("'");
}
if (i != fs.size() - 1) { if (i != fs.size() - 1) {
sql.append(","); sql.append(",");
} }
@ -236,7 +185,7 @@ public class ObjectClient extends Client {
} }
System.out.println("SQL ==> " + sql.toString()); System.out.println("SQL ==> " + sql.toString());
return executeUpdate(sql.toString()); return executeUpdate(sql.toString(), values);
} }
} catch (IllegalAccessException e) { } catch (IllegalAccessException e) {
e.printStackTrace(); e.printStackTrace();
@ -348,9 +297,12 @@ public class ObjectClient extends Client {
} else if (f.getType() == Byte[].class || f.getType() == byte[].class) { } else if (f.getType() == Byte[].class || f.getType() == byte[].class) {
byte[] v = resultSet.getBytes(f.getName()); byte[] v = resultSet.getBytes(f.getName());
f.set(o, v); f.set(o, v);
} else if (f.getType() == Date.class) {
Date v = resultSet.getDate(f.getName());
f.set(o, v);
} else { } else {
String string = resultSet.getString(f.getName()); String v = resultSet.getString(f.getName());
f.set(o, string); f.set(o, v);
} }
} }
rsList.add(o); rsList.add(o);

22
src/main/java/xyz/wbsite/wsqlite/entity/Example.java
Unescape View File

@ -2,6 +2,8 @@ package xyz.wbsite.wsqlite.entity;
import xyz.wbsite.wsqlite.anonation.TableField; import xyz.wbsite.wsqlite.anonation.TableField;
import java.util.Date;
public class Example { public class Example {
@TableField(40) @TableField(40)
@ -15,8 +17,6 @@ public class Example {
@TableField @TableField
private short s1; private short s1;
@TableField @TableField
private char c1;
@TableField
private int age; private int age;
@TableField @TableField
private long id; private long id;
@ -24,6 +24,8 @@ public class Example {
private float f1; private float f1;
@TableField @TableField
private double d1; private double d1;
@TableField
private Date date;
public String getS() { public String getS() {
return s; return s;
@ -65,14 +67,6 @@ public class Example {
this.s1 = s1; this.s1 = s1;
} }
public char getC1() {
return c1;
}
public void setC1(char c1) {
this.c1 = c1;
}
public int getAge() { public int getAge() {
return age; return age;
} }
@ -104,4 +98,12 @@ public class Example {
public void setD1(double d1) { public void setD1(double d1) {
this.d1 = d1; this.d1 = d1;
} }
public Date getDate() {
return date;
}
public void setDate(Date date) {
this.date = date;
}
} }

12
src/test/java/WsqliteTest.java
Unescape View File

@ -3,8 +3,8 @@ import xyz.wbsite.wsqlite.entity.Example;
import java.io.File; import java.io.File;
import java.sql.*; import java.sql.*;
import java.util.ArrayList; import java.util.*;
import java.util.List; import java.util.Date;
public class WsqliteTest { public class WsqliteTest {
public static void main(String[] args) { public static void main(String[] args) {
@ -17,14 +17,14 @@ public class WsqliteTest {
Example example = new Example(); Example example = new Example();
example.setS("wangbing"); example.setS("wangbing");
example.setB(false); example.setB(false);
example.setBs(null); example.setBs(new byte[]{-1, 0, 1});
example.setB1((byte) 1); example.setB1((byte) 1);
example.setS1((short) 1); example.setS1((short) 1);
example.setC1((char) 1);
example.setAge(1); example.setAge(1);
example.setId(1L); example.setId(1L);
example.setF1(0.1f); example.setF1(0.1f);
example.setD1(0.1); example.setD1(0.1);
example.setDate(new Date());
h.insert(Example.class, example); h.insert(Example.class, example);
@ -35,14 +35,14 @@ public class WsqliteTest {
System.out.println("测试 update"); System.out.println("测试 update");
example.setS("wangbing"); example.setS("wangbing");
example.setB(false); example.setB(false);
example.setBs(null); example.setBs(new byte[]{-2, 0, 2});
example.setB1((byte) 1); example.setB1((byte) 1);
example.setS1((short) 1); example.setS1((short) 1);
example.setC1((char) 1);
example.setAge(1); example.setAge(1);
example.setId(1L); example.setId(1L);
example.setF1(0.1f); example.setF1(0.1f);
example.setD1(0.1); example.setD1(0.1);
example.setDate(new Date());
int update = h.update(Example.class, example); int update = h.update(Example.class, example);
System.out.println(update); System.out.println(update);

Write Preview
Loading…
Cancel
Save

Powered by TurnKey Linux.