commit code

gallery/internal/model/imagesmodel.go

@@ -0,0 +1,29 @@
+package model
+
+import "github.com/zeromicro/go-zero/core/stores/sqlx"
+
+var _ ImagesModel = (*customImagesModel)(nil)
+
+type (
+	// ImagesModel is an interface to be customized, add more methods here,
+	// and implement the added methods in customImagesModel.
+	ImagesModel interface {
+		imagesModel
+		withSession(session sqlx.Session) ImagesModel
+	}
+
+	customImagesModel struct {
+		*defaultImagesModel
+	}
+)
+
+// NewImagesModel returns a model for the database table.
+func NewImagesModel(conn sqlx.SqlConn) ImagesModel {
+	return &customImagesModel{
+		defaultImagesModel: newImagesModel(conn),
+	}
+}
+
+func (m *customImagesModel) withSession(session sqlx.Session) ImagesModel {
+	return NewImagesModel(sqlx.NewSqlConnFromSession(session))
+}