style: adjust code

Author: zhufuyi

internal/database/init.go.mgo

@@ -10,7 +10,7 @@ import (
 )
 
 var (
-	mdb     *mgo.DB
+	mdb     *mgo.Database
 	mdbOnce sync.Once
 )
 
@@ -27,7 +27,7 @@ func InitDB() {
 }
 
 // GetDB get db
-func GetDB() *mgo.DB {
+func GetDB() *mgo.Database {
 	if mdb == nil {
 		mdbOnce.Do(func() {
 			InitDB()

internal/database/mongodb.go.mgo

@@ -18,7 +18,7 @@ var ErrRecordNotFound = mgo.ErrNoDocuments
 
 // InitMongodb connect mongodb
 // For more information on connecting to mongodb, see https://pkg.go.dev/go.mongodb.org/mongo-driver/mongo#Connect
-func InitMongodb() *mgo.DB {
+func InitMongodb() *mgo.Database {
 	dsn := utils.AdaptiveMongodbDsn(config.Get().Database.Mongodb.Dsn)
 	mdb, err := mgo.Init(dsn)
 	if err != nil {

pkg/gofile/filePath.go

@@ -33,11 +33,28 @@ func GetFilename(filePath string) string {
 	return name
 }
 
+// GetFileSuffixName get file suffix name, example: ".txt"
+func GetFileSuffixName(filePath string) string {
+	return filepath.Ext(filePath)
+}
+
 // GetDir get dir, not include the last separator
 func GetDir(filePath string) string {
 	return filepath.Dir(filePath)
 }
 
+// GetSuffixDir get suffix dir, not include the last separator
+func GetSuffixDir(filePath string) string {
+	fileInfo, err := os.Stat(filePath)
+	if err != nil {
+		return filepath.Base(filePath)
+	}
+	if !fileInfo.IsDir() {
+		filePath = strings.TrimSuffix(filePath, fileInfo.Name())
+	}
+	return filepath.Base(filePath)
+}
+
 // GetFileDir get dir, include the last separator
 func GetFileDir(filePath string) string {
 	dir, _ := filepath.Split(filePath)

pkg/gofile/filePath_test.go

@@ -83,9 +83,15 @@ func TestGetFilename(t *testing.T) {
 	name := GetFilename("./README.md")
 	assert.Equal(t, "README.md", name)
 
+	name = GetFileSuffixName("./README.md")
+	assert.Equal(t, ".md", name)
+
 	name = GetDir("gofile/README.md")
 	assert.Equal(t, "gofile", name)
 
+	name = GetSuffixDir("gofile/")
+	assert.Equal(t, "gofile", name)
+
 	name = GetFileDir("gofile/README.md")
 	assert.Equal(t, "gofile/", name)
 

pkg/mgo/mongo.go

@@ -12,7 +12,7 @@ import (
 	"go.uber.org/zap"
 )
 
-type DB = mongo.Database
+type Database = mongo.Database
 
 var ErrNoDocuments = mongo.ErrNoDocuments