Ver código fonte

refactor: move postgres to pg package (#1781)

Kevin Wan 3 anos atrás
pai
commit
ba8ac974aa

+ 2 - 2
core/collection/timingwheel.go

@@ -71,8 +71,8 @@ func NewTimingWheel(interval time.Duration, numSlots int, execute Execute) (*Tim
 	return newTimingWheelWithClock(interval, numSlots, execute, timex.NewTicker(interval))
 }
 
-func newTimingWheelWithClock(interval time.Duration, numSlots int, execute Execute, ticker timex.Ticker) (
-	*TimingWheel, error) {
+func newTimingWheelWithClock(interval time.Duration, numSlots int, execute Execute,
+	ticker timex.Ticker) (*TimingWheel, error) {
 	tw := &TimingWheel{
 		interval:      interval,
 		ticker:        ticker,

+ 14 - 0
core/stores/pg/postgresql.go

@@ -0,0 +1,14 @@
+package pg
+
+import (
+	// imports the driver, don't remove this comment, golint requires.
+	_ "github.com/lib/pq"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+const postgresDriverName = "postgres"
+
+// New returns a postgres connection.
+func New(datasource string, opts ...sqlx.SqlOption) sqlx.SqlConn {
+	return sqlx.NewSqlConn(postgresDriverName, datasource, opts...)
+}

+ 11 - 0
core/stores/pg/postgresql_test.go

@@ -0,0 +1,11 @@
+package pg
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPostgreSql(t *testing.T) {
+	assert.NotNil(t, New("postgre"))
+}

+ 4 - 11
core/stores/postgres/postgresql.go

@@ -1,14 +1,7 @@
 package postgres
 
-import (
-	// imports the driver, don't remove this comment, golint requires.
-	_ "github.com/lib/pq"
-	"github.com/zeromicro/go-zero/core/stores/sqlx"
-)
+import "github.com/zeromicro/go-zero/core/stores/pg"
 
-const postgresDriverName = "postgres"
-
-// New returns a postgres connection.
-func New(datasource string, opts ...sqlx.SqlOption) sqlx.SqlConn {
-	return sqlx.NewSqlConn(postgresDriverName, datasource, opts...)
-}
+// New creates a new postgresql store.
+// Deprecated: use pg.New instead.
+var New = pg.New