diff options
| author | Chris Lu <chrislusf@users.noreply.github.com> | 2021-03-30 00:25:01 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-03-30 00:25:01 -0700 |
| commit | bec74c3e6a4279288c790fb052eb2844e3b9bc36 (patch) | |
| tree | 555d0e3edf71e66f404185c3148b7553b7c9c50c /weed/filer/postgres2/postgres2_store.go | |
| parent | a95929e53ca609181dddda04ece0e6d482234161 (diff) | |
| parent | 9f1cab179c1c41deb1db627e8bfbc632a2dffba7 (diff) | |
| download | seaweedfs-bec74c3e6a4279288c790fb052eb2844e3b9bc36.tar.xz seaweedfs-bec74c3e6a4279288c790fb052eb2844e3b9bc36.zip | |
Merge pull request #1948 from LazyDBA247-Anyvision/master
add enableUpsert=true
Diffstat (limited to 'weed/filer/postgres2/postgres2_store.go')
| -rw-r--r-- | weed/filer/postgres2/postgres2_store.go | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/weed/filer/postgres2/postgres2_store.go b/weed/filer/postgres2/postgres2_store.go index b83945db6..01fc4a869 100644 --- a/weed/filer/postgres2/postgres2_store.go +++ b/weed/filer/postgres2/postgres2_store.go @@ -32,7 +32,8 @@ func (store *PostgresStore2) GetName() string { func (store *PostgresStore2) Initialize(configuration util.Configuration, prefix string) (err error) { return store.initialize( configuration.GetString(prefix+"createTable"), - configuration.GetString(prefix+"insertQuery"), + configuration.GetString(prefix+"upsertQuery"), + configuration.GetBool(prefix+"enableUpsert"), configuration.GetString(prefix+"username"), configuration.GetString(prefix+"password"), configuration.GetString(prefix+"hostname"), @@ -46,13 +47,16 @@ func (store *PostgresStore2) Initialize(configuration util.Configuration, prefix ) } -func (store *PostgresStore2) initialize(createTable, insertQuery, user, password, hostname string, port int, database, schema, sslmode string, maxIdle, maxOpen, maxLifetimeSeconds int) (err error) { +func (store *PostgresStore2) initialize(createTable, upsertQuery string, enableUpsert bool, user, password, hostname string, port int, database, schema, sslmode string, maxIdle, maxOpen, maxLifetimeSeconds int) (err error) { store.SupportBucketTable = true + if !enableUpsert { + upsertQuery = "" + } store.SqlGenerator = &postgres.SqlGenPostgres{ CreateTableSqlTemplate: createTable, DropTableSqlTemplate: `drop table "%s"`, - InsertQueryTemplate: insertQuery, + UpsertQueryTemplate: upsertQuery, } sqlUrl := fmt.Sprintf(CONNECTION_URL_PATTERN, hostname, port, sslmode) |