add dbutil for FindMulti/Single

db/chat_log.go

@@ -25,15 +25,11 @@ func (r *ChatLogRepository) Insert(l *types.ChatLog) error {
 }
 
 func (r *ChatLogRepository) Search(channel string, from, to int64) ([]*types.ChatLog, error) {
-	var list []*types.ChatLog
-	err := r.g.Where("timestamp > ?", from).Where("timestamp < ?", to).Where(types.ChatLog{Channel: channel}).Find(&list).Error
-	return list, err
+	return FindMulti[types.ChatLog](r.g.Where("timestamp > ?", from).Where("timestamp < ?", to).Where(types.ChatLog{Channel: channel}))
 }
 
 func (r *ChatLogRepository) GetLatest(channel string, limit int) ([]*types.ChatLog, error) {
-	var list []*types.ChatLog
-	err := r.g.Where(types.ChatLog{Channel: channel}).Order("timestamp ASC").Limit(limit).Find(&list).Error
-	return list, err
+	return FindMulti[types.ChatLog](r.g.Where(types.ChatLog{Channel: channel}).Order("timestamp ASC"))
 }
 
 func (r *ChatLogRepository) DeleteBefore(timestamp int64) error {

db/config.go

@@ -12,12 +12,7 @@ type ConfigRepository struct {
 }
 
 func (r *ConfigRepository) GetByKey(key types.ConfigKey) (*types.ConfigEntry, error) {
-	var list []*types.ConfigEntry
-	err := r.g.Where(types.ConfigEntry{Key: key}).Limit(1).Find(&list).Error
-	if len(list) == 0 {
-		return nil, err
-	}
-	return list[0], err
+	return FindSingle[types.ConfigEntry](r.g.Where(types.ConfigEntry{Key: key}))
 }
 
 func (r *ConfigRepository) Set(c *types.ConfigEntry) error {

db/feature.go

@@ -19,16 +19,9 @@ func (r *FeatureRepository) Set(m *types.Feature) error {
 }
 
 func (r *FeatureRepository) GetAll() ([]*types.Feature, error) {
-	var list []*types.Feature
-	err := r.g.Find(&list).Error
-	return list, err
+	return FindMulti[types.Feature](r.g)
 }
 
 func (r *FeatureRepository) GetByName(name string) (*types.Feature, error) {
-	var list []*types.Feature
-	err := r.g.Where(types.Feature{Name: name}).Limit(1).Find(&list).Error
-	if len(list) == 0 {
-		return nil, err
-	}
-	return list[0], err
+	return FindSingle[types.Feature](r.g.Where(types.Feature{Name: name}))
 }

db/log.go

@@ -109,9 +109,7 @@ func (r *LogRepository) query(s *types.LogSearch) *gorm.DB {
 }
 
 func (r *LogRepository) Search(s *types.LogSearch) ([]*types.Log, error) {
-	var list []*types.Log
-	err := r.query(s).Find(&list).Error
-	return list, err
+	return FindMulti[types.Log](r.query(s))
 }
 
 func (r *LogRepository) Count(s *types.LogSearch) (int, error) {

db/mesecons.go

@@ -19,18 +19,11 @@ func (r *MeseconsRepository) Save(m *types.Mesecons) error {
 }
 
 func (r *MeseconsRepository) GetByPlayerName(playername string) ([]*types.Mesecons, error) {
-	var list []*types.Mesecons
-	err := r.g.Where(types.Mesecons{PlayerName: playername}).Order("order_id ASC").Find(&list).Error
-	return list, err
+	return FindMulti[types.Mesecons](r.g.Where(types.Mesecons{PlayerName: playername}).Order("order_id ASC"))
 }
 
 func (r *MeseconsRepository) GetByPoskey(poskey string) (*types.Mesecons, error) {
-	var list []*types.Mesecons
-	err := r.g.Where(types.Mesecons{PosKey: poskey}).Find(&list).Error
-	if len(list) == 0 {
-		return nil, err
-	}
-	return list[0], err
+	return FindSingle[types.Mesecons](r.g.Where(types.Mesecons{PosKey: poskey}))
 }
 
 func (r *MeseconsRepository) Remove(poskey string) error {

db/mod.go

@@ -19,27 +19,15 @@ func (r *ModRepository) Create(m *types.Mod) error {
 }
 
 func (r *ModRepository) GetAll() ([]*types.Mod, error) {
-	var list []*types.Mod
-	err := r.g.Find(&list).Error
-	return list, err
+	return FindMulti[types.Mod](r.g)
 }
 
 func (r *ModRepository) GetByName(name string) (*types.Mod, error) {
-	var list []*types.Mod
-	err := r.g.Where(types.Mod{Name: name}).Limit(1).Find(&list).Error
-	if len(list) == 0 {
-		return nil, err
-	}
-	return list[0], err
+	return FindSingle[types.Mod](r.g.Where(types.Mod{Name: name}))
 }
 
 func (r *ModRepository) GetByID(id string) (*types.Mod, error) {
-	var list []*types.Mod
-	err := r.g.Where(&types.Mod{ID: id}).Limit(1).Find(&list).Error
-	if len(list) == 0 {
-		return nil, err
-	}
-	return list[0], err
+	return FindSingle[types.Mod](r.g.Where(types.Mod{ID: id}))
 }
 
 func (r *ModRepository) Update(m *types.Mod) error {

db/util.go

@@ -0,0 +1,19 @@
+package db
+
+import "gorm.io/gorm"
+
+func FindSingle[T any](g *gorm.DB) (*T, error) {
+	list := make([]*T, 0)
+	err := g.Limit(1).Find(&list).Error
+	if err != nil || len(list) == 0 {
+		return nil, err
+	} else {
+		return list[0], err
+	}
+}
+
+func FindMulti[T any](g *gorm.DB) ([]*T, error) {
+	list := make([]*T, 0)
+	err := g.Find(&list).Error
+	return list, err
+}