summaryrefslogtreecommitdiffstats
path: root/models/models.go
diff options
context:
space:
mode:
authorLunny Xiao <xiaolunwen@gmail.com>2021-09-19 19:49:59 +0800
committerGitHub <noreply@github.com>2021-09-19 19:49:59 +0800
commita4bfef265d9e512830350635a0489c2cdcd6508f (patch)
tree1e3c2ec94276dfcb2f8ba73a2ac075ba39c4a34a /models/models.go
parent462306e263db5a809dbe2cdf62e99307aeff28de (diff)
downloadgitea-a4bfef265d9e512830350635a0489c2cdcd6508f.tar.gz
gitea-a4bfef265d9e512830350635a0489c2cdcd6508f.zip
Move db related basic functions to models/db (#17075)
* Move db related basic functions to models/db * Fix lint * Fix lint * Fix test * Fix lint * Fix lint * revert unnecessary change * Fix test * Fix wrong replace string * Use *Context * Correct committer spelling and fix wrong replaced words Co-authored-by: zeripath <art27@cantab.net>
Diffstat (limited to 'models/models.go')
-rwxr-xr-xmodels/models.go408
1 files changed, 0 insertions, 408 deletions
diff --git a/models/models.go b/models/models.go
deleted file mode 100755
index 4e1448241a..0000000000
--- a/models/models.go
+++ /dev/null
@@ -1,408 +0,0 @@
-// Copyright 2014 The Gogs Authors. All rights reserved.
-// Copyright 2018 The Gitea Authors. All rights reserved.
-// Use of this source code is governed by a MIT-style
-// license that can be found in the LICENSE file.
-
-package models
-
-import (
- "context"
- "database/sql"
- "errors"
- "fmt"
- "reflect"
- "strings"
-
- "code.gitea.io/gitea/modules/setting"
-
- // Needed for the MySQL driver
- _ "github.com/go-sql-driver/mysql"
- lru "github.com/hashicorp/golang-lru"
- "xorm.io/xorm"
- "xorm.io/xorm/names"
- "xorm.io/xorm/schemas"
-
- // Needed for the Postgresql driver
- _ "github.com/lib/pq"
-
- // Needed for the MSSQL driver
- _ "github.com/denisenkom/go-mssqldb"
-)
-
-// Engine represents a xorm engine or session.
-type Engine interface {
- Table(tableNameOrBean interface{}) *xorm.Session
- Count(...interface{}) (int64, error)
- Decr(column string, arg ...interface{}) *xorm.Session
- Delete(...interface{}) (int64, error)
- Exec(...interface{}) (sql.Result, error)
- Find(interface{}, ...interface{}) error
- Get(interface{}) (bool, error)
- ID(interface{}) *xorm.Session
- In(string, ...interface{}) *xorm.Session
- Incr(column string, arg ...interface{}) *xorm.Session
- Insert(...interface{}) (int64, error)
- InsertOne(interface{}) (int64, error)
- Iterate(interface{}, xorm.IterFunc) error
- Join(joinOperator string, tablename interface{}, condition string, args ...interface{}) *xorm.Session
- SQL(interface{}, ...interface{}) *xorm.Session
- Where(interface{}, ...interface{}) *xorm.Session
- Asc(colNames ...string) *xorm.Session
- Desc(colNames ...string) *xorm.Session
- Limit(limit int, start ...int) *xorm.Session
- SumInt(bean interface{}, columnName string) (res int64, err error)
-}
-
-const (
- // When queries are broken down in parts because of the number
- // of parameters, attempt to break by this amount
- maxQueryParameters = 300
-)
-
-var (
- x *xorm.Engine
- tables []interface{}
-
- // HasEngine specifies if we have a xorm.Engine
- HasEngine bool
-)
-
-func init() {
- tables = append(tables,
- new(User),
- new(PublicKey),
- new(AccessToken),
- new(Repository),
- new(DeployKey),
- new(Collaboration),
- new(Access),
- new(Upload),
- new(Watch),
- new(Star),
- new(Follow),
- new(Action),
- new(Issue),
- new(PullRequest),
- new(Comment),
- new(Attachment),
- new(Label),
- new(IssueLabel),
- new(Milestone),
- new(Mirror),
- new(Release),
- new(LoginSource),
- new(Webhook),
- new(HookTask),
- new(Team),
- new(OrgUser),
- new(TeamUser),
- new(TeamRepo),
- new(Notice),
- new(EmailAddress),
- new(Notification),
- new(IssueUser),
- new(LFSMetaObject),
- new(TwoFactor),
- new(GPGKey),
- new(GPGKeyImport),
- new(RepoUnit),
- new(RepoRedirect),
- new(ExternalLoginUser),
- new(ProtectedBranch),
- new(UserOpenID),
- new(IssueWatch),
- new(CommitStatus),
- new(Stopwatch),
- new(TrackedTime),
- new(DeletedBranch),
- new(RepoIndexerStatus),
- new(IssueDependency),
- new(LFSLock),
- new(Reaction),
- new(IssueAssignees),
- new(U2FRegistration),
- new(TeamUnit),
- new(Review),
- new(OAuth2Application),
- new(OAuth2AuthorizationCode),
- new(OAuth2Grant),
- new(Task),
- new(LanguageStat),
- new(EmailHash),
- new(UserRedirect),
- new(Project),
- new(ProjectBoard),
- new(ProjectIssue),
- new(Session),
- new(RepoTransfer),
- new(IssueIndex),
- new(PushMirror),
- new(RepoArchiver),
- new(ProtectedTag),
- )
-
- gonicNames := []string{"SSL", "UID"}
- for _, name := range gonicNames {
- names.LintGonicMapper[name] = true
- }
-}
-
-// GetNewEngine returns a new xorm engine from the configuration
-func GetNewEngine() (*xorm.Engine, error) {
- connStr, err := setting.DBConnStr()
- if err != nil {
- return nil, err
- }
-
- var engine *xorm.Engine
-
- if setting.Database.UsePostgreSQL && len(setting.Database.Schema) > 0 {
- // OK whilst we sort out our schema issues - create a schema aware postgres
- registerPostgresSchemaDriver()
- engine, err = xorm.NewEngine("postgresschema", connStr)
- } else {
- engine, err = xorm.NewEngine(setting.Database.Type, connStr)
- }
-
- if err != nil {
- return nil, err
- }
- if setting.Database.Type == "mysql" {
- engine.Dialect().SetParams(map[string]string{"rowFormat": "DYNAMIC"})
- } else if setting.Database.Type == "mssql" {
- engine.Dialect().SetParams(map[string]string{"DEFAULT_VARCHAR": "nvarchar"})
- }
- engine.SetSchema(setting.Database.Schema)
- return engine, nil
-}
-
-func syncTables() error {
- return x.StoreEngine("InnoDB").Sync2(tables...)
-}
-
-// NewTestEngine sets a new test xorm.Engine
-func NewTestEngine() (err error) {
- x, err = GetNewEngine()
- if err != nil {
- return fmt.Errorf("Connect to database: %v", err)
- }
-
- x.SetMapper(names.GonicMapper{})
- x.SetLogger(NewXORMLogger(!setting.IsProd()))
- x.ShowSQL(!setting.IsProd())
- return syncTables()
-}
-
-// SetEngine sets the xorm.Engine
-func SetEngine() (err error) {
- x, err = GetNewEngine()
- if err != nil {
- return fmt.Errorf("Failed to connect to database: %v", err)
- }
-
- x.SetMapper(names.GonicMapper{})
- // WARNING: for serv command, MUST remove the output to os.stdout,
- // so use log file to instead print to stdout.
- x.SetLogger(NewXORMLogger(setting.Database.LogSQL))
- x.ShowSQL(setting.Database.LogSQL)
- x.SetMaxOpenConns(setting.Database.MaxOpenConns)
- x.SetMaxIdleConns(setting.Database.MaxIdleConns)
- x.SetConnMaxLifetime(setting.Database.ConnMaxLifetime)
- return nil
-}
-
-// NewEngine initializes a new xorm.Engine
-// This function must never call .Sync2() if the provided migration function fails.
-// When called from the "doctor" command, the migration function is a version check
-// that prevents the doctor from fixing anything in the database if the migration level
-// is different from the expected value.
-func NewEngine(ctx context.Context, migrateFunc func(*xorm.Engine) error) (err error) {
- if err = SetEngine(); err != nil {
- return err
- }
-
- x.SetDefaultContext(ctx)
-
- if err = x.Ping(); err != nil {
- return err
- }
-
- if err = migrateFunc(x); err != nil {
- return fmt.Errorf("migrate: %v", err)
- }
-
- if err = syncTables(); err != nil {
- return fmt.Errorf("sync database struct error: %v", err)
- }
-
- if setting.SuccessfulTokensCacheSize > 0 {
- successfulAccessTokenCache, err = lru.New(setting.SuccessfulTokensCacheSize)
- if err != nil {
- return fmt.Errorf("unable to allocate AccessToken cache: %v", err)
- }
- } else {
- successfulAccessTokenCache = nil
- }
-
- return nil
-}
-
-// NamesToBean return a list of beans or an error
-func NamesToBean(names ...string) ([]interface{}, error) {
- beans := []interface{}{}
- if len(names) == 0 {
- beans = append(beans, tables...)
- return beans, nil
- }
- // Need to map provided names to beans...
- beanMap := make(map[string]interface{})
- for _, bean := range tables {
-
- beanMap[strings.ToLower(reflect.Indirect(reflect.ValueOf(bean)).Type().Name())] = bean
- beanMap[strings.ToLower(x.TableName(bean))] = bean
- beanMap[strings.ToLower(x.TableName(bean, true))] = bean
- }
-
- gotBean := make(map[interface{}]bool)
- for _, name := range names {
- bean, ok := beanMap[strings.ToLower(strings.TrimSpace(name))]
- if !ok {
- return nil, fmt.Errorf("No table found that matches: %s", name)
- }
- if !gotBean[bean] {
- beans = append(beans, bean)
- gotBean[bean] = true
- }
- }
- return beans, nil
-}
-
-// Statistic contains the database statistics
-type Statistic struct {
- Counter struct {
- User, Org, PublicKey,
- Repo, Watch, Star, Action, Access,
- Issue, IssueClosed, IssueOpen,
- Comment, Oauth, Follow,
- Mirror, Release, LoginSource, Webhook,
- Milestone, Label, HookTask,
- Team, UpdateTask, Attachment int64
- }
-}
-
-// GetStatistic returns the database statistics
-func GetStatistic() (stats Statistic) {
- stats.Counter.User = CountUsers()
- stats.Counter.Org = CountOrganizations()
- stats.Counter.PublicKey, _ = x.Count(new(PublicKey))
- stats.Counter.Repo = CountRepositories(true)
- stats.Counter.Watch, _ = x.Count(new(Watch))
- stats.Counter.Star, _ = x.Count(new(Star))
- stats.Counter.Action, _ = x.Count(new(Action))
- stats.Counter.Access, _ = x.Count(new(Access))
-
- type IssueCount struct {
- Count int64
- IsClosed bool
- }
- issueCounts := []IssueCount{}
-
- _ = x.Select("COUNT(*) AS count, is_closed").Table("issue").GroupBy("is_closed").Find(&issueCounts)
- for _, c := range issueCounts {
- if c.IsClosed {
- stats.Counter.IssueClosed = c.Count
- } else {
- stats.Counter.IssueOpen = c.Count
- }
- }
-
- stats.Counter.Issue = stats.Counter.IssueClosed + stats.Counter.IssueOpen
-
- stats.Counter.Comment, _ = x.Count(new(Comment))
- stats.Counter.Oauth = 0
- stats.Counter.Follow, _ = x.Count(new(Follow))
- stats.Counter.Mirror, _ = x.Count(new(Mirror))
- stats.Counter.Release, _ = x.Count(new(Release))
- stats.Counter.LoginSource = CountLoginSources()
- stats.Counter.Webhook, _ = x.Count(new(Webhook))
- stats.Counter.Milestone, _ = x.Count(new(Milestone))
- stats.Counter.Label, _ = x.Count(new(Label))
- stats.Counter.HookTask, _ = x.Count(new(HookTask))
- stats.Counter.Team, _ = x.Count(new(Team))
- stats.Counter.Attachment, _ = x.Count(new(Attachment))
- return
-}
-
-// Ping tests if database is alive
-func Ping() error {
- if x != nil {
- return x.Ping()
- }
- return errors.New("database not configured")
-}
-
-// DumpDatabase dumps all data from database according the special database SQL syntax to file system.
-func DumpDatabase(filePath, dbType string) error {
- var tbs []*schemas.Table
- for _, t := range tables {
- t, err := x.TableInfo(t)
- if err != nil {
- return err
- }
- tbs = append(tbs, t)
- }
-
- type Version struct {
- ID int64 `xorm:"pk autoincr"`
- Version int64
- }
- t, err := x.TableInfo(&Version{})
- if err != nil {
- return err
- }
- tbs = append(tbs, t)
-
- if len(dbType) > 0 {
- return x.DumpTablesToFile(tbs, filePath, schemas.DBType(dbType))
- }
- return x.DumpTablesToFile(tbs, filePath)
-}
-
-// MaxBatchInsertSize returns the table's max batch insert size
-func MaxBatchInsertSize(bean interface{}) int {
- t, err := x.TableInfo(bean)
- if err != nil {
- return 50
- }
- return 999 / len(t.ColumnsSeq())
-}
-
-// Count returns records number according struct's fields as database query conditions
-func Count(bean interface{}) (int64, error) {
- return x.Count(bean)
-}
-
-// IsTableNotEmpty returns true if table has at least one record
-func IsTableNotEmpty(tableName string) (bool, error) {
- return x.Table(tableName).Exist()
-}
-
-// DeleteAllRecords will delete all the records of this table
-func DeleteAllRecords(tableName string) error {
- _, err := x.Exec(fmt.Sprintf("DELETE FROM %s", tableName))
- return err
-}
-
-// GetMaxID will return max id of the table
-func GetMaxID(beanOrTableName interface{}) (maxID int64, err error) {
- _, err = x.Select("MAX(id)").Table(beanOrTableName).Get(&maxID)
- return
-}
-
-// FindByMaxID filled results as the condition from database
-func FindByMaxID(maxID int64, limit int, results interface{}) error {
- return x.Where("id <= ?", maxID).
- OrderBy("id DESC").
- Limit(limit).
- Find(results)
-}