feat: 初始提交云印享付费打印服务后端

cloudprint-backend/internal/repository/admin_repository.go

@@ -0,0 +1,49 @@
+package repository
+
+import (
+	"context"
+	"errors"
+
+	"github.com/cloudprint/cloudprint-backend/internal/model"
+	"gorm.io/gorm"
+)
+
+type AdminRepository struct {
+	db *gorm.DB
+}
+
+func NewAdminRepository(db *gorm.DB) *AdminRepository {
+	return &AdminRepository{db: db}
+}
+
+func (r *AdminRepository) Create(ctx context.Context, admin *model.Admin) error {
+	return r.db.WithContext(ctx).Create(admin).Error
+}
+
+func (r *AdminRepository) GetByID(ctx context.Context, id int32) (*model.Admin, error) {
+	var admin model.Admin
+	err := r.db.WithContext(ctx).First(&admin, id).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &admin, nil
+}
+
+func (r *AdminRepository) GetByUsername(ctx context.Context, username string) (*model.Admin, error) {
+	var admin model.Admin
+	err := r.db.WithContext(ctx).Where("username = ?", username).First(&admin).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return &admin, nil
+}
+
+func (r *AdminRepository) Update(ctx context.Context, admin *model.Admin) error {
+	return r.db.WithContext(ctx).Save(admin).Error
+}