github.com/e154/smart-home@v0.17.2-0.20240311175135-e530a6e5cd45/db/usermeta.go (about) 1 // This file is part of the Smart Home 2 // Program complex distribution https://github.com/e154/smart-home 3 // Copyright (C) 2016-2023, Filippov Alex 4 // 5 // This library is free software: you can redistribute it and/or 6 // modify it under the terms of the GNU Lesser General Public 7 // License as published by the Free Software Foundation; either 8 // version 3 of the License, or (at your option) any later version. 9 // 10 // This library is distributed in the hope that it will be useful, 11 // but WITHOUT ANY WARRANTY; without even the implied warranty of 12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 13 // Library General Public License for more details. 14 // 15 // You should have received a copy of the GNU Lesser General Public 16 // License along with this library. If not, see 17 // <https://www.gnu.org/licenses/>. 18 19 package db 20 21 import ( 22 "context" 23 24 "github.com/e154/smart-home/common/apperr" 25 "github.com/pkg/errors" 26 "gorm.io/gorm" 27 ) 28 29 // UserMetas ... 30 type UserMetas struct { 31 Db *gorm.DB 32 } 33 34 // UserMeta ... 35 type UserMeta struct { 36 Id int64 `gorm:"primary_key"` 37 UserId int64 38 Key string 39 Value string 40 } 41 42 // TableName ... 43 func (m *UserMeta) TableName() string { 44 return "user_metas" 45 } 46 47 // UpdateOrCreate ... 48 func (m *UserMetas) UpdateOrCreate(ctx context.Context, meta *UserMeta) (id int64, err error) { 49 50 err = m.Db.WithContext(ctx).Model(&UserMeta{}). 51 Where("user_id = ? and key = ?", meta.UserId, meta.Key). 52 Updates(map[string]interface{}{"value": meta.Value}). 53 Error 54 55 if err != nil { 56 if err = m.Db.WithContext(ctx).Create(&meta).Error; err != nil { 57 err = errors.Wrap(apperr.ErrUserMetaAdd, err.Error()) 58 return 59 } 60 id = meta.Id 61 } 62 63 return 64 }