seaweedfs/weed/filer/postgres/postgres_sql_gen.go

59 lines
1.9 KiB
Go
Raw Normal View History

2021-01-20 02:07:29 +00:00
package postgres
import (
2021-02-18 04:57:08 +00:00
"fmt"
2021-01-20 02:07:29 +00:00
2021-02-18 04:57:08 +00:00
"github.com/chrislusf/seaweedfs/weed/filer/abstract_sql"
_ "github.com/lib/pq"
2021-01-20 02:07:29 +00:00
)
type SqlGenPostgres struct {
CreateTableSqlTemplate string
DropTableSqlTemplate string
UpsertQueryTemplate string
2021-01-20 02:07:29 +00:00
}
var (
_ = abstract_sql.SqlGenerator(&SqlGenPostgres{})
)
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlInsert(tableName string) string {
if gen.UpsertQueryTemplate != "" {
return fmt.Sprintf(gen.UpsertQueryTemplate, tableName)
} else {
return fmt.Sprintf(`INSERT INTO "%s" (dirhash,name,directory,meta) VALUES($1,$2,$3,$4)`, tableName)
}
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlUpdate(tableName string) string {
return fmt.Sprintf(`UPDATE "%s" SET meta=$1 WHERE dirhash=$2 AND name=$3 AND directory=$4`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlFind(tableName string) string {
return fmt.Sprintf(`SELECT meta FROM "%s" WHERE dirhash=$1 AND name=$2 AND directory=$3`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlDelete(tableName string) string {
return fmt.Sprintf(`DELETE FROM "%s" WHERE dirhash=$1 AND name=$2 AND directory=$3`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlDeleteFolderChildren(tableName string) string {
return fmt.Sprintf(`DELETE FROM "%s" WHERE dirhash=$1 AND directory=$2`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlListExclusive(tableName string) string {
return fmt.Sprintf(`SELECT NAME, meta FROM "%s" WHERE dirhash=$1 AND name>$2 AND directory=$3 AND name like $4 ORDER BY NAME ASC LIMIT $5`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlListInclusive(tableName string) string {
return fmt.Sprintf(`SELECT NAME, meta FROM "%s" WHERE dirhash=$1 AND name>=$2 AND directory=$3 AND name like $4 ORDER BY NAME ASC LIMIT $5`, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlCreateTable(tableName string) string {
return fmt.Sprintf(gen.CreateTableSqlTemplate, tableName)
2021-01-20 02:07:29 +00:00
}
2021-03-25 19:05:51 +00:00
func (gen *SqlGenPostgres) GetSqlDropTable(tableName string) string {
return fmt.Sprintf(gen.DropTableSqlTemplate, tableName)
2021-01-20 02:07:29 +00:00
}