chore: import upstream snapshot with attribution

internal/repo/user_notification_config/user_notification_config_repo.go

@@ -0,0 +1,128 @@
+/*
+ * 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 user_notification_config
+
+import (
+	"context"
+
+	"github.com/apache/answer/internal/base/constant"
+	"github.com/apache/answer/internal/base/data"
+	"github.com/apache/answer/internal/base/reason"
+	"github.com/apache/answer/internal/entity"
+	"github.com/apache/answer/internal/service/user_notification_config"
+	"github.com/segmentfault/pacman/errors"
+)
+
+// userNotificationConfigRepo notification repository
+type userNotificationConfigRepo struct {
+	data *data.Data
+}
+
+// NewUserNotificationConfigRepo new repository
+func NewUserNotificationConfigRepo(data *data.Data) user_notification_config.UserNotificationConfigRepo {
+	return &userNotificationConfigRepo{
+		data: data,
+	}
+}
+
+// Add add notification config
+func (ur *userNotificationConfigRepo) Add(ctx context.Context, userIDs []string, source, channels string) (err error) {
+	var configs []*entity.UserNotificationConfig
+	for _, userID := range userIDs {
+		configs = append(configs, &entity.UserNotificationConfig{
+			UserID:   userID,
+			Source:   source,
+			Channels: channels,
+			Enabled:  true,
+		})
+	}
+	_, err = ur.data.DB.Context(ctx).Insert(configs)
+	if err != nil {
+		return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return nil
+}
+
+// Save save notification config, if existed, update, if not exist, insert
+func (ur *userNotificationConfigRepo) Save(ctx context.Context, uc *entity.UserNotificationConfig) (err error) {
+	old := &entity.UserNotificationConfig{UserID: uc.UserID, Source: uc.Source}
+	exist, err := ur.data.DB.Context(ctx).Get(old)
+	if err != nil {
+		return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	if exist {
+		old.Channels = uc.Channels
+		old.Enabled = uc.Enabled
+		_, err = ur.data.DB.Context(ctx).ID(old.ID).UseBool("enabled").Cols("channels", "enabled").Update(old)
+	} else {
+		_, err = ur.data.DB.Context(ctx).Insert(uc)
+	}
+	if err != nil {
+		return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return nil
+}
+
+// GetByUserID get notification config by user id
+func (ur *userNotificationConfigRepo) GetByUserID(ctx context.Context, userID string) (
+	[]*entity.UserNotificationConfig, error) {
+	var configs []*entity.UserNotificationConfig
+	err := ur.data.DB.Context(ctx).Where("user_id = ?", userID).Find(&configs)
+	if err != nil {
+		return nil, errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return configs, nil
+}
+
+// GetBySource get notification config by source
+func (ur *userNotificationConfigRepo) GetBySource(ctx context.Context, source constant.NotificationSource) (
+	[]*entity.UserNotificationConfig, error) {
+	var configs []*entity.UserNotificationConfig
+	err := ur.data.DB.Context(ctx).UseBool("enabled").
+		Find(&configs, &entity.UserNotificationConfig{Source: string(source), Enabled: true})
+	if err != nil {
+		return nil, errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return configs, nil
+}
+
+// GetByUserIDAndSource get notification config by user id and source
+func (ur *userNotificationConfigRepo) GetByUserIDAndSource(ctx context.Context, userID string, source constant.NotificationSource) (
+	conf *entity.UserNotificationConfig, exist bool, err error) {
+	config := &entity.UserNotificationConfig{UserID: userID, Source: string(source)}
+	exist, err = ur.data.DB.Context(ctx).Get(config)
+	if err != nil {
+		return nil, false, errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return config, exist, nil
+}
+
+// GetByUsersAndSource get notification config by user ids and source
+func (ur *userNotificationConfigRepo) GetByUsersAndSource(
+	ctx context.Context, userIDs []string, source constant.NotificationSource) (
+	[]*entity.UserNotificationConfig, error) {
+	var configs []*entity.UserNotificationConfig
+	err := ur.data.DB.Context(ctx).UseBool("enabled").In("user_id", userIDs).
+		Find(&configs, &entity.UserNotificationConfig{Source: string(source), Enabled: true})
+	if err != nil {
+		return nil, errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
+	}
+	return configs, nil
+}