代码拉取完成,页面将自动刷新
package impl
import (
"context"
"fmt"
"gitee.com/talk-about-articles/keyauth-tz/apps/token"
"github.com/infraboard/mcube/exception"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
func (s *service) save(ctx context.Context, ins *token.Token) error {
// insertOne 和 insertMany 会根据定义模型的bson标签完成对应的文档类型入库,主要是bson标签
if _, err := s.col.InsertOne(ctx, ins); err != nil {
return exception.NewInternalServerError("inserted token(%s) document error, %s",
ins.AccessToken, err)
}
return nil
}
// GET,Describe
// filter 过滤器(Collection),类似于mysql的where条件
// 过滤器内容就是按照文档中的任意一个键值对作为过滤条件 完成过滤查询
// 调用Decode方法来进行反序列化 bytes --> Object (查询出来数据的反序列化)
func (s *service) get(ctx context.Context, accessToken string) (*token.Token, error) {
// 以主键的键值对为条件过滤 就是在json串中匹配 "_id: 1" 的json串
filter := bson.M{"_id": accessToken}
ins := token.NewDefaultToken()
// 通过Decode反序列化为Object对象
if err := s.col.FindOne(ctx, filter).Decode(ins); err != nil {
if err == mongo.ErrNoDocuments {
return nil, exception.NewNotFound("token %s not found", accessToken)
}
return nil, exception.NewInternalServerError("find token %s error, %s", accessToken, err)
}
return ins, nil
}
func newQueryTokenRequest(r *token.QueryTokenRequest) *queryTokenRequest {
return &queryTokenRequest{
r,
}
}
type queryTokenRequest struct {
*token.QueryTokenRequest
}
func (r *queryTokenRequest) FindOptions() *options.FindOptions {
pageSize := int64(r.Page.PageSize)
skip := int64(r.Page.PageSize) * int64(r.Page.PageNumber-1)
opt := &options.FindOptions{
// 排序: Order By create_at Desc
Sort: bson.D{
{Key: "create_at", Value: -1},
},
Limit: &pageSize,
Skip: &skip,
}
return opt
}
// 过滤条件
// 由于Mongodb支持嵌套,JSON,如何过滤嵌套里面的条件,使用.访问嵌套对象属性
func (r *queryTokenRequest) FindFilter() bson.M {
filter := bson.M{}
if r.Keywords != "" {
filter["$or"] = bson.A{
bson.M{"data.name": bson.M{"$regex": r.Keywords, "$options": "im"}},
bson.M{"data.author": bson.M{"$regex": r.Keywords, "$options": "im"}},
}
}
return filter
}
func (s *service) query(ctx context.Context, req *queryTokenRequest) (*token.TokenSet, error) {
resp, err := s.col.Find(ctx, req.FindFilter(), req.FindOptions())
if err != nil {
return nil, exception.NewInternalServerError("find token error, error is %s", err)
}
set := token.NewTokenSet()
// 循环
for resp.Next(ctx) {
ins := token.NewDefaultToken()
if err := resp.Decode(ins); err != nil {
return nil, exception.NewInternalServerError("decode token error, error is %s", err)
}
set.Add(ins)
}
// count
count, err := s.col.CountDocuments(ctx, req.FindFilter())
if err != nil {
return nil, exception.NewInternalServerError("get token count error, error is %s", err)
}
set.Total = count
return set, nil
}
func (s *service) update(ctx context.Context, ins *token.Token) error {
if _, err := s.col.UpdateByID(ctx, ins.AccessToken, ins); err != nil {
return exception.NewInternalServerError("update token(%s) document error, %s",
ins.AccessToken, err)
}
return nil
}
func (s *service) deleteToken(ctx context.Context, ins *token.Token) error {
if ins == nil || ins.AccessToken == "" {
return fmt.Errorf("token is nil")
}
result, err := s.col.DeleteOne(ctx, bson.M{"_id": ins.AccessToken})
if err != nil {
return exception.NewInternalServerError("delete token(%s) error, %s", ins.AccessToken, err)
}
if result.DeletedCount == 0 {
return exception.NewNotFound("token %s not found", ins.AccessToken)
}
return nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。