wehub-resource-sync
87 lines
2.7 KiB
Go
87 lines
2.7 KiB
Go
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
|
|
package collection
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/apache/answer/internal/entity"
|
|
"github.com/apache/answer/internal/schema"
|
|
collectioncommon "github.com/apache/answer/internal/service/collection_common"
|
|
questioncommon "github.com/apache/answer/internal/service/question_common"
|
|
)
|
|
|
|
// CollectionService user service
|
|
type CollectionService struct {
|
|
collectionRepo collectioncommon.CollectionRepo
|
|
collectionGroupRepo CollectionGroupRepo
|
|
questionCommon *questioncommon.QuestionCommon
|
|
}
|
|
|
|
func NewCollectionService(
|
|
collectionRepo collectioncommon.CollectionRepo,
|
|
collectionGroupRepo CollectionGroupRepo,
|
|
questionCommon *questioncommon.QuestionCommon,
|
|
) *CollectionService {
|
|
return &CollectionService{
|
|
collectionRepo: collectionRepo,
|
|
collectionGroupRepo: collectionGroupRepo,
|
|
questionCommon: questionCommon,
|
|
}
|
|
}
|
|
|
|
func (cs *CollectionService) CollectionSwitch(ctx context.Context, req *schema.CollectionSwitchReq) (
|
|
resp *schema.CollectionSwitchResp, err error) {
|
|
collectionGroup, err := cs.collectionGroupRepo.CreateDefaultGroupIfNotExist(ctx, req.UserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
collection, exist, err := cs.collectionRepo.GetOneByObjectIDAndUser(ctx, req.UserID, req.ObjectID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if (!req.Bookmark && !exist) || (req.Bookmark && exist) {
|
|
return nil, nil
|
|
}
|
|
|
|
if req.Bookmark {
|
|
collection = &entity.Collection{
|
|
UserID: req.UserID,
|
|
ObjectID: req.ObjectID,
|
|
UserCollectionGroupID: collectionGroup.ID,
|
|
}
|
|
err = cs.collectionRepo.AddCollection(ctx, collection)
|
|
} else {
|
|
err = cs.collectionRepo.RemoveCollection(ctx, collection.ID)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// For now, we only support bookmark for question, so we just update question collection count
|
|
resp = &schema.CollectionSwitchResp{}
|
|
resp.ObjectCollectionCount, err = cs.questionCommon.UpdateCollectionCount(ctx, req.ObjectID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return resp, nil
|
|
}
|