feat: geo ip.

bank_card/assets/assets.go

@@ -0,0 +1,6 @@
+package assets
+
+import _ "embed"
+
+//go:embed bank_card.db
+var BankCardDatabase []byte

bank_card/database.go

@@ -3,17 +3,15 @@ package bank_card
 import (
 	"context"
 	"database/sql"
-	_ "embed"
 	"errors"
 	"fmt"
 	"log"
 
 	"github.com/mattn/go-sqlite3"
 	_ "github.com/mattn/go-sqlite3"
-)
 
-//go:embed bank_card.db
-var embeddedDatabase []byte
+	"github.com/tx7do/kratos-utils/bank_card/assets"
+)
 
 type Database struct {
 	db *sql.DB
@@ -33,7 +31,7 @@ func NewDatabase(openFile bool) *Database {
 
 // openFromFile 从文件打开数据库
 func (d *Database) openFromFile() {
-	db, err := sql.Open("sqlite3", "bank_card.db")
+	db, err := sql.Open("sqlite3", "assets/bank_card.db")
 	if err != nil {
 		panic("failed to connect database")
 	}
@@ -59,7 +57,7 @@ func (d *Database) openFromEmbed() error {
 	defer conn.Close()
 
 	if err = conn.Raw(func(raw interface{}) error {
-		return raw.(*sqlite3.SQLiteConn).Deserialize(embeddedDatabase, "")
+		return raw.(*sqlite3.SQLiteConn).Deserialize(assets.BankCardDatabase, "")
 	}); err != nil {
 		log.Fatal(err)
 		return err

bank_card/database_test.go

@@ -18,7 +18,7 @@ func TestImportBankName(t *testing.T) {
 
 	//db.openFromFile()
 
-	file, err := os.Open("name.csv")
+	file, err := os.Open("assets/name.csv")
 	if err != nil {
 		fmt.Println("Read file err, err =", err)
 		return
@@ -65,7 +65,7 @@ func TestImportBankCard(t *testing.T) {
 
 	//db.openFromFile()
 
-	file, err := os.Open("bin.csv")
+	file, err := os.Open("assets/bin.csv")
 	if err != nil {
 		fmt.Println("Read file err, err =", err)
 		return