Spanner: support for lastInsertId (#101770)

* Support for returning inserted ID in Spanner.
* Convert bytes to string when storing them into text column.

pkg/services/sqlstore/migrator/migrator.go

@@ -240,19 +240,19 @@ func (mg *Migrator) run(ctx context.Context) (err error) {
 
 	migrationLogExists, err := mg.DBEngine.IsTableExist(mg.tableName)
 	if err != nil {
-		return fmt.Errorf("%v: %w", "failed to check table existence", err)
+		return fmt.Errorf("failed to check table existence: %w", err)
 	}
 
 	if !migrationLogExists {
 		// Check if dialect can initialize database from a snapshot.
 		err := mg.Dialect.CreateDatabaseFromSnapshot(ctx, mg.DBEngine, mg.tableName)
 		if err != nil {
-			return fmt.Errorf("%v: %w", "failed to create database from snapshot", err)
+			return fmt.Errorf("failed to create database from snapshot: %w", err)
 		}
 
 		migrationLogExists, err = mg.DBEngine.IsTableExist(mg.tableName)
 		if err != nil {
-			return fmt.Errorf("%v: %w", "failed to check table existence after applying snapshot", err)
+			return fmt.Errorf("failed to check table existence after applying snapshot: %w", err)
 		}
 	}
 

pkg/services/sqlstore/session.go

@@ -11,6 +11,7 @@ import (
 	"go.opentelemetry.io/otel/attribute"
 	"go.opentelemetry.io/otel/trace"
 	"go.opentelemetry.io/otel/trace/noop"
+
 	"xorm.io/xorm"
 
 	"github.com/grafana/grafana/pkg/apimachinery/errutil"
@@ -140,14 +141,18 @@ func (sess *DBSession) InsertId(bean any, dialect migrator.Dialect) error {
 }
 
 func (sess *DBSession) WithReturningID(driverName string, query string, args []any) (int64, error) {
-	supported := driverName != migrator.Postgres
 	var id int64
-	if !supported {
+	if driverName == migrator.Postgres {
 		query = fmt.Sprintf("%s RETURNING id", query)
 		if _, err := sess.SQL(query, args...).Get(&id); err != nil {
 			return id, err
 		}
 	} else {
+		if driverName == migrator.Spanner {
+			// Only works with INSERT statements.
+			query = fmt.Sprintf("%s THEN RETURN id", query)
+		}
+
 		sqlOrArgs := append([]any{query}, args...)
 		res, err := sess.Exec(sqlOrArgs...)
 		if err != nil {