Как использовать "INSERT" в пуле соединений psycopg2?

Я использую psycopg2 для подключения к PostgreSQL на Python, и я хочу использовать пул соединений.

Я не знаю, что мне делать вместо commit() и rollback () при выполнении запроса INSERT.

db = pool.SimpleConnectionPool(1, 10,host=conf_hostname,database=conf_dbname,user=conf_dbuser,password=conf_dbpass,port=conf_dbport)


# Get Cursor
@contextmanager
def get_cursor():
    con = db.getconn()
    try:
        yield con.cursor()
    finally:
        db.putconn(con)


with get_cursor() as cursor:
    cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
    id = cursor.fetchone()

Я не получаю идентификатор вставленной записи без commit ().

7
connection-pooling psycopg psycopg2 python

2 ответов

обновление Я не могу проверить код, но я дать вам некоторые идеи: Вы делаете фиксацию в связи не в db

# Get Cursor
@contextmanager
def get_cursor():
    con = db.getconn()
    try:
        yield con
    finally:
        db.putconn(con)

with get_cursor() as cursor:
    con.cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
    con.commit()
    id = cursor.fetchone()

или

# Get Cursor
@contextmanager
def get_cursor():
    con = db.getconn()
    try:
        yield con.cursor()
        con.commit()
    finally:
        db.putconn(con)


with get_cursor() as cursor:
    con.cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
    id = cursor.fetchone()

пул соединений существует, потому что создание нового соединения с БД может быть дорогостоящим и не избегать коммитов или откатов. Таким образом, вы можете зафиксировать свои данные без каких-либо проблем, фиксация данных не уничтожит соединение.

5

вот мой рабочий пример: 

db = pool.SimpleConnectionPool(1, 10,host=conf_hostname,database=conf_dbname,user=conf_dbuser,password=conf_dbpass,port=conf_dbport)


@contextmanager
def get_connection():
    con = db.getconn()
    try:
        yield con
    finally:
        db.putconn(con)

def write_to_db():
    with get_connection() as conn:
        try:
            cursor = conn.cursor()
            cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
            id = cursor.fetchone()
            cursor.close()
            conn.commit()
        except:
            conn.rollback()
3