# HG changeset patch # User Christophe de Vienne <christophe@cdevienne.info> # Date 1618003691 -7200 # Fri Apr 09 23:28:11 2021 +0200 # Node ID fa4e7de857924acebccd3fa32dbd4c08328b2984 # Parent 47a0379860f0f8b1178911727507e36bbb5bfe0e sequence: small cleanup diff --git a/database/sequence.go b/database/sequence.go --- a/database/sequence.go +++ b/database/sequence.go @@ -12,16 +12,18 @@ return SequenceBuilder{name} } -func (s SequenceBuilder) Set(value int) sq.SelectBuilder { - return sq.Select().Column("setval(?, ?)", s.name, value) +func (s SequenceBuilder) Set(value int, isCalled bool) sq.SelectBuilder { + return sq.Select().Column("setval(?, ?, ?)", s.name, value, isCalled) +} + +func (s SequenceBuilder) Next() sq.SelectBuilder { + return sq.Select().Column("nextval(?)", s.name) +} + +func (s SequenceBuilder) Current() sq.SelectBuilder { + return sq.Select().Column("currval(?)", s.name) } func (s SequenceBuilder) Last() sq.SelectBuilder { - return sq.Select().Column("last_value").From(s.name) + return sq.Select("last_value").From(s.name) } -func (s SequenceBuilder) Next() sq.SelectBuilder { - return sq.Select().Column("nextval(?)", s.name) -} -func (s SequenceBuilder) Current() sq.SelectBuilder { - return sq.Select().Column("currval(?)", s.name) -}