diff --git a/database/sequence.go b/database/sequence.go index 47a0379860f0f8b1178911727507e36bbb5bfe0e_ZGF0YWJhc2Uvc2VxdWVuY2UuZ28=..fa4e7de857924acebccd3fa32dbd4c08328b2984_ZGF0YWJhc2Uvc2VxdWVuY2UuZ28= 100644 --- a/database/sequence.go +++ b/database/sequence.go @@ -12,8 +12,16 @@ 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 { @@ -17,5 +25,5 @@ } func (s SequenceBuilder) Last() sq.SelectBuilder { - return sq.Select().Column("last_value").From(s.name) + return sq.Select("last_value").From(s.name) } @@ -21,7 +29,1 @@ } -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) -}