chengkun
2 years ago
3 changed files with 671 additions and 33 deletions
@ -0,0 +1,492 @@ |
|||
//打开数据库(创建数据库或者有该数据库就打开)
|
|||
var name = "scancode" // 数据库名称
|
|||
|
|||
// 防止数据中的 单引号 报错
|
|||
export function html2Escape(str) { |
|||
// console.log("str",str,typeof str)
|
|||
if(typeof str === "string"){ |
|||
return str.replace(/'/g, "''"); |
|||
}else if(typeof str === "object"){ |
|||
return JSON.stringify(str).replace(/'/g, "''"); |
|||
}else{ |
|||
return str |
|||
} |
|||
|
|||
} |
|||
|
|||
export async function openSqlite(){ |
|||
//创建数据库或者打开
|
|||
//这plus.sqlite只在手机上运行
|
|||
return new Promise((resolve,reject) =>{ |
|||
if(isOpen()){ |
|||
// console.log("数据库已经打开了")
|
|||
resolve("数据库已经打开"); |
|||
return true |
|||
} |
|||
// console.log("开始打开数据库")
|
|||
plus.sqlite.openDatabase({ |
|||
name, //数据库名称
|
|||
path:`_doc/${name}.db`, //数据库地址，uniapp推荐以下划线为开头，这到底存在哪里去了，我也不清楚，哈哈
|
|||
success(){ |
|||
// console.log("成功打开数据库")
|
|||
resolve(); //成功回调
|
|||
}, |
|||
fail(e){ |
|||
// console.log("失败打开数据库",e)
|
|||
reject(e); //失败回调
|
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
//判断数据库是否打开
|
|||
export function isOpen(){ |
|||
//数据库打开了就返回true,否则返回false
|
|||
var open = plus.sqlite.isOpenDatabase({ |
|||
name, |
|||
path: `_doc/${name}.db` |
|||
}) |
|||
return open; |
|||
} |
|||
//关闭数据库
|
|||
export function closedb(){ |
|||
return new Promise((resolve,reject) =>{ |
|||
plus.sqlite.closeDatabase({ |
|||
name, |
|||
success(e){ |
|||
resolve(e); |
|||
}, |
|||
fail(e){ |
|||
reject(e); |
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
// 执行sql语句
|
|||
export function executeSql(sql){ |
|||
return new Promise((resolve,reject) =>{ |
|||
//创建表格在executeSql方法里写
|
|||
// console.log("开始执行自定义sql",sql)
|
|||
plus.sqlite.executeSql({ |
|||
name, |
|||
//表格创建或者打开，后面为表格结构
|
|||
sql:sql, |
|||
success(e){ |
|||
resolve(e); |
|||
}, |
|||
fail(e){ |
|||
console.error("自定义sql报错",e) |
|||
reject(e); |
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
// 查询所有数据表名
|
|||
export async function getTable(){ |
|||
return selectSql("select * FROM sqlite_master where type='table'") |
|||
} |
|||
// 获取表的所有字段
|
|||
export async function getAllField(tableName){ |
|||
return selectSql(`PRAGMA table_info([${tableName}])`) |
|||
} |
|||
|
|||
// 查询表数据总条数
|
|||
export const getCount = (tableName,whereObj={}) => { |
|||
let sql = `select count(*) as num from ${tableName} where 1 ` |
|||
if(Object.keys(whereObj).length>0){ |
|||
for(let i in whereObj){ |
|||
sql += ` and ${i} = '${whereObj[i]}' ` |
|||
} |
|||
} |
|||
console.log("getCount的SQL",sql) |
|||
return selectSql(sql) |
|||
} |
|||
// 查询表是否存在
|
|||
export const isTable = (tableName) => { |
|||
return new Promise((resolve, reject) => { |
|||
plus.sqlite.selectSql({ |
|||
name: name, |
|||
sql: `select count(*) as isTable FROM sqlite_master where type='table' and name='${tableName}'`, |
|||
success(e) { |
|||
resolve(e[0].isTable ? true : false); |
|||
}, |
|||
fail(e) { |
|||
console.log(e) |
|||
reject(e); |
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
|
|||
// 添加数据
|
|||
export async function addSql(tableName,obj={}) { |
|||
|
|||
if (Object.keys(obj).length>0) { |
|||
// 获取数据库的字段
|
|||
let fieldList = await getAllField(tableName) |
|||
// 取数据库有的字段
|
|||
let keys = intersection(fieldList,obj) |
|||
let keyStr = keys.toString() |
|||
let valStr = '' |
|||
keys.forEach((item, index) => { |
|||
valStr += (`'${html2Escape(obj[item])}',`) |
|||
}) |
|||
valStr = valStr.substring(0,valStr.length-1) |
|||
let sqlStr = `insert into ${tableName}(${keyStr}) values(${valStr})` |
|||
console.log("addSql",sqlStr) |
|||
return executeSql(sqlStr) |
|||
} else { |
|||
return new Promise((resolve, reject) => { |
|||
reject("错误") |
|||
}) |
|||
} |
|||
} |
|||
|
|||
|
|||
// 修改数据
|
|||
export async function updateSql(tableName, setData, whereObj={}){ |
|||
if(!Object.keys(whereObj).length){ |
|||
return new Promise((resolve, reject) => { |
|||
reject("请传whereObj") |
|||
}); |
|||
return false |
|||
} |
|||
if (setData&&JSON.stringify(setData) !== '{}') { |
|||
// 获取数据库的字段
|
|||
let fieldList = await getAllField(tableName) |
|||
let dataKeys = intersection(fieldList,setData) |
|||
let setStr = '' |
|||
// console.log("dataKeys",tableName,dataKeys)
|
|||
dataKeys.forEach((item, index) => { |
|||
// console.log(setData[item])
|
|||
setStr += ( |
|||
`${item} = '${setData[item]}',`) |
|||
}) |
|||
setStr = setStr.substring(0,setStr.length-1) |
|||
let sql = `update ${tableName} set ${setStr} where 1 ` |
|||
for(let i in whereObj){ |
|||
sql += ` and ${i} = '${whereObj[i]}' ` |
|||
} |
|||
// console.log("修改数据的sql",sql)
|
|||
return executeSql(sql) |
|||
} else { |
|||
return new Promise((resolve, reject) => { |
|||
reject("错误") |
|||
}); |
|||
} |
|||
} |
|||
|
|||
//删除数据库数据
|
|||
export async function deleteSql(tableName,whereObj={}){ |
|||
if (tableName&&JSON.stringify(whereObj) !== '{}') { |
|||
let dataKeys = Object.keys(whereObj) |
|||
let setStr = '' |
|||
dataKeys.forEach((item, index) => { |
|||
console.log(whereObj[item]) |
|||
setStr += ( |
|||
`${item}='${whereObj[item]}' ${dataKeys.length - 1 !== index ? " and " : ""}`) |
|||
}) |
|||
console.log("删除sql",`-- delete from ${tableName} where ${setStr}`) |
|||
let sql = `delete from ${tableName} where ${setStr}` |
|||
return executeSql(sql) |
|||
} else { |
|||
return new Promise((resolve, reject) => { |
|||
reject("错误") |
|||
}); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 获取分页数据库数据 |
|||
* @param tableName 表名称 |
|||
* @param pageObj 分页数据，如 {current:1,size:10} |
|||
* @param whereObj 查询 |
|||
* @param orderBy 排序 |
|||
* @returns {Promise<unknown>} |
|||
*/ |
|||
export const getPageList = async (tableName,pageObj,whereObj={},orderBy) => { |
|||
// console.log("数据库",tableName,pageObj,whereObj,orderBy)
|
|||
let current = parseInt(pageObj.current) |
|||
let size = parseInt(pageObj.size) |
|||
// console.log("数据库11")
|
|||
let pages = 0,total = 0 // pages为总页数，total为总数据条数
|
|||
let sql = '' |
|||
let numindex = 0 |
|||
await getCount(tableName,whereObj).then((resNum) => { |
|||
console.log("resNum",resNum,size) |
|||
pages = Math.ceil(resNum[0].num / size) |
|||
total = resNum[0].num |
|||
}) |
|||
if(((current - 1) * size) == 0) { |
|||
numindex = 0 |
|||
} else { |
|||
numindex = ((current - 1) * size) + 1 |
|||
} |
|||
sql = `select * from ${tableName} where 1 ` |
|||
if(Object.keys(whereObj).length>0){ |
|||
for(let i in whereObj){ |
|||
sql += ` and ${i} = '${whereObj[i]}' ` |
|||
} |
|||
} |
|||
if(orderBy) { |
|||
sql += ` order by ${orderBy} ` |
|||
} |
|||
sql += ` limit ${numindex},${size}` |
|||
console.log("分页sql，",sql) |
|||
if (pages < current - 1) { |
|||
return new Promise((resolve, reject) => { |
|||
reject("无数据") |
|||
}); |
|||
} else { |
|||
return new Promise((resolve, reject) => { |
|||
plus.sqlite.selectSql({ |
|||
name: name, |
|||
sql:sql , |
|||
success(e) { |
|||
resolve({ |
|||
code:0, |
|||
data:{ |
|||
data:{ |
|||
records:e, |
|||
pages, |
|||
total |
|||
} |
|||
}, |
|||
msg:null |
|||
}); |
|||
}, |
|||
fail(e) { |
|||
reject({data:e,code:500,msg:e}); |
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
} |
|||
/** |
|||
* 自定义语句查询，数据库数据 |
|||
* @param sql sql语句 |
|||
* @returns {Promise<any>} |
|||
*/ |
|||
export const selectSql = (sql) => { |
|||
// console.log("查询的sql",sql)
|
|||
return new Promise((resolve, reject) => { |
|||
plus.sqlite.selectSql({ |
|||
name: name, |
|||
sql: sql, |
|||
success(e) { |
|||
resolve(e); |
|||
}, |
|||
fail(e) { |
|||
console.log("查询报错",e) |
|||
reject(e); |
|||
} |
|||
}) |
|||
}) |
|||
} |
|||
|
|||
/** |
|||
* 查询数据库数据 |
|||
* @param tableName 表名 |
|||
* @param searchObj 查询的对象 |
|||
* @param orderBy 排序, 如 "id desc" |
|||
* @param selectStr |
|||
* @returns {Promise<any>} |
|||
*/ |
|||
export const selectList = (tableName,searchObj={},orderBy="") => { |
|||
let setStr = '' |
|||
let sql = '' |
|||
if (tableName) { |
|||
if (JSON.stringify(searchObj) !== '{}') { |
|||
let dataKeys = Object.keys(searchObj) |
|||
dataKeys.forEach((item, index) => { |
|||
// console.log(searchObj[item])
|
|||
setStr += ( |
|||
`${item}='${html2Escape(searchObj[item])}' ${dataKeys.length - 1 !== index ? " and " : ""}`) |
|||
}) |
|||
} |
|||
sql = `select * from ${tableName} ` |
|||
sql += setStr ? ` where ${setStr} ` : "" |
|||
|
|||
if(orderBy) { |
|||
sql += ` order by ${orderBy} ` |
|||
} |
|||
console.log("查询数据库数据",sql) |
|||
return selectSql(sql) |
|||
} else { |
|||
return new Promise((resolve, reject) => { |
|||
reject("错误") |
|||
}); |
|||
} |
|||
} |
|||
|
|||
|
|||
//将SQL语句进行拼接values(),()...然后再一次性插入，
|
|||
export async function insertAll(tableName="",data=[]){ |
|||
let sql = "" |
|||
// 获取数据库的字段
|
|||
try{ |
|||
let fieldList = await getAllField(tableName) |
|||
// console.log("fieldList",fieldList)
|
|||
if (tableName && data.length > 0) { |
|||
// 取交集的字段，才是合法的，能存进数据库的字段（注意：取的字段，是按照第一条数据的字段，进行赋值的）
|
|||
let keys = intersection(fieldList,data[0]) |
|||
let keyStr = keys.toString() |
|||
console.log("keyStr",keyStr) |
|||
|
|||
sql = `insert into ${tableName} (${keyStr}) values`; |
|||
for (let i = 0; i < data.length; i++) { |
|||
sql += "("; |
|||
for (const [k, v] of keys.entries()) { |
|||
// console.log("data[i][v]",k,v,data[i][v],data)
|
|||
sql += `'${html2Escape(data[i][v])}',` |
|||
} |
|||
sql = sql.substring(0, sql.length - 1) |
|||
sql += "),"; |
|||
} |
|||
sql = sql.substring(0, sql.length - 1) |
|||
console.log("批量插入的sql", sql) |
|||
} else { |
|||
this.$msg("表名为空或者插入数据为空") |
|||
} |
|||
}catch(e){ |
|||
console.error("批量插入报错",e) |
|||
} |
|||
return executeSql(sql) |
|||
} |
|||
|
|||
/** |
|||
* 批量修改 data数组 field关键字段 批量修改函数 传参方式 |
|||
* @param tableName 表名 |
|||
* @param data 需要修改的数据 |
|||
* @param field 表主键字段，通常是 id |
|||
* @returns {Promise<unknown>} |
|||
*/ |
|||
export async function batchUpdate(tableName='',data=[],field=''){ |
|||
|
|||
let sql = "" |
|||
let con = []; |
|||
let con_sql = []; |
|||
let fields = []; |
|||
let value |
|||
let temp |
|||
try{ |
|||
|
|||
if (!tableName || !data || !field) { |
|||
return false; |
|||
} else { |
|||
sql = 'UPDATE ' + tableName; |
|||
} |
|||
|
|||
for (let key in data) { |
|||
value = data[key] |
|||
let x = 0; |
|||
let v |
|||
// console.log("value",value)
|
|||
for (let k in value) { |
|||
v = value[k] |
|||
if (k != field && !con[x] && x == 0) { |
|||
con[x] = ` set ${k} = (CASE ${field} `; |
|||
} else if (k != field && !con[x] && x > 0) { |
|||
con[x] = ` ${k} = (CASE ${field} `; |
|||
} |
|||
if (k != field) { |
|||
temp = value[field]; |
|||
con_sql[x] = con_sql[x] ? con_sql[x] : "" |
|||
con_sql[x] += ` WHEN '${temp}' THEN '${html2Escape(v)}' `; |
|||
x++; |
|||
} |
|||
} |
|||
temp = value[field]; |
|||
// console.log("temp",fields,temp)
|
|||
if (fields.indexOf(temp)===-1) { |
|||
fields.push(temp) |
|||
} |
|||
} |
|||
let num = con.length - 1; |
|||
for (let [key, value] of con.entries()) { |
|||
for (let [k, v] of con_sql.entries()) { |
|||
if (k == key && key < num) { |
|||
sql += value + v + ' end),'; |
|||
} else if (k == key && key == num) { |
|||
sql += value + v + ' end)'; |
|||
} |
|||
} |
|||
} |
|||
let str = fields.join(","); |
|||
sql += ` where ${field} in(${str})`; |
|||
console.log("批量更新的sql", sql) |
|||
|
|||
|
|||
}catch(e){ |
|||
console.error("批量修改报错",e) |
|||
} |
|||
return executeSql(sql) |
|||
} |
|||
|
|||
|
|||
|
|||
//////////////////////////////////////////////////////////////////////
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
|
|||
// 合并数据
|
|||
export const mergeSql = (name,tableName,tabs) => { |
|||
if (!tabs || tabs.length == 0) { |
|||
return new Promise((resolve, reject) => { |
|||
reject("错误") |
|||
}) |
|||
} |
|||
let itemValStr = '' |
|||
tabs.forEach((item, index) => { |
|||
let itemKey = Object.keys(item) |
|||
let itemVal = '' |
|||
itemKey.forEach((key, i) => { |
|||
if (itemKey.length - 1 == i) { |
|||
if (typeof item[key] == 'object') { |
|||
itemVal += (`'${JSON.stringify(item[key])}'`) |
|||
} else { |
|||
itemVal += (`'${item[key]}'`) |
|||
} |
|||
} else { |
|||
if (typeof item[key] == 'object') { |
|||
itemVal += (`'${JSON.stringify(item[key])}',`) |
|||
} else { |
|||
itemVal += (`'${item[key]}',`) |
|||
} |
|||
} |
|||
}) |
|||
if (tabs.length - 1 == index) { |
|||
itemValStr += ('(' + itemVal + ')') |
|||
} else { |
|||
itemValStr += ('(' + itemVal + '),') |
|||
} |
|||
}) |
|||
let keys = Object.keys(tabs[0]) |
|||
let keyStr = keys.toString() |
|||
let sql = `insert or ignore into ${tableName} (${keyStr}) values ${itemValStr}` |
|||
return executeSql(sql) |
|||
} |
|||
/** |
|||
* 过滤非数据表的字段，使其不报错 |
|||
* @param fieldList ，使用 getAllField()函数查询出来的数据 |
|||
* @param obj 数据 |
|||
* @returns [] |
|||
*/ |
|||
export function intersection(fieldList=[],obj){ |
|||
// 获取数据库的字段
|
|||
let arrField = {} |
|||
for(let i in fieldList){ |
|||
arrField[fieldList[i].name] = null |
|||
} |
|||
// 取交集的字段，才是合法的，能存进数据库的字段
|
|||
let keys = [] |
|||
for (let i in arrField) { |
|||
if (obj.hasOwnProperty(i)) { |
|||
keys.push(i) |
|||
} |
|||
} |
|||
|
|||
return keys |
|||
} |
|||
Loading…
Reference in new issue