mysql.go

// Copyright 2022 Teamgram Authors
//  All rights reserved.
//
// Licensed 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.
//
// Author: teamgramio (teamgram.io@gmail.com)
//

package dao

import (
	"context"
	"github.com/teamgram/marmota/pkg/stores/sqlx"
	"github.com/teamgram/teamgram-server/app/service/biz/user/internal/dal/dao/mysql_dao"
)

type Mysql struct {
	*sqlx.DB
	*mysql_dao.BotCommandsDAO
	*mysql_dao.BotsDAO
	*mysql_dao.ImportedContactsDAO
	*mysql_dao.PhoneBooksDAO
	*mysql_dao.PopularContactsDAO
	*mysql_dao.PredefinedUsersDAO
	*mysql_dao.UserContactsDAO
	*mysql_dao.UserGlobalPrivacySettingsDAO
	*mysql_dao.UserNotifySettingsDAO
	*mysql_dao.UserPeerBlocksDAO
	*mysql_dao.UserPeerSettingsDAO
	*mysql_dao.UserPresencesDAO
	*mysql_dao.UserPrivaciesDAO
	*mysql_dao.UserSettingsDAO
	*mysql_dao.UsersDAO
	*mysql_dao.UserProfilePhotosDAO
	*mysql_dao.UnregisteredContactsDAO
	*sqlx.CommonDAO
}

func newMysqlDao(db *sqlx.DB) *Mysql {
	return &Mysql{
		DB:                           db,
		BotCommandsDAO:               mysql_dao.NewBotCommandsDAO(db),
		BotsDAO:                      mysql_dao.NewBotsDAO(db),
		ImportedContactsDAO:          mysql_dao.NewImportedContactsDAO(db),
		PhoneBooksDAO:                mysql_dao.NewPhoneBooksDAO(db),
		PopularContactsDAO:           mysql_dao.NewPopularContactsDAO(db),
		PredefinedUsersDAO:           mysql_dao.NewPredefinedUsersDAO(db),
		UserContactsDAO:              mysql_dao.NewUserContactsDAO(db),
		UserGlobalPrivacySettingsDAO: mysql_dao.NewUserGlobalPrivacySettingsDAO(db),
		UserNotifySettingsDAO:        mysql_dao.NewUserNotifySettingsDAO(db),
		UserPeerBlocksDAO:            mysql_dao.NewUserPeerBlocksDAO(db),
		UserPeerSettingsDAO:          mysql_dao.NewUserPeerSettingsDAO(db),
		UserPresencesDAO:             mysql_dao.NewUserPresencesDAO(db),
		UserPrivaciesDAO:             mysql_dao.NewUserPrivaciesDAO(db),
		UserSettingsDAO:              mysql_dao.NewUserSettingsDAO(db),
		UsersDAO:                     mysql_dao.NewUsersDAO(db),
		UserProfilePhotosDAO:         mysql_dao.NewUserProfilePhotosDAO(db),
		UnregisteredContactsDAO:      mysql_dao.NewUnregisteredContactsDAO(db),
		CommonDAO:                    sqlx.NewCommonDAO(db),
	}
}

func (d *Mysql) Close() error {
	return d.DB.Close()
}

func (d *Mysql) Ping(ctx context.Context) (err error) {
	return d.DB.Ping(ctx)
}