go17/devcloud-mini/cmdb/apps/resource/impl/resource.go

package impl

import (
	"context"

	"gitlab.com/go-course-project/go17/devcloud-mini/cmdb/apps/resource"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
)

func (s *ResourceServiceImpl) Search(ctx context.Context, in *resource.SearchRequest) (*resource.ResourceSet, error) {
	set := resource.NewResourceSet()
	filter := bson.M{}

	if in.Keywords != "" {
		filter["name"] = bson.M{"$regex": in.Keywords, "$options": "im"}
	}
	if in.Type != nil {
		filter["type"] = *in.Type
	}
	for k, v := range in.Tag {
		filter[k] = v
	}

	// mongodb query
	result, err := s.col.Find(ctx, filter, options.Find().SetLimit(in.PageSize).SetSkip(in.Skip()))
	if err != nil {
		return nil, err
	}
	for result.Next(ctx) {
		res := &resource.Resource{}
		if err := result.Decode(res); err != nil {
			return nil, err
		}
		set.Items = append(set.Items, res)
	}

	return set, nil
}

func (s *ResourceServiceImpl) Save(ctx context.Context, in *resource.Resource) (*resource.Resource, error) {
	if err := in.Validate(); err != nil {
		return nil, err
	}

	// 保持数据, 需要从ioc里面获取一个mongodb实例
	_, err := s.col.InsertOne(ctx, in)
	if err != nil {
		return nil, err
	}

	return in, nil
}

// $in
func (s *ResourceServiceImpl) DeleteResource(ctx context.Context, in *resource.DeleteResourceRequest) error {
	_, err := s.col.DeleteMany(ctx, bson.M{"_id": bson.M{"$in": in.ResourceIds}})
	return err
}