@本文來源於公衆號:csdn2299,喜歡可以關注公衆號 程序員學府
今天小編就爲大家分享一篇使用Python開發SQLite代理服務器的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
SQLite數據庫使用單個磁盤文件,並且不需要像Oracle、MSSQL、MySQL等數據庫管理系統那樣啓動服務,使用非常靈活方便。但是SQLite也有個很嚴重的問題,就是沒有相應的服務,也沒有監聽任何端口,因此相應的程序只能訪問本地數據庫。也就是說,無法分離程序和數據庫,只能把程序和數據庫放在同一臺計算機上。
本文使用Python開發了一個SQLite數據庫的服務程序,可以完美地分離程序和數據庫。技術要點是Socket編程,在數據庫服務器上運行服務程序,該服務程序監聽特定端口、執行代理程序發來的SQL語句並返回結果;代理程序負責接收客戶端的SQL語句並轉發給服務器,然後再把服務器返回的結果轉發給客戶端。在具體使用時可以在本文代碼基礎上進行簡化和擴展。
服務程序:
##
# 服務器程序,接收代理服務器轉發來的SQL指令,並返回結果
#
import sqlite3
import socket
import struct
def getData(sql):
'''通過給定的SQL SELECT語句返回結果'''
with sqlite3.connect(r'database.db') as conn:
cur = conn.cursor()
cur.execute(sql)
result = cur.fetchall()
return result
def doSql(sql):
'''適用於DELETE/UPDATE/INSERT INTO語句,返回影響的記錄條數'''
with sqlite3.connect(r'database.db') as conn:
cur = conn.cursor()
result = cur.execute(sql)
return result.rowcount
# 創建socket對象,默認使用IPV4+TCP
sockServer = socket.socket()
sockServer.bind(('', 3030))
sockServer.listen(1)
while True:
# 接收客戶端連接
try:
conn, addr = sockServer.accept()
except:
continue
sql = conn.recv(1024).decode('gbk').lower()
if sql.startswith(('update','delete','insert')):
try:
# 首先發送要發送的字節總數量
# 然後再發送真實數據
result = str(doSql(sql)).encode('gbk')
conn.send(struct.pack('i', len(result)))
conn.send(result)
except:
message = b'error'
conn.send(struct.pack('i', len(message)))
conn.send(message)
elif sql.startswith('select'):
try:
result = str(getData(sql)).encode('gbk')
conn.send(struct.pack('i', len(result)))
conn.send(result)
except:
message = b'error'
conn.send(struct.pack('i', len(message)))
conn.send(message)
代理程序:
##
# 代理服務器,在SQLite數據庫服務器和客戶端之間進行指令和數據的轉發
# 這樣可以把數據庫和程序放到兩個服務器上進行分離
##
import socket
from threading import Thread
import struct
sockServer = socket.socket()
sockServer.bind(('',5050))
sockServer.listen(50)
def agent(conn):
# 接收客戶端發來的指令,並進行過濾
sql = conn.recv(1024)
if not sql.decode('gbk').startswith(('select', 'delete', 'insert','update')):
message = b'not a sql statement'
conn.send(struct.pack('i', len(message)))
conn.send(message)
return
else:
sockClient = socket.socket()
# 嘗試連接服務器
try:
sockClient.connect(('10.2.1.3', 3030))
except:
message = b'Server not alive'
conn.send(struct.pack('i', len(message)))
conn.send(message)
return
# 向服務程序轉發SQL語句
sockClient.send(sql)
# 數據量大小,使用sturct序列化一個整數需要4個字節
size = sockClient.recv(4)
conn.send(size)
size = struct.unpack('i', size)[0]
while True:
if size == 0:
break
elif size > 4096:
data = sockClient.recv(4096)
conn.send(data)
size -= len(data)
else:
data = sockClient.recv(size)
conn.send(data)
size -= len(data)
sockClient.close()
conn.close()
while True:
conn, _ = sockServer.accept()
Thread(target=agent, args=(conn,)).start()
模擬客戶端程序:
##
# 模擬客戶端,向SQLite代理服務器發送指令並接收數據
#
import sqlite3
import socket
import struct
while True:
sql = input('輸入一個要執行的SQL語句:\n')
# 沒有輸入,進入下一次循環
if sql.strip() == '':
continue
# 輸入exit或quit,退出客戶端
if sql in ('exit', 'quit'):
break
# 建立socket,嘗試連接
sockClient = socket.socket()
try:
sockClient.connect(('10.2.1.3', 5050))
except:
print('服務器異常,請檢查')
else:
# 發送遠程SQL語句
sockClient.send(sql.encode('gbk'))
size = sockClient.recv(4)
size = struct.unpack('i', size)[0]
data = b''
while True:
if size == 0:
break
elif size > 4096:
# 注意斷包和粘包
# 雖然設置了4096,但是不一定能夠接收4096字節
#即使緩衝區的數據遠多於4096
t = sockClient.recv(4096)
data += t
size -= len(t)
else:
t = sockClient.recv(size)
data += t
size -= len(t)
data = data.decode('gbk')
try:
data = eval(data)
except:
pass
sockClient.close()
print(data)
非常感謝你的閱讀
大學的時候選擇了自學python,工作了發現吃了計算機基礎不好的虧,學歷不行這是沒辦法的事,只能後天彌補,於是在編碼之外開啓了自己的逆襲之路,不斷的學習python核心知識,深入的研習計算機基礎知識,整理好了,我放在我們的微信公衆號《程序員學府》,如果你也不甘平庸,那就與我一起在編碼之外,不斷成長吧!
其實這裏不僅有技術,更有那些技術之外的東西,比如,如何做一個精緻的程序員,而不是“屌絲”,程序員本身就是高貴的一種存在啊,難道不是嗎?[點擊加入]想做你自己想成爲高尚人,加油!