Method Upsert

kontrol/postgres.go:243–293

(kiteProt *protocol.Kite, value *kontrolprotocol.RegisterValue)

}

func (p *Postgres) Upsert(kiteProt *protocol.Kite, value *kontrolprotocol.RegisterValue) (err error) {
	// check that the incoming URL is valid to prevent malformed input
	_, err = url.Parse(value.URL)
	if err != nil {
		return err
	}

	if value.KeyID == "" {
		return errors.New("postgres: keyId is empty. Aborting upsert")
	}

	// we are going to try an UPDATE, if it's not successful we are going to
	// INSERT the document, all ine one single transaction
	tx, err := p.DB.Begin()
	if err != nil {
		return err
	}

	defer func() {
		if err != nil {
			err = tx.Rollback()
		} else {
			// it calls Rollback inside if it fails again :)
			err = tx.Commit()
		}
	}()

	res, err := tx.Exec(`UPDATE kite.kite SET url = $1, key_id = $3, updated_at = (now() at time zone 'utc') WHERE id = $2`,
		value.URL, kiteProt.ID, value.KeyID)
	if err != nil {
		return err
	}

	rowAffected, err := res.RowsAffected()
	if err != nil {
		return err
	}

	// we got an update! so this was successful, just return without an error
	if rowAffected != 0 {
		return nil
	}

	insertSQL, args, err := insertKiteQuery(kiteProt, value.URL, value.KeyID)
	if err != nil {
		return err
	}

	_, err = tx.Exec(insertSQL, args...)
	return err
}

func (p *Postgres) Add(kiteProt *protocol.Kite, value *kontrolprotocol.RegisterValue) error {
	// check that the incoming URL is valid to prevent malformed input