ba18887ed8
Some checks failed
Build Push and Restart Compose / deploy (push) Failing after 1m47s
- `github.com/lib/pq` PostgreSQL 드라이버 vendor 디렉토리에 추가. - PostgreSQL 관련 내부 패키지(`pqsql`, `proto`, `pqtime`, `pgpass`, `pgservice`, `pqutil`) 구현: - SQL 어휘 처리, 프로토콜 상수 및 구조 정의, 시간 파서/포맷터(`Parse`, `Format`). - `.pgpass` 파일 및 `pg_service.conf` 관리 기능 추가. - 파일/사용자 권한 검증 및 플랫폼별 사용자 정보 조회 기능 포함. - 데이터베이스 초기화 로직 추가 (`services/db.go`): - PostgreSQL 연결 설정 및 초기 스키마 생성. - 자동매매 관련 DB 레포지토리(`services/autotrade_repo.go`) 구현: - 자동매매 규칙 및 포지션 관리 로직 추가 (`dbInsertRule`, `dbLoadRules` 등).
70 lines
2.5 KiB
Go
70 lines
2.5 KiB
Go
package pq
|
|
|
|
import (
|
|
"context"
|
|
"database/sql/driver"
|
|
)
|
|
|
|
// NoticeHandler returns the notice handler on the given connection, if any. A
|
|
// runtime panic occurs if c is not a pq connection. This is rarely used
|
|
// directly, use [ConnectorNoticeHandler] and [ConnectorWithNoticeHandler] instead.
|
|
func NoticeHandler(c driver.Conn) func(*Error) {
|
|
return c.(*conn).noticeHandler
|
|
}
|
|
|
|
// SetNoticeHandler sets the given notice handler on the given connection. A
|
|
// runtime panic occurs if c is not a pq connection. A nil handler may be used
|
|
// to unset it. This is rarely used directly, use ConnectorNoticeHandler and
|
|
// [ConnectorWithNoticeHandler] instead.
|
|
//
|
|
// Note: Notice handlers are executed synchronously by pq meaning commands
|
|
// won't continue to be processed until the handler returns.
|
|
func SetNoticeHandler(c driver.Conn, handler func(*Error)) {
|
|
c.(*conn).noticeHandler = handler
|
|
}
|
|
|
|
// NoticeHandlerConnector wraps a regular connector and sets a notice handler
|
|
// on it.
|
|
type NoticeHandlerConnector struct {
|
|
driver.Connector
|
|
noticeHandler func(*Error)
|
|
}
|
|
|
|
// Connect calls the underlying connector's connect method and then sets the
|
|
// notice handler.
|
|
func (n *NoticeHandlerConnector) Connect(ctx context.Context) (driver.Conn, error) {
|
|
c, err := n.Connector.Connect(ctx)
|
|
if err == nil {
|
|
SetNoticeHandler(c, n.noticeHandler)
|
|
}
|
|
return c, err
|
|
}
|
|
|
|
// ConnectorNoticeHandler returns the currently set notice handler, if any. If
|
|
// the given connector is not a result of [ConnectorWithNoticeHandler], nil is
|
|
// returned.
|
|
func ConnectorNoticeHandler(c driver.Connector) func(*Error) {
|
|
if c, ok := c.(*NoticeHandlerConnector); ok {
|
|
return c.noticeHandler
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ConnectorWithNoticeHandler creates or sets the given handler for the given
|
|
// connector. If the given connector is a result of calling this function
|
|
// previously, it is simply set on the given connector and returned. Otherwise,
|
|
// this returns a new connector wrapping the given one and setting the notice
|
|
// handler. A nil notice handler may be used to unset it.
|
|
//
|
|
// The returned connector is intended to be used with database/sql.OpenDB.
|
|
//
|
|
// Note: Notice handlers are executed synchronously by pq meaning commands
|
|
// won't continue to be processed until the handler returns.
|
|
func ConnectorWithNoticeHandler(c driver.Connector, handler func(*Error)) *NoticeHandlerConnector {
|
|
if c, ok := c.(*NoticeHandlerConnector); ok {
|
|
c.noticeHandler = handler
|
|
return c
|
|
}
|
|
return &NoticeHandlerConnector{Connector: c, noticeHandler: handler}
|
|
}
|