box-cost/boxcost/api/utils.go

package api

import (
	"box-cost/db/model"
	"box-cost/db/repo"
	"fmt"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

func incrementer(ctx *ApiSession, typeName string) (serial string, err error) {
	// 获取类型
	cate := &model.Category{}
	found, err := repo.RepoSeachDoc(ctx.CreateRepoCtx(), &repo.DocSearchOptions{
		CollectName: "cates",
		Project:     []string{"letterName"},
		Query:       repo.Map{"name": typeName},
		Sort:        bson.M{"_id": -1},
	}, cate)
	if !found || err != nil {
		return "", fmt.Errorf("未找到该分类")
	}
	// increment index加1
	increment := &model.Increment{}
	repo.RepoSeachDoc(ctx.CreateRepoCtx(), &repo.DocSearchOptions{
		CollectName: repo.CollectionIncrement,
		Project:     []string{"index"},
		Sort:        bson.M{"_id": -1},
	}, increment)
	index := increment.Index + 1
	repo.RepoAddDoc(ctx.CreateRepoCtx(), repo.CollectionIncrement, &model.Increment{Index: index})

	// 拼接为序号
	return fmt.Sprintf("%s-%05d", cate.LetterName, index), nil

}

func SearchBillTypeById(ctx *ApiSession, collectName string, id primitive.ObjectID) (string, error) {
	found, curbill := repo.RepoSeachDocMap(ctx.CreateRepoCtx(), &repo.DocSearchOptions{
		CollectName: collectName,
		Project:     []string{"type"},
		Query:       repo.Map{"_id": id},
		Sort:        bson.M{"_id": -1},
	})
	if !found {
		return "", fmt.Errorf("未找到该分类")
	}
	return curbill["type"].(string), nil
}