mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 05:42:25 -05:00 
			
		
		
		
	[feature] add worker task serialization logic (#2989)
* improved server shutdown with more precise shutdown of modules + deferring of ALL of it
* move delivery and workers into separate files
* add worker task model and Serialize() / Deserialize() methods for message types
* start adding message serialize / deserialize tests
* start adding test cases
* update body rewinding to rely on standard library mechanism of r.GetBody()
* remove request rewinding (http.Client{} should already handle this)
* standard library already handles rewinding
* improved code comment
* move the newPOST() function contents to prepare(), fits better with current API
* add Serialize() / Deserialize() implementations for Delivery{} type
* finish writing FromClientAPI sserialize / deserialize tests
* start adding FromFediAPI{} serialize / deserialize test cases
* fix FromFediAPI{} tests
* add tests for delivery message type
* fix repeat code
* missing license header
* use testrig status and accounts for marshal / unmarshaling tests
* add a specific test for checking account RSA keys are preserved
	
	
This commit is contained in:
		
					parent
					
						
							
								4ce5c37df5
							
						
					
				
			
			
				commit
				
					
						7b1ccbd65a
					
				
			
		
					 31 changed files with 6318 additions and 457 deletions
				
			
		
							
								
								
									
										1
									
								
								go.mod
									
										
									
									
									
								
							
							
						
						
									
										1
									
								
								go.mod
									
										
									
									
									
								
							|  | @ -35,6 +35,7 @@ require ( | ||||||
| 	github.com/gin-gonic/gin v1.10.0 | 	github.com/gin-gonic/gin v1.10.0 | ||||||
| 	github.com/go-playground/form/v4 v4.2.1 | 	github.com/go-playground/form/v4 v4.2.1 | ||||||
| 	github.com/go-swagger/go-swagger v0.31.0 | 	github.com/go-swagger/go-swagger v0.31.0 | ||||||
|  | 	github.com/google/go-cmp v0.6.0 | ||||||
| 	github.com/google/uuid v1.6.0 | 	github.com/google/uuid v1.6.0 | ||||||
| 	github.com/gorilla/feeds v1.1.2 | 	github.com/gorilla/feeds v1.1.2 | ||||||
| 	github.com/gorilla/websocket v1.5.2 | 	github.com/gorilla/websocket v1.5.2 | ||||||
|  |  | ||||||
							
								
								
									
										41
									
								
								internal/gtsmodel/workertask.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								internal/gtsmodel/workertask.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,41 @@ | ||||||
|  | // GoToSocial | ||||||
|  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||||
|  | // SPDX-License-Identifier: AGPL-3.0-or-later | ||||||
|  | // | ||||||
|  | // This program is free software: you can redistribute it and/or modify | ||||||
|  | // it under the terms of the GNU Affero General Public License as published by | ||||||
|  | // the Free Software Foundation, either version 3 of the License, or | ||||||
|  | // (at your option) any later version. | ||||||
|  | // | ||||||
|  | // This program is distributed in the hope that it will be useful, | ||||||
|  | // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||||
|  | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||||
|  | // GNU Affero General Public License for more details. | ||||||
|  | // | ||||||
|  | // You should have received a copy of the GNU Affero General Public License | ||||||
|  | // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||||
|  | 
 | ||||||
|  | package gtsmodel | ||||||
|  | 
 | ||||||
|  | import "time" | ||||||
|  | 
 | ||||||
|  | type WorkerType uint8 | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	DeliveryWorker  WorkerType = 1 | ||||||
|  | 	FederatorWorker WorkerType = 2 | ||||||
|  | 	ClientWorker    WorkerType = 3 | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // WorkerTask represents a queued worker task | ||||||
|  | // that was persisted to the database on shutdown. | ||||||
|  | // This is only ever used on startup to pickup | ||||||
|  | // where we left off, and on shutdown to prevent | ||||||
|  | // queued tasks from being lost. It is simply a | ||||||
|  | // means to store a blob of serialized task data. | ||||||
|  | type WorkerTask struct { | ||||||
|  | 	ID         uint      `bun:""` | ||||||
|  | 	WorkerType uint8     `bun:""` | ||||||
|  | 	TaskData   []byte    `bun:""` | ||||||
|  | 	CreatedAt  time.Time `bun:""` | ||||||
|  | } | ||||||
|  | @ -21,7 +21,6 @@ import ( | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"time" | 	"time" | ||||||
| 
 | 
 | ||||||
| 	"codeberg.org/gruf/go-byteutil" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" | 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
|  | @ -45,12 +44,6 @@ func (t *signingtransport) RoundTrip(r *http.Request) (*http.Response, error) { | ||||||
| 		r.Header.Del("Signature") | 		r.Header.Del("Signature") | ||||||
| 		r.Header.Del("Digest") | 		r.Header.Del("Digest") | ||||||
| 
 | 
 | ||||||
| 		// Rewind body reader and content-length if set. |  | ||||||
| 		if rc, ok := r.Body.(*byteutil.ReadNopCloser); ok { |  | ||||||
| 			rc.Rewind() // set len AFTER rewind |  | ||||||
| 			r.ContentLength = int64(rc.Len()) |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Sign the outgoing request. | 		// Sign the outgoing request. | ||||||
| 		if err := sign(r); err != nil { | 		if err := sign(r); err != nil { | ||||||
| 			return nil, err | 			return nil, err | ||||||
|  | @ -18,9 +18,15 @@ | ||||||
| package messages | package messages | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"encoding/json" | ||||||
| 	"net/url" | 	"net/url" | ||||||
|  | 	"reflect" | ||||||
| 
 | 
 | ||||||
| 	"codeberg.org/gruf/go-structr" | 	"codeberg.org/gruf/go-structr" | ||||||
|  | 	"github.com/superseriousbusiness/activity/streams" | ||||||
|  | 	"github.com/superseriousbusiness/activity/streams/vocab" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtserror" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
|  | @ -50,6 +56,105 @@ type FromClientAPI struct { | ||||||
| 	Target *gtsmodel.Account | 	Target *gtsmodel.Account | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // fromClientAPI is an internal type | ||||||
|  | // for FromClientAPI that provides a | ||||||
|  | // json serialize / deserialize -able | ||||||
|  | // shape that minimizes required data. | ||||||
|  | type fromClientAPI struct { | ||||||
|  | 	APObjectType   string          `json:"ap_object_type,omitempty"` | ||||||
|  | 	APActivityType string          `json:"ap_activity_type,omitempty"` | ||||||
|  | 	GTSModel       json.RawMessage `json:"gts_model,omitempty"` | ||||||
|  | 	GTSModelType   string          `json:"gts_model_type,omitempty"` | ||||||
|  | 	TargetURI      string          `json:"target_uri,omitempty"` | ||||||
|  | 	OriginID       string          `json:"origin_id,omitempty"` | ||||||
|  | 	TargetID       string          `json:"target_id,omitempty"` | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Serialize will serialize the worker data as data blob for storage, | ||||||
|  | // note that this will flatten some of the data e.g. only account IDs. | ||||||
|  | func (msg *FromClientAPI) Serialize() ([]byte, error) { | ||||||
|  | 	var ( | ||||||
|  | 		modelType string | ||||||
|  | 		originID  string | ||||||
|  | 		targetID  string | ||||||
|  | 	) | ||||||
|  | 
 | ||||||
|  | 	// Set database model type if any provided. | ||||||
|  | 	if t := reflect.TypeOf(msg.GTSModel); t != nil { | ||||||
|  | 		modelType = t.String() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set origin account ID. | ||||||
|  | 	if msg.Origin != nil { | ||||||
|  | 		originID = msg.Origin.ID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set target account ID. | ||||||
|  | 	if msg.Target != nil { | ||||||
|  | 		targetID = msg.Target.ID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Marshal GTS model as raw JSON block. | ||||||
|  | 	modelJSON, err := json.Marshal(msg.GTSModel) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Marshal as internal JSON type. | ||||||
|  | 	return json.Marshal(fromClientAPI{ | ||||||
|  | 		APObjectType:   msg.APObjectType, | ||||||
|  | 		APActivityType: msg.APActivityType, | ||||||
|  | 		GTSModel:       modelJSON, | ||||||
|  | 		GTSModelType:   modelType, | ||||||
|  | 		TargetURI:      msg.TargetURI, | ||||||
|  | 		OriginID:       originID, | ||||||
|  | 		TargetID:       targetID, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Deserialize will attempt to deserialize a blob of task data, | ||||||
|  | // which will involve unflattening previously serialized data and | ||||||
|  | // leave some message structures as placeholders to holding IDs. | ||||||
|  | func (msg *FromClientAPI) Deserialize(data []byte) error { | ||||||
|  | 	var imsg fromClientAPI | ||||||
|  | 
 | ||||||
|  | 	// Unmarshal as internal JSON type. | ||||||
|  | 	err := json.Unmarshal(data, &imsg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Copy over the simplest fields. | ||||||
|  | 	msg.APObjectType = imsg.APObjectType | ||||||
|  | 	msg.APActivityType = imsg.APActivityType | ||||||
|  | 	msg.TargetURI = imsg.TargetURI | ||||||
|  | 
 | ||||||
|  | 	// Resolve Go type from JSON data. | ||||||
|  | 	msg.GTSModel, err = resolveGTSModel( | ||||||
|  | 		imsg.GTSModelType, | ||||||
|  | 		imsg.GTSModel, | ||||||
|  | 	) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if imsg.OriginID != "" { | ||||||
|  | 		// Set origin account ID using a | ||||||
|  | 		// barebones model (later filled in). | ||||||
|  | 		msg.Origin = new(gtsmodel.Account) | ||||||
|  | 		msg.Origin.ID = imsg.OriginID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if imsg.TargetID != "" { | ||||||
|  | 		// Set target account ID using a | ||||||
|  | 		// barebones model (later filled in). | ||||||
|  | 		msg.Target = new(gtsmodel.Account) | ||||||
|  | 		msg.Target.ID = imsg.TargetID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| // ClientMsgIndices defines queue indices this | // ClientMsgIndices defines queue indices this | ||||||
| // message type should be accessible / stored under. | // message type should be accessible / stored under. | ||||||
| func ClientMsgIndices() []structr.IndexConfig { | func ClientMsgIndices() []structr.IndexConfig { | ||||||
|  | @ -91,6 +196,133 @@ type FromFediAPI struct { | ||||||
| 	Receiving *gtsmodel.Account | 	Receiving *gtsmodel.Account | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // fromFediAPI is an internal type | ||||||
|  | // for FromFediAPI that provides a | ||||||
|  | // json serialize / deserialize -able | ||||||
|  | // shape that minimizes required data. | ||||||
|  | type fromFediAPI struct { | ||||||
|  | 	APObjectType   string                 `json:"ap_object_type,omitempty"` | ||||||
|  | 	APActivityType string                 `json:"ap_activity_type,omitempty"` | ||||||
|  | 	APIRI          string                 `json:"ap_iri,omitempty"` | ||||||
|  | 	APObject       map[string]interface{} `json:"ap_object,omitempty"` | ||||||
|  | 	GTSModel       json.RawMessage        `json:"gts_model,omitempty"` | ||||||
|  | 	GTSModelType   string                 `json:"gts_model_type,omitempty"` | ||||||
|  | 	TargetURI      string                 `json:"target_uri,omitempty"` | ||||||
|  | 	RequestingID   string                 `json:"requesting_id,omitempty"` | ||||||
|  | 	ReceivingID    string                 `json:"receiving_id,omitempty"` | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Serialize will serialize the worker data as data blob for storage, | ||||||
|  | // note that this will flatten some of the data e.g. only account IDs. | ||||||
|  | func (msg *FromFediAPI) Serialize() ([]byte, error) { | ||||||
|  | 	var ( | ||||||
|  | 		gtsModelType string | ||||||
|  | 		apIRI        string | ||||||
|  | 		apObject     map[string]interface{} | ||||||
|  | 		requestingID string | ||||||
|  | 		receivingID  string | ||||||
|  | 	) | ||||||
|  | 
 | ||||||
|  | 	// Set AP IRI string. | ||||||
|  | 	if msg.APIRI != nil { | ||||||
|  | 		apIRI = msg.APIRI.String() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set serialized AP object data if set. | ||||||
|  | 	if t, ok := msg.APObject.(vocab.Type); ok { | ||||||
|  | 		obj, err := t.Serialize() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		apObject = obj | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set database model type if any provided. | ||||||
|  | 	if t := reflect.TypeOf(msg.GTSModel); t != nil { | ||||||
|  | 		gtsModelType = t.String() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set requesting account ID. | ||||||
|  | 	if msg.Requesting != nil { | ||||||
|  | 		requestingID = msg.Requesting.ID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Set receiving account ID. | ||||||
|  | 	if msg.Receiving != nil { | ||||||
|  | 		receivingID = msg.Receiving.ID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Marshal GTS model as raw JSON block. | ||||||
|  | 	modelJSON, err := json.Marshal(msg.GTSModel) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Marshal as internal JSON type. | ||||||
|  | 	return json.Marshal(fromFediAPI{ | ||||||
|  | 		APObjectType:   msg.APObjectType, | ||||||
|  | 		APActivityType: msg.APActivityType, | ||||||
|  | 		APIRI:          apIRI, | ||||||
|  | 		APObject:       apObject, | ||||||
|  | 		GTSModel:       modelJSON, | ||||||
|  | 		GTSModelType:   gtsModelType, | ||||||
|  | 		TargetURI:      msg.TargetURI, | ||||||
|  | 		RequestingID:   requestingID, | ||||||
|  | 		ReceivingID:    receivingID, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Deserialize will attempt to deserialize a blob of task data, | ||||||
|  | // which will involve unflattening previously serialized data and | ||||||
|  | // leave some message structures as placeholders to holding IDs. | ||||||
|  | func (msg *FromFediAPI) Deserialize(data []byte) error { | ||||||
|  | 	var imsg fromFediAPI | ||||||
|  | 
 | ||||||
|  | 	// Unmarshal as internal JSON type. | ||||||
|  | 	err := json.Unmarshal(data, &imsg) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Copy over the simplest fields. | ||||||
|  | 	msg.APObjectType = imsg.APObjectType | ||||||
|  | 	msg.APActivityType = imsg.APActivityType | ||||||
|  | 	msg.TargetURI = imsg.TargetURI | ||||||
|  | 
 | ||||||
|  | 	// Resolve AP object from JSON data. | ||||||
|  | 	msg.APObject, err = resolveAPObject( | ||||||
|  | 		imsg.APObject, | ||||||
|  | 	) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Resolve Go type from JSON data. | ||||||
|  | 	msg.GTSModel, err = resolveGTSModel( | ||||||
|  | 		imsg.GTSModelType, | ||||||
|  | 		imsg.GTSModel, | ||||||
|  | 	) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if imsg.RequestingID != "" { | ||||||
|  | 		// Set requesting account ID using a | ||||||
|  | 		// barebones model (later filled in). | ||||||
|  | 		msg.Requesting = new(gtsmodel.Account) | ||||||
|  | 		msg.Requesting.ID = imsg.RequestingID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if imsg.ReceivingID != "" { | ||||||
|  | 		// Set target account ID using a | ||||||
|  | 		// barebones model (later filled in). | ||||||
|  | 		msg.Receiving = new(gtsmodel.Account) | ||||||
|  | 		msg.Receiving.ID = imsg.ReceivingID | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| // FederatorMsgIndices defines queue indices this | // FederatorMsgIndices defines queue indices this | ||||||
| // message type should be accessible / stored under. | // message type should be accessible / stored under. | ||||||
| func FederatorMsgIndices() []structr.IndexConfig { | func FederatorMsgIndices() []structr.IndexConfig { | ||||||
|  | @ -101,3 +333,61 @@ func FederatorMsgIndices() []structr.IndexConfig { | ||||||
| 		{Fields: "Receiving.ID", Multiple: true}, | 		{Fields: "Receiving.ID", Multiple: true}, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  | 
 | ||||||
|  | // resolveAPObject resolves an ActivityPub object from its "serialized" JSON map | ||||||
|  | // (yes the terminology here is weird, but that's how go-fed/activity is written). | ||||||
|  | func resolveAPObject(data map[string]interface{}) (interface{}, error) { | ||||||
|  | 	if len(data) == 0 { | ||||||
|  | 		// No data given. | ||||||
|  | 		return nil, nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Resolve vocab.Type from "raw" input data map. | ||||||
|  | 	return streams.ToType(context.Background(), data) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // resolveGTSModel is unfortunately where things get messy... our data is stored as JSON | ||||||
|  | // in the database, which serializes struct types as key-value pairs surrounded by curly | ||||||
|  | // braces. Deserializing from that gives us back a data blob of key-value pairs, which | ||||||
|  | // we then need to wrangle back into the original type. So we also store the type name | ||||||
|  | // and use this to determine the appropriate Go structure type to unmarshal into to. | ||||||
|  | func resolveGTSModel(typ string, data []byte) (interface{}, error) { | ||||||
|  | 	if typ == "" && data == nil { | ||||||
|  | 		// No data given. | ||||||
|  | 		return nil, nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	var value interface{} | ||||||
|  | 
 | ||||||
|  | 	switch typ { | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Account)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Account) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Block)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Block) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Follow)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Follow) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.FollowRequest)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.FollowRequest) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Move)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Move) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Poll)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Poll) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.PollVote)(nil)).String(): | ||||||
|  | 		value = new(*gtsmodel.PollVote) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Report)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Report) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.Status)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.Status) | ||||||
|  | 	case reflect.TypeOf((*gtsmodel.StatusFave)(nil)).String(): | ||||||
|  | 		value = new(gtsmodel.StatusFave) | ||||||
|  | 	default: | ||||||
|  | 		return nil, gtserror.Newf("unknown type: %s", typ) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Attempt to unmarshal value JSON into destination. | ||||||
|  | 	if err := json.Unmarshal(data, &value); err != nil { | ||||||
|  | 		return nil, gtserror.Newf("error unmarshaling %s value data: %w", typ, err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return value, nil | ||||||
|  | } | ||||||
|  |  | ||||||
							
								
								
									
										292
									
								
								internal/messages/messages_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										292
									
								
								internal/messages/messages_test.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,292 @@ | ||||||
|  | // GoToSocial | ||||||
|  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||||
|  | // SPDX-License-Identifier: AGPL-3.0-or-later | ||||||
|  | // | ||||||
|  | // This program is free software: you can redistribute it and/or modify | ||||||
|  | // it under the terms of the GNU Affero General Public License as published by | ||||||
|  | // the Free Software Foundation, either version 3 of the License, or | ||||||
|  | // (at your option) any later version. | ||||||
|  | // | ||||||
|  | // This program is distributed in the hope that it will be useful, | ||||||
|  | // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||||
|  | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||||
|  | // GNU Affero General Public License for more details. | ||||||
|  | // | ||||||
|  | // You should have received a copy of the GNU Affero General Public License | ||||||
|  | // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||||
|  | 
 | ||||||
|  | package messages_test | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"encoding/json" | ||||||
|  | 	"net/url" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/ap" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/messages" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/testrig" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var testStatus = testrig.NewTestStatuses()["admin_account_status_1"] | ||||||
|  | 
 | ||||||
|  | var testAccount = testrig.NewTestAccounts()["admin_account"] | ||||||
|  | 
 | ||||||
|  | var fromClientAPICases = []struct { | ||||||
|  | 	msg  messages.FromClientAPI | ||||||
|  | 	data []byte | ||||||
|  | }{ | ||||||
|  | 	{ | ||||||
|  | 		msg: messages.FromClientAPI{ | ||||||
|  | 			APObjectType:   ap.ObjectNote, | ||||||
|  | 			APActivityType: ap.ActivityCreate, | ||||||
|  | 			GTSModel:       testStatus, | ||||||
|  | 			TargetURI:      "https://gotosocial.org", | ||||||
|  | 			Origin:         >smodel.Account{ID: "654321"}, | ||||||
|  | 			Target:         >smodel.Account{ID: "123456"}, | ||||||
|  | 		}, | ||||||
|  | 		data: toJSON(map[string]any{ | ||||||
|  | 			"ap_object_type":   ap.ObjectNote, | ||||||
|  | 			"ap_activity_type": ap.ActivityCreate, | ||||||
|  | 			"gts_model":        json.RawMessage(toJSON(testStatus)), | ||||||
|  | 			"gts_model_type":   "*gtsmodel.Status", | ||||||
|  | 			"target_uri":       "https://gotosocial.org", | ||||||
|  | 			"origin_id":        "654321", | ||||||
|  | 			"target_id":        "123456", | ||||||
|  | 		}), | ||||||
|  | 	}, | ||||||
|  | 	{ | ||||||
|  | 		msg: messages.FromClientAPI{ | ||||||
|  | 			APObjectType:   ap.ObjectProfile, | ||||||
|  | 			APActivityType: ap.ActivityUpdate, | ||||||
|  | 			GTSModel:       testAccount, | ||||||
|  | 			TargetURI:      "https://uk-queen-is-dead.org", | ||||||
|  | 			Origin:         >smodel.Account{ID: "123456"}, | ||||||
|  | 			Target:         >smodel.Account{ID: "654321"}, | ||||||
|  | 		}, | ||||||
|  | 		data: toJSON(map[string]any{ | ||||||
|  | 			"ap_object_type":   ap.ObjectProfile, | ||||||
|  | 			"ap_activity_type": ap.ActivityUpdate, | ||||||
|  | 			"gts_model":        json.RawMessage(toJSON(testAccount)), | ||||||
|  | 			"gts_model_type":   "*gtsmodel.Account", | ||||||
|  | 			"target_uri":       "https://uk-queen-is-dead.org", | ||||||
|  | 			"origin_id":        "123456", | ||||||
|  | 			"target_id":        "654321", | ||||||
|  | 		}), | ||||||
|  | 	}, | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | var fromFediAPICases = []struct { | ||||||
|  | 	msg  messages.FromFediAPI | ||||||
|  | 	data []byte | ||||||
|  | }{ | ||||||
|  | 	{ | ||||||
|  | 		msg: messages.FromFediAPI{ | ||||||
|  | 			APObjectType:   ap.ObjectNote, | ||||||
|  | 			APActivityType: ap.ActivityCreate, | ||||||
|  | 			GTSModel:       testStatus, | ||||||
|  | 			TargetURI:      "https://gotosocial.org", | ||||||
|  | 			Requesting:     >smodel.Account{ID: "654321"}, | ||||||
|  | 			Receiving:      >smodel.Account{ID: "123456"}, | ||||||
|  | 		}, | ||||||
|  | 		data: toJSON(map[string]any{ | ||||||
|  | 			"ap_object_type":   ap.ObjectNote, | ||||||
|  | 			"ap_activity_type": ap.ActivityCreate, | ||||||
|  | 			"gts_model":        json.RawMessage(toJSON(testStatus)), | ||||||
|  | 			"gts_model_type":   "*gtsmodel.Status", | ||||||
|  | 			"target_uri":       "https://gotosocial.org", | ||||||
|  | 			"requesting_id":    "654321", | ||||||
|  | 			"receiving_id":     "123456", | ||||||
|  | 		}), | ||||||
|  | 	}, | ||||||
|  | 	{ | ||||||
|  | 		msg: messages.FromFediAPI{ | ||||||
|  | 			APObjectType:   ap.ObjectProfile, | ||||||
|  | 			APActivityType: ap.ActivityUpdate, | ||||||
|  | 			GTSModel:       testAccount, | ||||||
|  | 			TargetURI:      "https://uk-queen-is-dead.org", | ||||||
|  | 			Requesting:     >smodel.Account{ID: "123456"}, | ||||||
|  | 			Receiving:      >smodel.Account{ID: "654321"}, | ||||||
|  | 		}, | ||||||
|  | 		data: toJSON(map[string]any{ | ||||||
|  | 			"ap_object_type":   ap.ObjectProfile, | ||||||
|  | 			"ap_activity_type": ap.ActivityUpdate, | ||||||
|  | 			"gts_model":        json.RawMessage(toJSON(testAccount)), | ||||||
|  | 			"gts_model_type":   "*gtsmodel.Account", | ||||||
|  | 			"target_uri":       "https://uk-queen-is-dead.org", | ||||||
|  | 			"requesting_id":    "123456", | ||||||
|  | 			"receiving_id":     "654321", | ||||||
|  | 		}), | ||||||
|  | 	}, | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestSerializeFromClientAPI(t *testing.T) { | ||||||
|  | 	for _, test := range fromClientAPICases { | ||||||
|  | 		// Serialize test message to blob. | ||||||
|  | 		data, err := test.msg.Serialize() | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check serialized JSON data as expected. | ||||||
|  | 		assertJSONEqual(t, test.data, data) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestDeserializeFromClientAPI(t *testing.T) { | ||||||
|  | 	for _, test := range fromClientAPICases { | ||||||
|  | 		var msg messages.FromClientAPI | ||||||
|  | 
 | ||||||
|  | 		// Deserialize test message blob. | ||||||
|  | 		err := msg.Deserialize(test.data) | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check that msg is as expected. | ||||||
|  | 		assertEqual(t, test.msg.APActivityType, msg.APActivityType) | ||||||
|  | 		assertEqual(t, test.msg.APObjectType, msg.APObjectType) | ||||||
|  | 		assertEqual(t, test.msg.GTSModel, msg.GTSModel) | ||||||
|  | 		assertEqual(t, test.msg.TargetURI, msg.TargetURI) | ||||||
|  | 		assertEqual(t, accountID(test.msg.Origin), accountID(msg.Origin)) | ||||||
|  | 		assertEqual(t, accountID(test.msg.Target), accountID(msg.Target)) | ||||||
|  | 
 | ||||||
|  | 		// Perform final check to ensure | ||||||
|  | 		// account model keys deserialized. | ||||||
|  | 		assertEqualRSA(t, test.msg.GTSModel, msg.GTSModel) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestSerializeFromFediAPI(t *testing.T) { | ||||||
|  | 	for _, test := range fromFediAPICases { | ||||||
|  | 		// Serialize test message to blob. | ||||||
|  | 		data, err := test.msg.Serialize() | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check serialized JSON data as expected. | ||||||
|  | 		assertJSONEqual(t, test.data, data) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func TestDeserializeFromFediAPI(t *testing.T) { | ||||||
|  | 	for _, test := range fromFediAPICases { | ||||||
|  | 		var msg messages.FromFediAPI | ||||||
|  | 
 | ||||||
|  | 		// Deserialize test message blob. | ||||||
|  | 		err := msg.Deserialize(test.data) | ||||||
|  | 		if err != nil { | ||||||
|  | 			t.Fatal(err) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check that msg is as expected. | ||||||
|  | 		assertEqual(t, test.msg.APActivityType, msg.APActivityType) | ||||||
|  | 		assertEqual(t, test.msg.APObjectType, msg.APObjectType) | ||||||
|  | 		assertEqual(t, urlStr(test.msg.APIRI), urlStr(msg.APIRI)) | ||||||
|  | 		assertEqual(t, test.msg.APObject, msg.APObject) | ||||||
|  | 		assertEqual(t, test.msg.GTSModel, msg.GTSModel) | ||||||
|  | 		assertEqual(t, test.msg.TargetURI, msg.TargetURI) | ||||||
|  | 		assertEqual(t, accountID(test.msg.Receiving), accountID(msg.Receiving)) | ||||||
|  | 		assertEqual(t, accountID(test.msg.Requesting), accountID(msg.Requesting)) | ||||||
|  | 
 | ||||||
|  | 		// Perform final check to ensure | ||||||
|  | 		// account model keys deserialized. | ||||||
|  | 		assertEqualRSA(t, test.msg.GTSModel, msg.GTSModel) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // assertEqualRSA asserts that test account model RSA keys are equal. | ||||||
|  | func assertEqualRSA(t *testing.T, expect, receive any) bool { | ||||||
|  | 	t.Helper() | ||||||
|  | 
 | ||||||
|  | 	account1, ok1 := expect.(*gtsmodel.Account) | ||||||
|  | 
 | ||||||
|  | 	account2, ok2 := receive.(*gtsmodel.Account) | ||||||
|  | 
 | ||||||
|  | 	if ok1 != ok2 { | ||||||
|  | 		t.Errorf("different model types: expect=%T receive=%T", expect, receive) | ||||||
|  | 		return false | ||||||
|  | 	} else if !ok1 { | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if !account1.PublicKey.Equal(account2.PublicKey) { | ||||||
|  | 		t.Error("public keys do not match") | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	t.Logf("publickey=%v", account1.PublicKey) | ||||||
|  | 
 | ||||||
|  | 	if !account1.PrivateKey.Equal(account2.PrivateKey) { | ||||||
|  | 		t.Error("private keys do not match") | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	t.Logf("privatekey=%v", account1.PrivateKey) | ||||||
|  | 
 | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // assertEqual asserts that two values (of any type!) are equal, | ||||||
|  | // note we use the 'cmp' library here as it's much more useful in | ||||||
|  | // outputting debug information than testify, and handles more complex | ||||||
|  | // types like rsa public / private key comparisons correctly. | ||||||
|  | func assertEqual(t *testing.T, expect, receive any) bool { | ||||||
|  | 	t.Helper() | ||||||
|  | 	if diff := cmp.Diff(expect, receive); diff != "" { | ||||||
|  | 		t.Error(diff) | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // assertJSONEqual asserts that two slices of JSON data are equal. | ||||||
|  | func assertJSONEqual(t *testing.T, expect, receive []byte) bool { | ||||||
|  | 	t.Helper() | ||||||
|  | 	return assertEqual(t, fromJSON(expect), fromJSON(receive)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // urlStr returns url as string, or empty. | ||||||
|  | func urlStr(url *url.URL) string { | ||||||
|  | 	if url == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	return url.String() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // accountID returns account's ID, or empty. | ||||||
|  | func accountID(account *gtsmodel.Account) string { | ||||||
|  | 	if account == nil { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	return account.ID | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // fromJSON unmarshals input data as JSON. | ||||||
|  | func fromJSON(b []byte) any { | ||||||
|  | 	r := bytes.NewReader(b) | ||||||
|  | 	d := json.NewDecoder(r) | ||||||
|  | 	d.UseNumber() | ||||||
|  | 	var a any | ||||||
|  | 	err := d.Decode(&a) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	if d.More() { | ||||||
|  | 		panic("multiple json values in b") | ||||||
|  | 	} | ||||||
|  | 	return a | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // toJSON marshals input type as JSON data. | ||||||
|  | func toJSON(a any) []byte { | ||||||
|  | 	b, err := json.Marshal(a) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  | @ -18,12 +18,12 @@ | ||||||
| package transport | package transport | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
|  | 	"bytes" | ||||||
| 	"context" | 	"context" | ||||||
| 	"encoding/json" | 	"encoding/json" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"net/url" | 	"net/url" | ||||||
| 
 | 
 | ||||||
| 	"codeberg.org/gruf/go-byteutil" |  | ||||||
| 	apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util" | 	apiutil "github.com/superseriousbusiness/gotosocial/internal/api/util" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/config" | 	"github.com/superseriousbusiness/gotosocial/internal/config" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" | 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" | ||||||
|  | @ -130,25 +130,28 @@ func (t *transport) prepare( | ||||||
| 	*delivery.Delivery, | 	*delivery.Delivery, | ||||||
| 	error, | 	error, | ||||||
| ) { | ) { | ||||||
| 	url := to.String() |  | ||||||
| 
 |  | ||||||
| 	// Use rewindable reader for body. |  | ||||||
| 	var body byteutil.ReadNopCloser |  | ||||||
| 	body.Reset(data) |  | ||||||
| 
 |  | ||||||
| 	// Prepare POST signer. | 	// Prepare POST signer. | ||||||
| 	sign := t.signPOST(data) | 	sign := t.signPOST(data) | ||||||
| 
 | 
 | ||||||
|  | 	// Use *bytes.Reader for request body, | ||||||
|  | 	// as NewRequest() automatically will | ||||||
|  | 	// set .GetBody and content-length. | ||||||
|  | 	// (this handles necessary rewinding). | ||||||
|  | 	body := bytes.NewReader(data) | ||||||
|  | 
 | ||||||
| 	// Update to-be-used request context with signing details. | 	// Update to-be-used request context with signing details. | ||||||
| 	ctx = gtscontext.SetOutgoingPublicKeyID(ctx, t.pubKeyID) | 	ctx = gtscontext.SetOutgoingPublicKeyID(ctx, t.pubKeyID) | ||||||
| 	ctx = gtscontext.SetHTTPClientSignFunc(ctx, sign) | 	ctx = gtscontext.SetHTTPClientSignFunc(ctx, sign) | ||||||
| 
 | 
 | ||||||
| 	// Prepare a new request with data body directed at URL. | 	// Prepare a new request with data body directed at URL. | ||||||
| 	r, err := http.NewRequestWithContext(ctx, "POST", url, &body) | 	r, err := http.NewRequestWithContext(ctx, "POST", to.String(), body) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, gtserror.Newf("error preparing request: %w", err) | 		return nil, gtserror.Newf("error preparing request: %w", err) | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 	// Set our predefined controller user-agent. | ||||||
|  | 	r.Header.Set("User-Agent", t.controller.userAgent) | ||||||
|  | 
 | ||||||
| 	// Set the standard ActivityPub content-type + charset headers. | 	// Set the standard ActivityPub content-type + charset headers. | ||||||
| 	r.Header.Add("Content-Type", string(apiutil.AppActivityLDJSON)) | 	r.Header.Add("Content-Type", string(apiutil.AppActivityLDJSON)) | ||||||
| 	r.Header.Add("Accept-Charset", "utf-8") | 	r.Header.Add("Accept-Charset", "utf-8") | ||||||
|  |  | ||||||
|  | @ -18,16 +18,13 @@ | ||||||
| package delivery | package delivery | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"bytes" | ||||||
| 	"slices" | 	"encoding/json" | ||||||
|  | 	"io" | ||||||
|  | 	"net/http" | ||||||
| 	"time" | 	"time" | ||||||
| 
 | 
 | ||||||
| 	"codeberg.org/gruf/go-runners" |  | ||||||
| 	"codeberg.org/gruf/go-structr" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/queue" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/util" |  | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| // Delivery wraps an httpclient.Request{} | // Delivery wraps an httpclient.Request{} | ||||||
|  | @ -36,6 +33,9 @@ import ( | ||||||
| // be indexed (and so, dropped from queue) | // be indexed (and so, dropped from queue) | ||||||
| // by any of these possible ID IRIs. | // by any of these possible ID IRIs. | ||||||
| type Delivery struct { | type Delivery struct { | ||||||
|  | 	// PubKeyID is the signing public key | ||||||
|  | 	// ID of the actor performing request. | ||||||
|  | 	PubKeyID string | ||||||
| 
 | 
 | ||||||
| 	// ActorID contains the ActivityPub | 	// ActorID contains the ActivityPub | ||||||
| 	// actor ID IRI (if any) of the activity | 	// actor ID IRI (if any) of the activity | ||||||
|  | @ -61,273 +61,98 @@ type Delivery struct { | ||||||
| 	next time.Time | 	next time.Time | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // delivery is an internal type | ||||||
|  | // for Delivery{} that provides | ||||||
|  | // a json serialize / deserialize | ||||||
|  | // able shape that minimizes data. | ||||||
|  | type delivery struct { | ||||||
|  | 	PubKeyID string              `json:"pub_key_id,omitempty"` | ||||||
|  | 	ActorID  string              `json:"actor_id,omitempty"` | ||||||
|  | 	ObjectID string              `json:"object_id,omitempty"` | ||||||
|  | 	TargetID string              `json:"target_id,omitempty"` | ||||||
|  | 	Method   string              `json:"method,omitempty"` | ||||||
|  | 	Header   map[string][]string `json:"header,omitempty"` | ||||||
|  | 	URL      string              `json:"url,omitempty"` | ||||||
|  | 	Body     []byte              `json:"body,omitempty"` | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Serialize will serialize the delivery data as data blob for storage, | ||||||
|  | // note that this will flatten some of the data, dropping signing funcs. | ||||||
|  | func (dlv *Delivery) Serialize() ([]byte, error) { | ||||||
|  | 	var body []byte | ||||||
|  | 
 | ||||||
|  | 	if dlv.Request.GetBody != nil { | ||||||
|  | 		// Fetch a fresh copy of request body. | ||||||
|  | 		rbody, err := dlv.Request.GetBody() | ||||||
|  | 		if err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Read request body into memory. | ||||||
|  | 		body, err = io.ReadAll(rbody) | ||||||
|  | 
 | ||||||
|  | 		// Done with body. | ||||||
|  | 		_ = rbody.Close() | ||||||
|  | 
 | ||||||
|  | 		if err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Marshal as internal JSON type. | ||||||
|  | 	return json.Marshal(delivery{ | ||||||
|  | 		PubKeyID: dlv.PubKeyID, | ||||||
|  | 		ActorID:  dlv.ActorID, | ||||||
|  | 		ObjectID: dlv.ObjectID, | ||||||
|  | 		TargetID: dlv.TargetID, | ||||||
|  | 		Method:   dlv.Request.Method, | ||||||
|  | 		Header:   dlv.Request.Header, | ||||||
|  | 		URL:      dlv.Request.URL.String(), | ||||||
|  | 		Body:     body, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Deserialize will attempt to deserialize a blob of task data, | ||||||
|  | // which will involve unflattening previously serialized data and | ||||||
|  | // leave delivery incomplete, still requiring signing func setup. | ||||||
|  | func (dlv *Delivery) Deserialize(data []byte) error { | ||||||
|  | 	var idlv delivery | ||||||
|  | 
 | ||||||
|  | 	// Unmarshal as internal JSON type. | ||||||
|  | 	err := json.Unmarshal(data, &idlv) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Copy over simplest fields. | ||||||
|  | 	dlv.PubKeyID = idlv.PubKeyID | ||||||
|  | 	dlv.ActorID = idlv.ActorID | ||||||
|  | 	dlv.ObjectID = idlv.ObjectID | ||||||
|  | 	dlv.TargetID = idlv.TargetID | ||||||
|  | 
 | ||||||
|  | 	var body io.Reader | ||||||
|  | 
 | ||||||
|  | 	if idlv.Body != nil { | ||||||
|  | 		// Create new body reader from data. | ||||||
|  | 		body = bytes.NewReader(idlv.Body) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Create a new request object from unmarshaled details. | ||||||
|  | 	r, err := http.NewRequest(idlv.Method, idlv.URL, body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Wrap request in httpclient type. | ||||||
|  | 	dlv.Request = httpclient.WrapRequest(r) | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // backoff returns a valid (>= 0) backoff duration. | ||||||
| func (dlv *Delivery) backoff() time.Duration { | func (dlv *Delivery) backoff() time.Duration { | ||||||
| 	if dlv.next.IsZero() { | 	if dlv.next.IsZero() { | ||||||
| 		return 0 | 		return 0 | ||||||
| 	} | 	} | ||||||
| 	return time.Until(dlv.next) | 	return time.Until(dlv.next) | ||||||
| } | } | ||||||
| 
 |  | ||||||
| // WorkerPool wraps multiple Worker{}s in |  | ||||||
| // a singular struct for easy multi start/stop. |  | ||||||
| type WorkerPool struct { |  | ||||||
| 
 |  | ||||||
| 	// Client defines httpclient.Client{} |  | ||||||
| 	// passed to each of delivery pool Worker{}s. |  | ||||||
| 	Client *httpclient.Client |  | ||||||
| 
 |  | ||||||
| 	// Queue is the embedded queue.StructQueue{} |  | ||||||
| 	// passed to each of delivery pool Worker{}s. |  | ||||||
| 	Queue queue.StructQueue[*Delivery] |  | ||||||
| 
 |  | ||||||
| 	// internal fields. |  | ||||||
| 	workers []*Worker |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Init will initialize the Worker{} pool |  | ||||||
| // with given http client, request queue to pull |  | ||||||
| // from and number of delivery workers to spawn. |  | ||||||
| func (p *WorkerPool) Init(client *httpclient.Client) { |  | ||||||
| 	p.Client = client |  | ||||||
| 	p.Queue.Init(structr.QueueConfig[*Delivery]{ |  | ||||||
| 		Indices: []structr.IndexConfig{ |  | ||||||
| 			{Fields: "ActorID", Multiple: true}, |  | ||||||
| 			{Fields: "ObjectID", Multiple: true}, |  | ||||||
| 			{Fields: "TargetID", Multiple: true}, |  | ||||||
| 		}, |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Start will attempt to start 'n' Worker{}s. |  | ||||||
| func (p *WorkerPool) Start(n int) { |  | ||||||
| 	// Check whether workers are |  | ||||||
| 	// set (is already running). |  | ||||||
| 	ok := (len(p.workers) > 0) |  | ||||||
| 	if ok { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Allocate new workers slice. |  | ||||||
| 	p.workers = make([]*Worker, n) |  | ||||||
| 	for i := range p.workers { |  | ||||||
| 
 |  | ||||||
| 		// Allocate new Worker{}. |  | ||||||
| 		p.workers[i] = new(Worker) |  | ||||||
| 		p.workers[i].Client = p.Client |  | ||||||
| 		p.workers[i].Queue = &p.Queue |  | ||||||
| 
 |  | ||||||
| 		// Attempt to start worker. |  | ||||||
| 		// Return bool not useful |  | ||||||
| 		// here, as true = started, |  | ||||||
| 		// false = already running. |  | ||||||
| 		_ = p.workers[i].Start() |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Stop will attempt to stop contained Worker{}s. |  | ||||||
| func (p *WorkerPool) Stop() { |  | ||||||
| 	// Check whether workers are |  | ||||||
| 	// set (is currently running). |  | ||||||
| 	ok := (len(p.workers) == 0) |  | ||||||
| 	if ok { |  | ||||||
| 		return |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Stop all running workers. |  | ||||||
| 	for i := range p.workers { |  | ||||||
| 
 |  | ||||||
| 		// return bool not useful |  | ||||||
| 		// here, as true = stopped, |  | ||||||
| 		// false = never running. |  | ||||||
| 		_ = p.workers[i].Stop() |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Unset workers slice. |  | ||||||
| 	p.workers = p.workers[:0] |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Worker wraps an httpclient.Client{} to feed |  | ||||||
| // from queue.StructQueue{} for ActivityPub reqs |  | ||||||
| // to deliver. It does so while prioritizing new |  | ||||||
| // queued requests over backlogged retries. |  | ||||||
| type Worker struct { |  | ||||||
| 
 |  | ||||||
| 	// Client is the httpclient.Client{} that |  | ||||||
| 	// delivery worker will use for requests. |  | ||||||
| 	Client *httpclient.Client |  | ||||||
| 
 |  | ||||||
| 	// Queue is the Delivery{} message queue |  | ||||||
| 	// that delivery worker will feed from. |  | ||||||
| 	Queue *queue.StructQueue[*Delivery] |  | ||||||
| 
 |  | ||||||
| 	// internal fields. |  | ||||||
| 	backlog []*Delivery |  | ||||||
| 	service runners.Service |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Start will attempt to start the Worker{}. |  | ||||||
| func (w *Worker) Start() bool { |  | ||||||
| 	return w.service.GoRun(w.run) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // Stop will attempt to stop the Worker{}. |  | ||||||
| func (w *Worker) Stop() bool { |  | ||||||
| 	return w.service.Stop() |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // run wraps process to restart on any panic. |  | ||||||
| func (w *Worker) run(ctx context.Context) { |  | ||||||
| 	if w.Client == nil || w.Queue == nil { |  | ||||||
| 		panic("not yet initialized") |  | ||||||
| 	} |  | ||||||
| 	util.Must(func() { w.process(ctx) }) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // process is the main delivery worker processing routine. |  | ||||||
| func (w *Worker) process(ctx context.Context) bool { |  | ||||||
| 	if w.Client == nil || w.Queue == nil { |  | ||||||
| 		// we perform this check here just |  | ||||||
| 		// to ensure the compiler knows these |  | ||||||
| 		// variables aren't nil in the loop, |  | ||||||
| 		// even if already checked by caller. |  | ||||||
| 		panic("not yet initialized") |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| loop: |  | ||||||
| 	for { |  | ||||||
| 		// Get next delivery. |  | ||||||
| 		dlv, ok := w.next(ctx) |  | ||||||
| 		if !ok { |  | ||||||
| 			return true |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Check whether backoff required. |  | ||||||
| 		const min = 100 * time.Millisecond |  | ||||||
| 		if d := dlv.backoff(); d > min { |  | ||||||
| 
 |  | ||||||
| 			// Start backoff sleep timer. |  | ||||||
| 			backoff := time.NewTimer(d) |  | ||||||
| 
 |  | ||||||
| 			select { |  | ||||||
| 			case <-ctx.Done(): |  | ||||||
| 				// Main ctx |  | ||||||
| 				// cancelled. |  | ||||||
| 				backoff.Stop() |  | ||||||
| 				return true |  | ||||||
| 
 |  | ||||||
| 			case <-w.Queue.Wait(): |  | ||||||
| 				// A new message was |  | ||||||
| 				// queued, re-add this |  | ||||||
| 				// to backlog + retry. |  | ||||||
| 				w.pushBacklog(dlv) |  | ||||||
| 				backoff.Stop() |  | ||||||
| 				continue loop |  | ||||||
| 
 |  | ||||||
| 			case <-backoff.C: |  | ||||||
| 				// success! |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Attempt delivery of AP request. |  | ||||||
| 		rsp, retry, err := w.Client.DoOnce( |  | ||||||
| 			&dlv.Request, |  | ||||||
| 		) |  | ||||||
| 
 |  | ||||||
| 		if err == nil { |  | ||||||
| 			// Ensure body closed. |  | ||||||
| 			_ = rsp.Body.Close() |  | ||||||
| 			continue loop |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		if !retry { |  | ||||||
| 			// Drop deliveries when no |  | ||||||
| 			// retry requested, or they |  | ||||||
| 			// reached max (either). |  | ||||||
| 			continue loop |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Determine next delivery attempt. |  | ||||||
| 		backoff := dlv.Request.BackOff() |  | ||||||
| 		dlv.next = time.Now().Add(backoff) |  | ||||||
| 
 |  | ||||||
| 		// Push to backlog. |  | ||||||
| 		w.pushBacklog(dlv) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // next gets the next available delivery, blocking until available if necessary. |  | ||||||
| func (w *Worker) next(ctx context.Context) (*Delivery, bool) { |  | ||||||
| loop: |  | ||||||
| 	for { |  | ||||||
| 		// Try pop next queued. |  | ||||||
| 		dlv, ok := w.Queue.Pop() |  | ||||||
| 
 |  | ||||||
| 		if !ok { |  | ||||||
| 			// Check the backlog. |  | ||||||
| 			if len(w.backlog) > 0 { |  | ||||||
| 
 |  | ||||||
| 				// Sort by 'next' time. |  | ||||||
| 				sortDeliveries(w.backlog) |  | ||||||
| 
 |  | ||||||
| 				// Pop next delivery. |  | ||||||
| 				dlv := w.popBacklog() |  | ||||||
| 
 |  | ||||||
| 				return dlv, true |  | ||||||
| 			} |  | ||||||
| 
 |  | ||||||
| 			select { |  | ||||||
| 			// Backlog is empty, we MUST |  | ||||||
| 			// block until next enqueued. |  | ||||||
| 			case <-w.Queue.Wait(): |  | ||||||
| 				continue loop |  | ||||||
| 
 |  | ||||||
| 			// Worker was stopped. |  | ||||||
| 			case <-ctx.Done(): |  | ||||||
| 				return nil, false |  | ||||||
| 			} |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Replace request context for worker state canceling. |  | ||||||
| 		ctx := gtscontext.WithValues(ctx, dlv.Request.Context()) |  | ||||||
| 		dlv.Request.Request = dlv.Request.Request.WithContext(ctx) |  | ||||||
| 
 |  | ||||||
| 		return dlv, true |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // popBacklog pops next available from the backlog. |  | ||||||
| func (w *Worker) popBacklog() *Delivery { |  | ||||||
| 	if len(w.backlog) == 0 { |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Pop from backlog. |  | ||||||
| 	dlv := w.backlog[0] |  | ||||||
| 
 |  | ||||||
| 	// Shift backlog down by one. |  | ||||||
| 	copy(w.backlog, w.backlog[1:]) |  | ||||||
| 	w.backlog = w.backlog[:len(w.backlog)-1] |  | ||||||
| 
 |  | ||||||
| 	return dlv |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // pushBacklog pushes the given delivery to backlog. |  | ||||||
| func (w *Worker) pushBacklog(dlv *Delivery) { |  | ||||||
| 	w.backlog = append(w.backlog, dlv) |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // sortDeliveries sorts deliveries according |  | ||||||
| // to when is the first requiring re-attempt. |  | ||||||
| func sortDeliveries(d []*Delivery) { |  | ||||||
| 	slices.SortFunc(d, func(a, b *Delivery) int { |  | ||||||
| 		const k = +1 |  | ||||||
| 		switch { |  | ||||||
| 		case a.next.Before(b.next): |  | ||||||
| 			return +k |  | ||||||
| 		case b.next.Before(a.next): |  | ||||||
| 			return -k |  | ||||||
| 		default: |  | ||||||
| 			return 0 |  | ||||||
| 		} |  | ||||||
| 	}) |  | ||||||
| } |  | ||||||
|  |  | ||||||
|  | @ -1,203 +1,134 @@ | ||||||
|  | // GoToSocial | ||||||
|  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||||
|  | // SPDX-License-Identifier: AGPL-3.0-or-later | ||||||
|  | // | ||||||
|  | // This program is free software: you can redistribute it and/or modify | ||||||
|  | // it under the terms of the GNU Affero General Public License as published by | ||||||
|  | // the Free Software Foundation, either version 3 of the License, or | ||||||
|  | // (at your option) any later version. | ||||||
|  | // | ||||||
|  | // This program is distributed in the hope that it will be useful, | ||||||
|  | // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||||
|  | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||||
|  | // GNU Affero General Public License for more details. | ||||||
|  | // | ||||||
|  | // You should have received a copy of the GNU Affero General Public License | ||||||
|  | // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||||
|  | 
 | ||||||
| package delivery_test | package delivery_test | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
| 	"fmt" | 	"bytes" | ||||||
|  | 	"encoding/json" | ||||||
| 	"io" | 	"io" | ||||||
| 	"math/rand" |  | ||||||
| 	"net" |  | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"strconv" |  | ||||||
| 	"strings" |  | ||||||
| 	"testing" | 	"testing" | ||||||
| 
 | 
 | ||||||
| 	"codeberg.org/gruf/go-byteutil" | 	"github.com/stretchr/testify/assert" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/config" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/queue" |  | ||||||
| 	"github.com/superseriousbusiness/gotosocial/internal/transport/delivery" | 	"github.com/superseriousbusiness/gotosocial/internal/transport/delivery" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| func TestDeliveryWorkerPool(t *testing.T) { | var deliveryCases = []struct { | ||||||
| 	for _, i := range []int{1, 2, 4, 8, 16, 32} { | 	msg  delivery.Delivery | ||||||
| 		t.Run("size="+strconv.Itoa(i), func(t *testing.T) { | 	data []byte | ||||||
| 			testDeliveryWorkerPool(t, i, generateInput(100*i)) | }{ | ||||||
| 		}) | 	{ | ||||||
| 	} | 		msg: delivery.Delivery{ | ||||||
| } | 			PubKeyID: "https://google.com/users/bigboy#pubkey", | ||||||
| 
 | 			ActorID:  "https://google.com/users/bigboy", | ||||||
| func testDeliveryWorkerPool(t *testing.T, sz int, input []*testrequest) { | 			ObjectID: "https://google.com/users/bigboy/follow/1", | ||||||
| 	wp := new(delivery.WorkerPool) | 			TargetID: "https://askjeeves.com/users/smallboy", | ||||||
| 	wp.Init(httpclient.New(httpclient.Config{ | 			Request:  toRequest("POST", "https://askjeeves.com/users/smallboy/inbox", []byte("data!")), | ||||||
| 		AllowRanges: config.MustParseIPPrefixes([]string{ | 		}, | ||||||
| 			"127.0.0.0/8", | 		data: toJSON(map[string]any{ | ||||||
|  | 			"pub_key_id": "https://google.com/users/bigboy#pubkey", | ||||||
|  | 			"actor_id":   "https://google.com/users/bigboy", | ||||||
|  | 			"object_id":  "https://google.com/users/bigboy/follow/1", | ||||||
|  | 			"target_id":  "https://askjeeves.com/users/smallboy", | ||||||
|  | 			"method":     "POST", | ||||||
|  | 			"url":        "https://askjeeves.com/users/smallboy/inbox", | ||||||
|  | 			"body":       []byte("data!"), | ||||||
|  | 			// "header":     map[string][]string{}, | ||||||
| 		}), | 		}), | ||||||
| 	})) | 	}, | ||||||
| 	wp.Start(sz) | 	{ | ||||||
| 	defer wp.Stop() | 		msg: delivery.Delivery{ | ||||||
| 	test(t, &wp.Queue, input) | 			Request: toRequest("GET", "https://google.com", []byte("uwu im just a wittle seawch engwin")), | ||||||
|  | 		}, | ||||||
|  | 		data: toJSON(map[string]any{ | ||||||
|  | 			"method": "GET", | ||||||
|  | 			"url":    "https://google.com", | ||||||
|  | 			"body":   []byte("uwu im just a wittle seawch engwin"), | ||||||
|  | 			// "header":     map[string][]string{}, | ||||||
|  | 		}), | ||||||
|  | 	}, | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| func test( | func TestSerializeDelivery(t *testing.T) { | ||||||
| 	t *testing.T, | 	for _, test := range deliveryCases { | ||||||
| 	queue *queue.StructQueue[*delivery.Delivery], | 		// Serialize test message to blob. | ||||||
| 	input []*testrequest, | 		data, err := test.msg.Serialize() | ||||||
| ) { |  | ||||||
| 	expect := make(chan *testrequest) |  | ||||||
| 	errors := make(chan error) |  | ||||||
| 
 |  | ||||||
| 	// Prepare an HTTP test handler that ensures expected delivery is received. |  | ||||||
| 	handler := http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { |  | ||||||
| 		errors <- (<-expect).Equal(r) |  | ||||||
| 	}) |  | ||||||
| 
 |  | ||||||
| 	// Start new HTTP test server listener. |  | ||||||
| 	l, err := net.Listen("tcp", "127.0.0.1:0") |  | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			t.Fatal(err) | 			t.Fatal(err) | ||||||
| 		} | 		} | ||||||
| 	defer l.Close() |  | ||||||
| 
 | 
 | ||||||
| 	// Start the HTTP server. | 		// Check that serialized JSON data is as expected. | ||||||
| 	// | 		assert.JSONEq(t, string(test.data), string(data)) | ||||||
| 	// specifically not using httptest.Server{} here as httptest |  | ||||||
| 	// links that server with its own http.Client{}, whereas we're |  | ||||||
| 	// using an httpclient.Client{} (well, delivery routine is). |  | ||||||
| 	srv := new(http.Server) |  | ||||||
| 	srv.Addr = "http://" + l.Addr().String() |  | ||||||
| 	srv.Handler = handler |  | ||||||
| 	go srv.Serve(l) |  | ||||||
| 	defer srv.Close() |  | ||||||
| 
 |  | ||||||
| 	// Range over test input. |  | ||||||
| 	for _, test := range input { |  | ||||||
| 
 |  | ||||||
| 		// Generate req for input. |  | ||||||
| 		req := test.Generate(srv.Addr) |  | ||||||
| 		r := httpclient.WrapRequest(req) |  | ||||||
| 
 |  | ||||||
| 		// Wrap the request in delivery. |  | ||||||
| 		dlv := new(delivery.Delivery) |  | ||||||
| 		dlv.Request = r |  | ||||||
| 
 |  | ||||||
| 		// Enqueue delivery! |  | ||||||
| 		queue.Push(dlv) |  | ||||||
| 		expect <- test |  | ||||||
| 
 |  | ||||||
| 		// Wait for errors from handler. |  | ||||||
| 		if err := <-errors; err != nil { |  | ||||||
| 			t.Error(err) |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| type testrequest struct { | func TestDeserializeDelivery(t *testing.T) { | ||||||
| 	method string | 	for _, test := range deliveryCases { | ||||||
| 	uri    string | 		var msg delivery.Delivery | ||||||
| 	body   []byte |  | ||||||
| } |  | ||||||
| 
 | 
 | ||||||
| // generateInput generates 'n' many testrequest cases. | 		// Deserialize test message blob. | ||||||
| func generateInput(n int) []*testrequest { | 		err := msg.Deserialize(test.data) | ||||||
| 	tests := make([]*testrequest, n) | 		if err != nil { | ||||||
| 	for i := range tests { | 			t.Fatal(err) | ||||||
| 		tests[i] = new(testrequest) |  | ||||||
| 		tests[i].method = randomMethod() |  | ||||||
| 		tests[i].uri = randomURI() |  | ||||||
| 		tests[i].body = randomBody(tests[i].method) |  | ||||||
| 		} | 		} | ||||||
| 	return tests |  | ||||||
| } |  | ||||||
| 
 | 
 | ||||||
| var methods = []string{ | 		// Check that delivery fields are as expected. | ||||||
| 	http.MethodConnect, | 		assert.Equal(t, test.msg.PubKeyID, msg.PubKeyID) | ||||||
| 	http.MethodDelete, | 		assert.Equal(t, test.msg.ActorID, msg.ActorID) | ||||||
| 	http.MethodGet, | 		assert.Equal(t, test.msg.ObjectID, msg.ObjectID) | ||||||
| 	http.MethodHead, | 		assert.Equal(t, test.msg.TargetID, msg.TargetID) | ||||||
| 	http.MethodOptions, | 		assert.Equal(t, test.msg.Request.Method, msg.Request.Method) | ||||||
| 	http.MethodPatch, | 		assert.Equal(t, test.msg.Request.URL, msg.Request.URL) | ||||||
| 	http.MethodPost, | 		assert.Equal(t, readBody(test.msg.Request.Body), readBody(msg.Request.Body)) | ||||||
| 	http.MethodPut, |  | ||||||
| 	http.MethodTrace, |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // randomMethod generates a random http method. |  | ||||||
| func randomMethod() string { |  | ||||||
| 	return methods[rand.Intn(len(methods))] |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // randomURI generates a random http uri. |  | ||||||
| func randomURI() string { |  | ||||||
| 	n := rand.Intn(5) |  | ||||||
| 	p := make([]string, n) |  | ||||||
| 	for i := range p { |  | ||||||
| 		p[i] = strconv.Itoa(rand.Int()) |  | ||||||
| 	} |  | ||||||
| 	return "/" + strings.Join(p, "/") |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // randomBody generates a random http body DEPENDING on method. |  | ||||||
| func randomBody(method string) []byte { |  | ||||||
| 	if requiresBody(method) { |  | ||||||
| 		return []byte(method + " " + randomURI()) |  | ||||||
| 	} |  | ||||||
| 	return nil |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| // requiresBody returns whether method requires body. |  | ||||||
| func requiresBody(method string) bool { |  | ||||||
| 	switch method { |  | ||||||
| 	case http.MethodPatch, |  | ||||||
| 		http.MethodPost, |  | ||||||
| 		http.MethodPut: |  | ||||||
| 		return true |  | ||||||
| 	default: |  | ||||||
| 		return false |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // Generate will generate a real http.Request{} from test data. | // toRequest creates httpclient.Request from HTTP method, URL and body data. | ||||||
| func (t *testrequest) Generate(addr string) *http.Request { | func toRequest(method string, url string, body []byte) httpclient.Request { | ||||||
| 	var body io.ReadCloser | 	var rbody io.Reader | ||||||
| 	if t.body != nil { | 	if body != nil { | ||||||
| 		var b byteutil.ReadNopCloser | 		rbody = bytes.NewReader(body) | ||||||
| 		b.Reset(t.body) |  | ||||||
| 		body = &b |  | ||||||
| 	} | 	} | ||||||
| 	req, err := http.NewRequest(t.method, addr+t.uri, body) | 	req, err := http.NewRequest(method, url, rbody) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		panic(err) | 		panic(err) | ||||||
| 	} | 	} | ||||||
| 	return req | 	return httpclient.WrapRequest(req) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // Equal checks if request matches receiving test request. | // readBody reads the content of body io.ReadCloser into memory as byte slice. | ||||||
| func (t *testrequest) Equal(r *http.Request) error { | func readBody(r io.ReadCloser) []byte { | ||||||
| 	// Ensure methods match. | 	if r == nil { | ||||||
| 	if t.method != r.Method { |  | ||||||
| 		return fmt.Errorf("differing request methods: t=%q r=%q", t.method, r.Method) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Ensure request URIs match. |  | ||||||
| 	if t.uri != r.URL.RequestURI() { |  | ||||||
| 		return fmt.Errorf("differing request urls: t=%q r=%q", t.uri, r.URL.RequestURI()) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	// Ensure body cases match. |  | ||||||
| 	if requiresBody(t.method) { |  | ||||||
| 
 |  | ||||||
| 		// Read request into memory. |  | ||||||
| 		b, err := io.ReadAll(r.Body) |  | ||||||
| 		if err != nil { |  | ||||||
| 			return fmt.Errorf("error reading request body: %v", err) |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		// Compare the request bodies. |  | ||||||
| 		st := strings.TrimSpace(string(t.body)) |  | ||||||
| 		sr := strings.TrimSpace(string(b)) |  | ||||||
| 		if st != sr { |  | ||||||
| 			return fmt.Errorf("differing request bodies: t=%q r=%q", st, sr) |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 		return nil | 		return nil | ||||||
|  | 	} | ||||||
|  | 	b, err := io.ReadAll(r) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // toJSON marshals input type as JSON data. | ||||||
|  | func toJSON(a any) []byte { | ||||||
|  | 	b, err := json.Marshal(a) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
| } | } | ||||||
|  |  | ||||||
							
								
								
									
										298
									
								
								internal/transport/delivery/worker.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										298
									
								
								internal/transport/delivery/worker.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,298 @@ | ||||||
|  | // GoToSocial | ||||||
|  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||||
|  | // SPDX-License-Identifier: AGPL-3.0-or-later | ||||||
|  | // | ||||||
|  | // This program is free software: you can redistribute it and/or modify | ||||||
|  | // it under the terms of the GNU Affero General Public License as published by | ||||||
|  | // the Free Software Foundation, either version 3 of the License, or | ||||||
|  | // (at your option) any later version. | ||||||
|  | // | ||||||
|  | // This program is distributed in the hope that it will be useful, | ||||||
|  | // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||||
|  | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||||
|  | // GNU Affero General Public License for more details. | ||||||
|  | // | ||||||
|  | // You should have received a copy of the GNU Affero General Public License | ||||||
|  | // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||||
|  | 
 | ||||||
|  | package delivery | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"slices" | ||||||
|  | 	"time" | ||||||
|  | 
 | ||||||
|  | 	"codeberg.org/gruf/go-runners" | ||||||
|  | 	"codeberg.org/gruf/go-structr" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/log" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/queue" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/util" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // WorkerPool wraps multiple Worker{}s in | ||||||
|  | // a singular struct for easy multi start/stop. | ||||||
|  | type WorkerPool struct { | ||||||
|  | 
 | ||||||
|  | 	// Client defines httpclient.Client{} | ||||||
|  | 	// passed to each of delivery pool Worker{}s. | ||||||
|  | 	Client *httpclient.Client | ||||||
|  | 
 | ||||||
|  | 	// Queue is the embedded queue.StructQueue{} | ||||||
|  | 	// passed to each of delivery pool Worker{}s. | ||||||
|  | 	Queue queue.StructQueue[*Delivery] | ||||||
|  | 
 | ||||||
|  | 	// internal fields. | ||||||
|  | 	workers []*Worker | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Init will initialize the Worker{} pool | ||||||
|  | // with given http client, request queue to pull | ||||||
|  | // from and number of delivery workers to spawn. | ||||||
|  | func (p *WorkerPool) Init(client *httpclient.Client) { | ||||||
|  | 	p.Client = client | ||||||
|  | 	p.Queue.Init(structr.QueueConfig[*Delivery]{ | ||||||
|  | 		Indices: []structr.IndexConfig{ | ||||||
|  | 			{Fields: "ActorID", Multiple: true}, | ||||||
|  | 			{Fields: "ObjectID", Multiple: true}, | ||||||
|  | 			{Fields: "TargetID", Multiple: true}, | ||||||
|  | 		}, | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Start will attempt to start 'n' Worker{}s. | ||||||
|  | func (p *WorkerPool) Start(n int) { | ||||||
|  | 	// Check whether workers are | ||||||
|  | 	// set (is already running). | ||||||
|  | 	ok := (len(p.workers) > 0) | ||||||
|  | 	if ok { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Allocate new workers slice. | ||||||
|  | 	p.workers = make([]*Worker, n) | ||||||
|  | 	for i := range p.workers { | ||||||
|  | 
 | ||||||
|  | 		// Allocate new Worker{}. | ||||||
|  | 		p.workers[i] = new(Worker) | ||||||
|  | 		p.workers[i].Client = p.Client | ||||||
|  | 		p.workers[i].Queue = &p.Queue | ||||||
|  | 
 | ||||||
|  | 		// Attempt to start worker. | ||||||
|  | 		// Return bool not useful | ||||||
|  | 		// here, as true = started, | ||||||
|  | 		// false = already running. | ||||||
|  | 		_ = p.workers[i].Start() | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Stop will attempt to stop contained Worker{}s. | ||||||
|  | func (p *WorkerPool) Stop() { | ||||||
|  | 	// Check whether workers are | ||||||
|  | 	// set (is currently running). | ||||||
|  | 	ok := (len(p.workers) == 0) | ||||||
|  | 	if ok { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Stop all running workers. | ||||||
|  | 	for i := range p.workers { | ||||||
|  | 
 | ||||||
|  | 		// return bool not useful | ||||||
|  | 		// here, as true = stopped, | ||||||
|  | 		// false = never running. | ||||||
|  | 		_ = p.workers[i].Stop() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Unset workers slice. | ||||||
|  | 	p.workers = p.workers[:0] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Worker wraps an httpclient.Client{} to feed | ||||||
|  | // from queue.StructQueue{} for ActivityPub reqs | ||||||
|  | // to deliver. It does so while prioritizing new | ||||||
|  | // queued requests over backlogged retries. | ||||||
|  | type Worker struct { | ||||||
|  | 
 | ||||||
|  | 	// Client is the httpclient.Client{} that | ||||||
|  | 	// delivery worker will use for requests. | ||||||
|  | 	Client *httpclient.Client | ||||||
|  | 
 | ||||||
|  | 	// Queue is the Delivery{} message queue | ||||||
|  | 	// that delivery worker will feed from. | ||||||
|  | 	Queue *queue.StructQueue[*Delivery] | ||||||
|  | 
 | ||||||
|  | 	// internal fields. | ||||||
|  | 	backlog []*Delivery | ||||||
|  | 	service runners.Service | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Start will attempt to start the Worker{}. | ||||||
|  | func (w *Worker) Start() bool { | ||||||
|  | 	return w.service.GoRun(w.run) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Stop will attempt to stop the Worker{}. | ||||||
|  | func (w *Worker) Stop() bool { | ||||||
|  | 	return w.service.Stop() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // run wraps process to restart on any panic. | ||||||
|  | func (w *Worker) run(ctx context.Context) { | ||||||
|  | 	if w.Client == nil || w.Queue == nil { | ||||||
|  | 		panic("not yet initialized") | ||||||
|  | 	} | ||||||
|  | 	log.Infof(ctx, "%p: starting worker", w) | ||||||
|  | 	defer log.Infof(ctx, "%p: stopped worker", w) | ||||||
|  | 	util.Must(func() { w.process(ctx) }) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // process is the main delivery worker processing routine. | ||||||
|  | func (w *Worker) process(ctx context.Context) bool { | ||||||
|  | 	if w.Client == nil || w.Queue == nil { | ||||||
|  | 		// we perform this check here just | ||||||
|  | 		// to ensure the compiler knows these | ||||||
|  | 		// variables aren't nil in the loop, | ||||||
|  | 		// even if already checked by caller. | ||||||
|  | 		panic("not yet initialized") | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | loop: | ||||||
|  | 	for { | ||||||
|  | 		// Get next delivery. | ||||||
|  | 		dlv, ok := w.next(ctx) | ||||||
|  | 		if !ok { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check whether backoff required. | ||||||
|  | 		const min = 100 * time.Millisecond | ||||||
|  | 		if d := dlv.backoff(); d > min { | ||||||
|  | 
 | ||||||
|  | 			// Start backoff sleep timer. | ||||||
|  | 			backoff := time.NewTimer(d) | ||||||
|  | 
 | ||||||
|  | 			select { | ||||||
|  | 			case <-ctx.Done(): | ||||||
|  | 				// Main ctx | ||||||
|  | 				// cancelled. | ||||||
|  | 				backoff.Stop() | ||||||
|  | 				return true | ||||||
|  | 
 | ||||||
|  | 			case <-w.Queue.Wait(): | ||||||
|  | 				// A new message was | ||||||
|  | 				// queued, re-add this | ||||||
|  | 				// to backlog + retry. | ||||||
|  | 				w.pushBacklog(dlv) | ||||||
|  | 				backoff.Stop() | ||||||
|  | 				continue loop | ||||||
|  | 
 | ||||||
|  | 			case <-backoff.C: | ||||||
|  | 				// success! | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Attempt delivery of AP request. | ||||||
|  | 		rsp, retry, err := w.Client.DoOnce( | ||||||
|  | 			&dlv.Request, | ||||||
|  | 		) | ||||||
|  | 
 | ||||||
|  | 		if err == nil { | ||||||
|  | 			// Ensure body closed. | ||||||
|  | 			_ = rsp.Body.Close() | ||||||
|  | 			continue loop | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		if !retry { | ||||||
|  | 			// Drop deliveries when no | ||||||
|  | 			// retry requested, or they | ||||||
|  | 			// reached max (either). | ||||||
|  | 			continue loop | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Determine next delivery attempt. | ||||||
|  | 		backoff := dlv.Request.BackOff() | ||||||
|  | 		dlv.next = time.Now().Add(backoff) | ||||||
|  | 
 | ||||||
|  | 		// Push to backlog. | ||||||
|  | 		w.pushBacklog(dlv) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // next gets the next available delivery, blocking until available if necessary. | ||||||
|  | func (w *Worker) next(ctx context.Context) (*Delivery, bool) { | ||||||
|  | loop: | ||||||
|  | 	for { | ||||||
|  | 		// Try pop next queued. | ||||||
|  | 		dlv, ok := w.Queue.Pop() | ||||||
|  | 
 | ||||||
|  | 		if !ok { | ||||||
|  | 			// Check the backlog. | ||||||
|  | 			if len(w.backlog) > 0 { | ||||||
|  | 
 | ||||||
|  | 				// Sort by 'next' time. | ||||||
|  | 				sortDeliveries(w.backlog) | ||||||
|  | 
 | ||||||
|  | 				// Pop next delivery. | ||||||
|  | 				dlv := w.popBacklog() | ||||||
|  | 
 | ||||||
|  | 				return dlv, true | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			select { | ||||||
|  | 			// Backlog is empty, we MUST | ||||||
|  | 			// block until next enqueued. | ||||||
|  | 			case <-w.Queue.Wait(): | ||||||
|  | 				continue loop | ||||||
|  | 
 | ||||||
|  | 			// Worker was stopped. | ||||||
|  | 			case <-ctx.Done(): | ||||||
|  | 				return nil, false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Replace request context for worker state canceling. | ||||||
|  | 		ctx := gtscontext.WithValues(ctx, dlv.Request.Context()) | ||||||
|  | 		dlv.Request.Request = dlv.Request.Request.WithContext(ctx) | ||||||
|  | 
 | ||||||
|  | 		return dlv, true | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // popBacklog pops next available from the backlog. | ||||||
|  | func (w *Worker) popBacklog() *Delivery { | ||||||
|  | 	if len(w.backlog) == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Pop from backlog. | ||||||
|  | 	dlv := w.backlog[0] | ||||||
|  | 
 | ||||||
|  | 	// Shift backlog down by one. | ||||||
|  | 	copy(w.backlog, w.backlog[1:]) | ||||||
|  | 	w.backlog = w.backlog[:len(w.backlog)-1] | ||||||
|  | 
 | ||||||
|  | 	return dlv | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // pushBacklog pushes the given delivery to backlog. | ||||||
|  | func (w *Worker) pushBacklog(dlv *Delivery) { | ||||||
|  | 	w.backlog = append(w.backlog, dlv) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // sortDeliveries sorts deliveries according | ||||||
|  | // to when is the first requiring re-attempt. | ||||||
|  | func sortDeliveries(d []*Delivery) { | ||||||
|  | 	slices.SortFunc(d, func(a, b *Delivery) int { | ||||||
|  | 		const k = +1 | ||||||
|  | 		switch { | ||||||
|  | 		case a.next.Before(b.next): | ||||||
|  | 			return +k | ||||||
|  | 		case b.next.Before(a.next): | ||||||
|  | 			return -k | ||||||
|  | 		default: | ||||||
|  | 			return 0 | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  | } | ||||||
							
								
								
									
										220
									
								
								internal/transport/delivery/worker_test.go
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										220
									
								
								internal/transport/delivery/worker_test.go
									
										
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,220 @@ | ||||||
|  | // GoToSocial | ||||||
|  | // Copyright (C) GoToSocial Authors admin@gotosocial.org | ||||||
|  | // SPDX-License-Identifier: AGPL-3.0-or-later | ||||||
|  | // | ||||||
|  | // This program is free software: you can redistribute it and/or modify | ||||||
|  | // it under the terms of the GNU Affero General Public License as published by | ||||||
|  | // the Free Software Foundation, either version 3 of the License, or | ||||||
|  | // (at your option) any later version. | ||||||
|  | // | ||||||
|  | // This program is distributed in the hope that it will be useful, | ||||||
|  | // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||||
|  | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | ||||||
|  | // GNU Affero General Public License for more details. | ||||||
|  | // | ||||||
|  | // You should have received a copy of the GNU Affero General Public License | ||||||
|  | // along with this program.  If not, see <http://www.gnu.org/licenses/>. | ||||||
|  | 
 | ||||||
|  | package delivery_test | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"io" | ||||||
|  | 	"math/rand" | ||||||
|  | 	"net" | ||||||
|  | 	"net/http" | ||||||
|  | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  | 	"testing" | ||||||
|  | 
 | ||||||
|  | 	"codeberg.org/gruf/go-byteutil" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/config" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/httpclient" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/queue" | ||||||
|  | 	"github.com/superseriousbusiness/gotosocial/internal/transport/delivery" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func TestDeliveryWorkerPool(t *testing.T) { | ||||||
|  | 	for _, i := range []int{1, 2, 4, 8, 16, 32} { | ||||||
|  | 		t.Run("size="+strconv.Itoa(i), func(t *testing.T) { | ||||||
|  | 			testDeliveryWorkerPool(t, i, generateInput(100*i)) | ||||||
|  | 		}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func testDeliveryWorkerPool(t *testing.T, sz int, input []*testrequest) { | ||||||
|  | 	wp := new(delivery.WorkerPool) | ||||||
|  | 	wp.Init(httpclient.New(httpclient.Config{ | ||||||
|  | 		AllowRanges: config.MustParseIPPrefixes([]string{ | ||||||
|  | 			"127.0.0.0/8", | ||||||
|  | 		}), | ||||||
|  | 	})) | ||||||
|  | 	wp.Start(sz) | ||||||
|  | 	defer wp.Stop() | ||||||
|  | 	test(t, &wp.Queue, input) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func test( | ||||||
|  | 	t *testing.T, | ||||||
|  | 	queue *queue.StructQueue[*delivery.Delivery], | ||||||
|  | 	input []*testrequest, | ||||||
|  | ) { | ||||||
|  | 	expect := make(chan *testrequest) | ||||||
|  | 	errors := make(chan error) | ||||||
|  | 
 | ||||||
|  | 	// Prepare an HTTP test handler that ensures expected delivery is received. | ||||||
|  | 	handler := http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { | ||||||
|  | 		errors <- (<-expect).Equal(r) | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	// Start new HTTP test server listener. | ||||||
|  | 	l, err := net.Listen("tcp", "127.0.0.1:0") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatal(err) | ||||||
|  | 	} | ||||||
|  | 	defer l.Close() | ||||||
|  | 
 | ||||||
|  | 	// Start the HTTP server. | ||||||
|  | 	// | ||||||
|  | 	// specifically not using httptest.Server{} here as httptest | ||||||
|  | 	// links that server with its own http.Client{}, whereas we're | ||||||
|  | 	// using an httpclient.Client{} (well, delivery routine is). | ||||||
|  | 	srv := new(http.Server) | ||||||
|  | 	srv.Addr = "http://" + l.Addr().String() | ||||||
|  | 	srv.Handler = handler | ||||||
|  | 	go srv.Serve(l) | ||||||
|  | 	defer srv.Close() | ||||||
|  | 
 | ||||||
|  | 	// Range over test input. | ||||||
|  | 	for _, test := range input { | ||||||
|  | 
 | ||||||
|  | 		// Generate req for input. | ||||||
|  | 		req := test.Generate(srv.Addr) | ||||||
|  | 		r := httpclient.WrapRequest(req) | ||||||
|  | 
 | ||||||
|  | 		// Wrap the request in delivery. | ||||||
|  | 		dlv := new(delivery.Delivery) | ||||||
|  | 		dlv.Request = r | ||||||
|  | 
 | ||||||
|  | 		// Enqueue delivery! | ||||||
|  | 		queue.Push(dlv) | ||||||
|  | 		expect <- test | ||||||
|  | 
 | ||||||
|  | 		// Wait for errors from handler. | ||||||
|  | 		if err := <-errors; err != nil { | ||||||
|  | 			t.Error(err) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type testrequest struct { | ||||||
|  | 	method string | ||||||
|  | 	uri    string | ||||||
|  | 	body   []byte | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // generateInput generates 'n' many testrequest cases. | ||||||
|  | func generateInput(n int) []*testrequest { | ||||||
|  | 	tests := make([]*testrequest, n) | ||||||
|  | 	for i := range tests { | ||||||
|  | 		tests[i] = new(testrequest) | ||||||
|  | 		tests[i].method = randomMethod() | ||||||
|  | 		tests[i].uri = randomURI() | ||||||
|  | 		tests[i].body = randomBody(tests[i].method) | ||||||
|  | 	} | ||||||
|  | 	return tests | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | var methods = []string{ | ||||||
|  | 	http.MethodConnect, | ||||||
|  | 	http.MethodDelete, | ||||||
|  | 	http.MethodGet, | ||||||
|  | 	http.MethodHead, | ||||||
|  | 	http.MethodOptions, | ||||||
|  | 	http.MethodPatch, | ||||||
|  | 	http.MethodPost, | ||||||
|  | 	http.MethodPut, | ||||||
|  | 	http.MethodTrace, | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // randomMethod generates a random http method. | ||||||
|  | func randomMethod() string { | ||||||
|  | 	return methods[rand.Intn(len(methods))] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // randomURI generates a random http uri. | ||||||
|  | func randomURI() string { | ||||||
|  | 	n := rand.Intn(5) | ||||||
|  | 	p := make([]string, n) | ||||||
|  | 	for i := range p { | ||||||
|  | 		p[i] = strconv.Itoa(rand.Int()) | ||||||
|  | 	} | ||||||
|  | 	return "/" + strings.Join(p, "/") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // randomBody generates a random http body DEPENDING on method. | ||||||
|  | func randomBody(method string) []byte { | ||||||
|  | 	if requiresBody(method) { | ||||||
|  | 		return []byte(method + " " + randomURI()) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // requiresBody returns whether method requires body. | ||||||
|  | func requiresBody(method string) bool { | ||||||
|  | 	switch method { | ||||||
|  | 	case http.MethodPatch, | ||||||
|  | 		http.MethodPost, | ||||||
|  | 		http.MethodPut: | ||||||
|  | 		return true | ||||||
|  | 	default: | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Generate will generate a real http.Request{} from test data. | ||||||
|  | func (t *testrequest) Generate(addr string) *http.Request { | ||||||
|  | 	var body io.ReadCloser | ||||||
|  | 	if t.body != nil { | ||||||
|  | 		var b byteutil.ReadNopCloser | ||||||
|  | 		b.Reset(t.body) | ||||||
|  | 		body = &b | ||||||
|  | 	} | ||||||
|  | 	req, err := http.NewRequest(t.method, addr+t.uri, body) | ||||||
|  | 	if err != nil { | ||||||
|  | 		panic(err) | ||||||
|  | 	} | ||||||
|  | 	return req | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Equal checks if request matches receiving test request. | ||||||
|  | func (t *testrequest) Equal(r *http.Request) error { | ||||||
|  | 	// Ensure methods match. | ||||||
|  | 	if t.method != r.Method { | ||||||
|  | 		return fmt.Errorf("differing request methods: t=%q r=%q", t.method, r.Method) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Ensure request URIs match. | ||||||
|  | 	if t.uri != r.URL.RequestURI() { | ||||||
|  | 		return fmt.Errorf("differing request urls: t=%q r=%q", t.uri, r.URL.RequestURI()) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Ensure body cases match. | ||||||
|  | 	if requiresBody(t.method) { | ||||||
|  | 
 | ||||||
|  | 		// Read request into memory. | ||||||
|  | 		b, err := io.ReadAll(r.Body) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return fmt.Errorf("error reading request body: %v", err) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Compare the request bodies. | ||||||
|  | 		st := strings.TrimSpace(string(t.body)) | ||||||
|  | 		sr := strings.TrimSpace(string(b)) | ||||||
|  | 		if st != sr { | ||||||
|  | 			return fmt.Errorf("differing request bodies: t=%q r=%q", st, sr) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
							
								
								
									
										27
									
								
								vendor/github.com/google/go-cmp/LICENSE
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								vendor/github.com/google/go-cmp/LICENSE
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,27 @@ | ||||||
|  | Copyright (c) 2017 The Go Authors. All rights reserved. | ||||||
|  | 
 | ||||||
|  | Redistribution and use in source and binary forms, with or without | ||||||
|  | modification, are permitted provided that the following conditions are | ||||||
|  | met: | ||||||
|  | 
 | ||||||
|  |    * Redistributions of source code must retain the above copyright | ||||||
|  | notice, this list of conditions and the following disclaimer. | ||||||
|  |    * Redistributions in binary form must reproduce the above | ||||||
|  | copyright notice, this list of conditions and the following disclaimer | ||||||
|  | in the documentation and/or other materials provided with the | ||||||
|  | distribution. | ||||||
|  |    * Neither the name of Google Inc. nor the names of its | ||||||
|  | contributors may be used to endorse or promote products derived from | ||||||
|  | this software without specific prior written permission. | ||||||
|  | 
 | ||||||
|  | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | ||||||
|  | "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | ||||||
|  | LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | ||||||
|  | A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | ||||||
|  | OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | ||||||
|  | SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | ||||||
|  | LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | ||||||
|  | DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | ||||||
|  | THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | ||||||
|  | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | ||||||
|  | OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||||||
							
								
								
									
										671
									
								
								vendor/github.com/google/go-cmp/cmp/compare.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										671
									
								
								vendor/github.com/google/go-cmp/cmp/compare.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,671 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | // Package cmp determines equality of values. | ||||||
|  | // | ||||||
|  | // This package is intended to be a more powerful and safer alternative to | ||||||
|  | // [reflect.DeepEqual] for comparing whether two values are semantically equal. | ||||||
|  | // It is intended to only be used in tests, as performance is not a goal and | ||||||
|  | // it may panic if it cannot compare the values. Its propensity towards | ||||||
|  | // panicking means that its unsuitable for production environments where a | ||||||
|  | // spurious panic may be fatal. | ||||||
|  | // | ||||||
|  | // The primary features of cmp are: | ||||||
|  | // | ||||||
|  | //   - When the default behavior of equality does not suit the test's needs, | ||||||
|  | //     custom equality functions can override the equality operation. | ||||||
|  | //     For example, an equality function may report floats as equal so long as | ||||||
|  | //     they are within some tolerance of each other. | ||||||
|  | // | ||||||
|  | //   - Types with an Equal method (e.g., [time.Time.Equal]) may use that method | ||||||
|  | //     to determine equality. This allows package authors to determine | ||||||
|  | //     the equality operation for the types that they define. | ||||||
|  | // | ||||||
|  | //   - If no custom equality functions are used and no Equal method is defined, | ||||||
|  | //     equality is determined by recursively comparing the primitive kinds on | ||||||
|  | //     both values, much like [reflect.DeepEqual]. Unlike [reflect.DeepEqual], | ||||||
|  | //     unexported fields are not compared by default; they result in panics | ||||||
|  | //     unless suppressed by using an [Ignore] option | ||||||
|  | //     (see [github.com/google/go-cmp/cmp/cmpopts.IgnoreUnexported]) | ||||||
|  | //     or explicitly compared using the [Exporter] option. | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strings" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/diff" | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/function" | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/value" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // TODO(≥go1.18): Use any instead of interface{}. | ||||||
|  | 
 | ||||||
|  | // Equal reports whether x and y are equal by recursively applying the | ||||||
|  | // following rules in the given order to x and y and all of their sub-values: | ||||||
|  | // | ||||||
|  | //   - Let S be the set of all [Ignore], [Transformer], and [Comparer] options that | ||||||
|  | //     remain after applying all path filters, value filters, and type filters. | ||||||
|  | //     If at least one [Ignore] exists in S, then the comparison is ignored. | ||||||
|  | //     If the number of [Transformer] and [Comparer] options in S is non-zero, | ||||||
|  | //     then Equal panics because it is ambiguous which option to use. | ||||||
|  | //     If S contains a single [Transformer], then use that to transform | ||||||
|  | //     the current values and recursively call Equal on the output values. | ||||||
|  | //     If S contains a single [Comparer], then use that to compare the current values. | ||||||
|  | //     Otherwise, evaluation proceeds to the next rule. | ||||||
|  | // | ||||||
|  | //   - If the values have an Equal method of the form "(T) Equal(T) bool" or | ||||||
|  | //     "(T) Equal(I) bool" where T is assignable to I, then use the result of | ||||||
|  | //     x.Equal(y) even if x or y is nil. Otherwise, no such method exists and | ||||||
|  | //     evaluation proceeds to the next rule. | ||||||
|  | // | ||||||
|  | //   - Lastly, try to compare x and y based on their basic kinds. | ||||||
|  | //     Simple kinds like booleans, integers, floats, complex numbers, strings, | ||||||
|  | //     and channels are compared using the equivalent of the == operator in Go. | ||||||
|  | //     Functions are only equal if they are both nil, otherwise they are unequal. | ||||||
|  | // | ||||||
|  | // Structs are equal if recursively calling Equal on all fields report equal. | ||||||
|  | // If a struct contains unexported fields, Equal panics unless an [Ignore] option | ||||||
|  | // (e.g., [github.com/google/go-cmp/cmp/cmpopts.IgnoreUnexported]) ignores that field | ||||||
|  | // or the [Exporter] option explicitly permits comparing the unexported field. | ||||||
|  | // | ||||||
|  | // Slices are equal if they are both nil or both non-nil, where recursively | ||||||
|  | // calling Equal on all non-ignored slice or array elements report equal. | ||||||
|  | // Empty non-nil slices and nil slices are not equal; to equate empty slices, | ||||||
|  | // consider using [github.com/google/go-cmp/cmp/cmpopts.EquateEmpty]. | ||||||
|  | // | ||||||
|  | // Maps are equal if they are both nil or both non-nil, where recursively | ||||||
|  | // calling Equal on all non-ignored map entries report equal. | ||||||
|  | // Map keys are equal according to the == operator. | ||||||
|  | // To use custom comparisons for map keys, consider using | ||||||
|  | // [github.com/google/go-cmp/cmp/cmpopts.SortMaps]. | ||||||
|  | // Empty non-nil maps and nil maps are not equal; to equate empty maps, | ||||||
|  | // consider using [github.com/google/go-cmp/cmp/cmpopts.EquateEmpty]. | ||||||
|  | // | ||||||
|  | // Pointers and interfaces are equal if they are both nil or both non-nil, | ||||||
|  | // where they have the same underlying concrete type and recursively | ||||||
|  | // calling Equal on the underlying values reports equal. | ||||||
|  | // | ||||||
|  | // Before recursing into a pointer, slice element, or map, the current path | ||||||
|  | // is checked to detect whether the address has already been visited. | ||||||
|  | // If there is a cycle, then the pointed at values are considered equal | ||||||
|  | // only if both addresses were previously visited in the same path step. | ||||||
|  | func Equal(x, y interface{}, opts ...Option) bool { | ||||||
|  | 	s := newState(opts) | ||||||
|  | 	s.compareAny(rootStep(x, y)) | ||||||
|  | 	return s.result.Equal() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Diff returns a human-readable report of the differences between two values: | ||||||
|  | // y - x. It returns an empty string if and only if Equal returns true for the | ||||||
|  | // same input values and options. | ||||||
|  | // | ||||||
|  | // The output is displayed as a literal in pseudo-Go syntax. | ||||||
|  | // At the start of each line, a "-" prefix indicates an element removed from x, | ||||||
|  | // a "+" prefix to indicates an element added from y, and the lack of a prefix | ||||||
|  | // indicates an element common to both x and y. If possible, the output | ||||||
|  | // uses fmt.Stringer.String or error.Error methods to produce more humanly | ||||||
|  | // readable outputs. In such cases, the string is prefixed with either an | ||||||
|  | // 's' or 'e' character, respectively, to indicate that the method was called. | ||||||
|  | // | ||||||
|  | // Do not depend on this output being stable. If you need the ability to | ||||||
|  | // programmatically interpret the difference, consider using a custom Reporter. | ||||||
|  | func Diff(x, y interface{}, opts ...Option) string { | ||||||
|  | 	s := newState(opts) | ||||||
|  | 
 | ||||||
|  | 	// Optimization: If there are no other reporters, we can optimize for the | ||||||
|  | 	// common case where the result is equal (and thus no reported difference). | ||||||
|  | 	// This avoids the expensive construction of a difference tree. | ||||||
|  | 	if len(s.reporters) == 0 { | ||||||
|  | 		s.compareAny(rootStep(x, y)) | ||||||
|  | 		if s.result.Equal() { | ||||||
|  | 			return "" | ||||||
|  | 		} | ||||||
|  | 		s.result = diff.Result{} // Reset results | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	r := new(defaultReporter) | ||||||
|  | 	s.reporters = append(s.reporters, reporter{r}) | ||||||
|  | 	s.compareAny(rootStep(x, y)) | ||||||
|  | 	d := r.String() | ||||||
|  | 	if (d == "") != s.result.Equal() { | ||||||
|  | 		panic("inconsistent difference and equality results") | ||||||
|  | 	} | ||||||
|  | 	return d | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // rootStep constructs the first path step. If x and y have differing types, | ||||||
|  | // then they are stored within an empty interface type. | ||||||
|  | func rootStep(x, y interface{}) PathStep { | ||||||
|  | 	vx := reflect.ValueOf(x) | ||||||
|  | 	vy := reflect.ValueOf(y) | ||||||
|  | 
 | ||||||
|  | 	// If the inputs are different types, auto-wrap them in an empty interface | ||||||
|  | 	// so that they have the same parent type. | ||||||
|  | 	var t reflect.Type | ||||||
|  | 	if !vx.IsValid() || !vy.IsValid() || vx.Type() != vy.Type() { | ||||||
|  | 		t = anyType | ||||||
|  | 		if vx.IsValid() { | ||||||
|  | 			vvx := reflect.New(t).Elem() | ||||||
|  | 			vvx.Set(vx) | ||||||
|  | 			vx = vvx | ||||||
|  | 		} | ||||||
|  | 		if vy.IsValid() { | ||||||
|  | 			vvy := reflect.New(t).Elem() | ||||||
|  | 			vvy.Set(vy) | ||||||
|  | 			vy = vvy | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		t = vx.Type() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &pathStep{t, vx, vy} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type state struct { | ||||||
|  | 	// These fields represent the "comparison state". | ||||||
|  | 	// Calling statelessCompare must not result in observable changes to these. | ||||||
|  | 	result    diff.Result // The current result of comparison | ||||||
|  | 	curPath   Path        // The current path in the value tree | ||||||
|  | 	curPtrs   pointerPath // The current set of visited pointers | ||||||
|  | 	reporters []reporter  // Optional reporters | ||||||
|  | 
 | ||||||
|  | 	// recChecker checks for infinite cycles applying the same set of | ||||||
|  | 	// transformers upon the output of itself. | ||||||
|  | 	recChecker recChecker | ||||||
|  | 
 | ||||||
|  | 	// dynChecker triggers pseudo-random checks for option correctness. | ||||||
|  | 	// It is safe for statelessCompare to mutate this value. | ||||||
|  | 	dynChecker dynChecker | ||||||
|  | 
 | ||||||
|  | 	// These fields, once set by processOption, will not change. | ||||||
|  | 	exporters []exporter // List of exporters for structs with unexported fields | ||||||
|  | 	opts      Options    // List of all fundamental and filter options | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func newState(opts []Option) *state { | ||||||
|  | 	// Always ensure a validator option exists to validate the inputs. | ||||||
|  | 	s := &state{opts: Options{validator{}}} | ||||||
|  | 	s.curPtrs.Init() | ||||||
|  | 	s.processOption(Options(opts)) | ||||||
|  | 	return s | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) processOption(opt Option) { | ||||||
|  | 	switch opt := opt.(type) { | ||||||
|  | 	case nil: | ||||||
|  | 	case Options: | ||||||
|  | 		for _, o := range opt { | ||||||
|  | 			s.processOption(o) | ||||||
|  | 		} | ||||||
|  | 	case coreOption: | ||||||
|  | 		type filtered interface { | ||||||
|  | 			isFiltered() bool | ||||||
|  | 		} | ||||||
|  | 		if fopt, ok := opt.(filtered); ok && !fopt.isFiltered() { | ||||||
|  | 			panic(fmt.Sprintf("cannot use an unfiltered option: %v", opt)) | ||||||
|  | 		} | ||||||
|  | 		s.opts = append(s.opts, opt) | ||||||
|  | 	case exporter: | ||||||
|  | 		s.exporters = append(s.exporters, opt) | ||||||
|  | 	case reporter: | ||||||
|  | 		s.reporters = append(s.reporters, opt) | ||||||
|  | 	default: | ||||||
|  | 		panic(fmt.Sprintf("unknown option %T", opt)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // statelessCompare compares two values and returns the result. | ||||||
|  | // This function is stateless in that it does not alter the current result, | ||||||
|  | // or output to any registered reporters. | ||||||
|  | func (s *state) statelessCompare(step PathStep) diff.Result { | ||||||
|  | 	// We do not save and restore curPath and curPtrs because all of the | ||||||
|  | 	// compareX methods should properly push and pop from them. | ||||||
|  | 	// It is an implementation bug if the contents of the paths differ from | ||||||
|  | 	// when calling this function to when returning from it. | ||||||
|  | 
 | ||||||
|  | 	oldResult, oldReporters := s.result, s.reporters | ||||||
|  | 	s.result = diff.Result{} // Reset result | ||||||
|  | 	s.reporters = nil        // Remove reporters to avoid spurious printouts | ||||||
|  | 	s.compareAny(step) | ||||||
|  | 	res := s.result | ||||||
|  | 	s.result, s.reporters = oldResult, oldReporters | ||||||
|  | 	return res | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) compareAny(step PathStep) { | ||||||
|  | 	// Update the path stack. | ||||||
|  | 	s.curPath.push(step) | ||||||
|  | 	defer s.curPath.pop() | ||||||
|  | 	for _, r := range s.reporters { | ||||||
|  | 		r.PushStep(step) | ||||||
|  | 		defer r.PopStep() | ||||||
|  | 	} | ||||||
|  | 	s.recChecker.Check(s.curPath) | ||||||
|  | 
 | ||||||
|  | 	// Cycle-detection for slice elements (see NOTE in compareSlice). | ||||||
|  | 	t := step.Type() | ||||||
|  | 	vx, vy := step.Values() | ||||||
|  | 	if si, ok := step.(SliceIndex); ok && si.isSlice && vx.IsValid() && vy.IsValid() { | ||||||
|  | 		px, py := vx.Addr(), vy.Addr() | ||||||
|  | 		if eq, visited := s.curPtrs.Push(px, py); visited { | ||||||
|  | 			s.report(eq, reportByCycle) | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  | 		defer s.curPtrs.Pop(px, py) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Rule 1: Check whether an option applies on this node in the value tree. | ||||||
|  | 	if s.tryOptions(t, vx, vy) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Rule 2: Check whether the type has a valid Equal method. | ||||||
|  | 	if s.tryMethod(t, vx, vy) { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Rule 3: Compare based on the underlying kind. | ||||||
|  | 	switch t.Kind() { | ||||||
|  | 	case reflect.Bool: | ||||||
|  | 		s.report(vx.Bool() == vy.Bool(), 0) | ||||||
|  | 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: | ||||||
|  | 		s.report(vx.Int() == vy.Int(), 0) | ||||||
|  | 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: | ||||||
|  | 		s.report(vx.Uint() == vy.Uint(), 0) | ||||||
|  | 	case reflect.Float32, reflect.Float64: | ||||||
|  | 		s.report(vx.Float() == vy.Float(), 0) | ||||||
|  | 	case reflect.Complex64, reflect.Complex128: | ||||||
|  | 		s.report(vx.Complex() == vy.Complex(), 0) | ||||||
|  | 	case reflect.String: | ||||||
|  | 		s.report(vx.String() == vy.String(), 0) | ||||||
|  | 	case reflect.Chan, reflect.UnsafePointer: | ||||||
|  | 		s.report(vx.Pointer() == vy.Pointer(), 0) | ||||||
|  | 	case reflect.Func: | ||||||
|  | 		s.report(vx.IsNil() && vy.IsNil(), 0) | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		s.compareStruct(t, vx, vy) | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		s.compareSlice(t, vx, vy) | ||||||
|  | 	case reflect.Map: | ||||||
|  | 		s.compareMap(t, vx, vy) | ||||||
|  | 	case reflect.Ptr: | ||||||
|  | 		s.comparePtr(t, vx, vy) | ||||||
|  | 	case reflect.Interface: | ||||||
|  | 		s.compareInterface(t, vx, vy) | ||||||
|  | 	default: | ||||||
|  | 		panic(fmt.Sprintf("%v kind not handled", t.Kind())) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) tryOptions(t reflect.Type, vx, vy reflect.Value) bool { | ||||||
|  | 	// Evaluate all filters and apply the remaining options. | ||||||
|  | 	if opt := s.opts.filter(s, t, vx, vy); opt != nil { | ||||||
|  | 		opt.apply(s, vx, vy) | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) tryMethod(t reflect.Type, vx, vy reflect.Value) bool { | ||||||
|  | 	// Check if this type even has an Equal method. | ||||||
|  | 	m, ok := t.MethodByName("Equal") | ||||||
|  | 	if !ok || !function.IsType(m.Type, function.EqualAssignable) { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	eq := s.callTTBFunc(m.Func, vx, vy) | ||||||
|  | 	s.report(eq, reportByMethod) | ||||||
|  | 	return true | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) callTRFunc(f, v reflect.Value, step Transform) reflect.Value { | ||||||
|  | 	if !s.dynChecker.Next() { | ||||||
|  | 		return f.Call([]reflect.Value{v})[0] | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Run the function twice and ensure that we get the same results back. | ||||||
|  | 	// We run in goroutines so that the race detector (if enabled) can detect | ||||||
|  | 	// unsafe mutations to the input. | ||||||
|  | 	c := make(chan reflect.Value) | ||||||
|  | 	go detectRaces(c, f, v) | ||||||
|  | 	got := <-c | ||||||
|  | 	want := f.Call([]reflect.Value{v})[0] | ||||||
|  | 	if step.vx, step.vy = got, want; !s.statelessCompare(step).Equal() { | ||||||
|  | 		// To avoid false-positives with non-reflexive equality operations, | ||||||
|  | 		// we sanity check whether a value is equal to itself. | ||||||
|  | 		if step.vx, step.vy = want, want; !s.statelessCompare(step).Equal() { | ||||||
|  | 			return want | ||||||
|  | 		} | ||||||
|  | 		panic(fmt.Sprintf("non-deterministic function detected: %s", function.NameOf(f))) | ||||||
|  | 	} | ||||||
|  | 	return want | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) callTTBFunc(f, x, y reflect.Value) bool { | ||||||
|  | 	if !s.dynChecker.Next() { | ||||||
|  | 		return f.Call([]reflect.Value{x, y})[0].Bool() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Swapping the input arguments is sufficient to check that | ||||||
|  | 	// f is symmetric and deterministic. | ||||||
|  | 	// We run in goroutines so that the race detector (if enabled) can detect | ||||||
|  | 	// unsafe mutations to the input. | ||||||
|  | 	c := make(chan reflect.Value) | ||||||
|  | 	go detectRaces(c, f, y, x) | ||||||
|  | 	got := <-c | ||||||
|  | 	want := f.Call([]reflect.Value{x, y})[0].Bool() | ||||||
|  | 	if !got.IsValid() || got.Bool() != want { | ||||||
|  | 		panic(fmt.Sprintf("non-deterministic or non-symmetric function detected: %s", function.NameOf(f))) | ||||||
|  | 	} | ||||||
|  | 	return want | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func detectRaces(c chan<- reflect.Value, f reflect.Value, vs ...reflect.Value) { | ||||||
|  | 	var ret reflect.Value | ||||||
|  | 	defer func() { | ||||||
|  | 		recover() // Ignore panics, let the other call to f panic instead | ||||||
|  | 		c <- ret | ||||||
|  | 	}() | ||||||
|  | 	ret = f.Call(vs)[0] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) compareStruct(t reflect.Type, vx, vy reflect.Value) { | ||||||
|  | 	var addr bool | ||||||
|  | 	var vax, vay reflect.Value // Addressable versions of vx and vy | ||||||
|  | 
 | ||||||
|  | 	var mayForce, mayForceInit bool | ||||||
|  | 	step := StructField{&structField{}} | ||||||
|  | 	for i := 0; i < t.NumField(); i++ { | ||||||
|  | 		step.typ = t.Field(i).Type | ||||||
|  | 		step.vx = vx.Field(i) | ||||||
|  | 		step.vy = vy.Field(i) | ||||||
|  | 		step.name = t.Field(i).Name | ||||||
|  | 		step.idx = i | ||||||
|  | 		step.unexported = !isExported(step.name) | ||||||
|  | 		if step.unexported { | ||||||
|  | 			if step.name == "_" { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			// Defer checking of unexported fields until later to give an | ||||||
|  | 			// Ignore a chance to ignore the field. | ||||||
|  | 			if !vax.IsValid() || !vay.IsValid() { | ||||||
|  | 				// For retrieveUnexportedField to work, the parent struct must | ||||||
|  | 				// be addressable. Create a new copy of the values if | ||||||
|  | 				// necessary to make them addressable. | ||||||
|  | 				addr = vx.CanAddr() || vy.CanAddr() | ||||||
|  | 				vax = makeAddressable(vx) | ||||||
|  | 				vay = makeAddressable(vy) | ||||||
|  | 			} | ||||||
|  | 			if !mayForceInit { | ||||||
|  | 				for _, xf := range s.exporters { | ||||||
|  | 					mayForce = mayForce || xf(t) | ||||||
|  | 				} | ||||||
|  | 				mayForceInit = true | ||||||
|  | 			} | ||||||
|  | 			step.mayForce = mayForce | ||||||
|  | 			step.paddr = addr | ||||||
|  | 			step.pvx = vax | ||||||
|  | 			step.pvy = vay | ||||||
|  | 			step.field = t.Field(i) | ||||||
|  | 		} | ||||||
|  | 		s.compareAny(step) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) compareSlice(t reflect.Type, vx, vy reflect.Value) { | ||||||
|  | 	isSlice := t.Kind() == reflect.Slice | ||||||
|  | 	if isSlice && (vx.IsNil() || vy.IsNil()) { | ||||||
|  | 		s.report(vx.IsNil() && vy.IsNil(), 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// NOTE: It is incorrect to call curPtrs.Push on the slice header pointer | ||||||
|  | 	// since slices represents a list of pointers, rather than a single pointer. | ||||||
|  | 	// The pointer checking logic must be handled on a per-element basis | ||||||
|  | 	// in compareAny. | ||||||
|  | 	// | ||||||
|  | 	// A slice header (see reflect.SliceHeader) in Go is a tuple of a starting | ||||||
|  | 	// pointer P, a length N, and a capacity C. Supposing each slice element has | ||||||
|  | 	// a memory size of M, then the slice is equivalent to the list of pointers: | ||||||
|  | 	//	[P+i*M for i in range(N)] | ||||||
|  | 	// | ||||||
|  | 	// For example, v[:0] and v[:1] are slices with the same starting pointer, | ||||||
|  | 	// but they are clearly different values. Using the slice pointer alone | ||||||
|  | 	// violates the assumption that equal pointers implies equal values. | ||||||
|  | 
 | ||||||
|  | 	step := SliceIndex{&sliceIndex{pathStep: pathStep{typ: t.Elem()}, isSlice: isSlice}} | ||||||
|  | 	withIndexes := func(ix, iy int) SliceIndex { | ||||||
|  | 		if ix >= 0 { | ||||||
|  | 			step.vx, step.xkey = vx.Index(ix), ix | ||||||
|  | 		} else { | ||||||
|  | 			step.vx, step.xkey = reflect.Value{}, -1 | ||||||
|  | 		} | ||||||
|  | 		if iy >= 0 { | ||||||
|  | 			step.vy, step.ykey = vy.Index(iy), iy | ||||||
|  | 		} else { | ||||||
|  | 			step.vy, step.ykey = reflect.Value{}, -1 | ||||||
|  | 		} | ||||||
|  | 		return step | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Ignore options are able to ignore missing elements in a slice. | ||||||
|  | 	// However, detecting these reliably requires an optimal differencing | ||||||
|  | 	// algorithm, for which diff.Difference is not. | ||||||
|  | 	// | ||||||
|  | 	// Instead, we first iterate through both slices to detect which elements | ||||||
|  | 	// would be ignored if standing alone. The index of non-discarded elements | ||||||
|  | 	// are stored in a separate slice, which diffing is then performed on. | ||||||
|  | 	var indexesX, indexesY []int | ||||||
|  | 	var ignoredX, ignoredY []bool | ||||||
|  | 	for ix := 0; ix < vx.Len(); ix++ { | ||||||
|  | 		ignored := s.statelessCompare(withIndexes(ix, -1)).NumDiff == 0 | ||||||
|  | 		if !ignored { | ||||||
|  | 			indexesX = append(indexesX, ix) | ||||||
|  | 		} | ||||||
|  | 		ignoredX = append(ignoredX, ignored) | ||||||
|  | 	} | ||||||
|  | 	for iy := 0; iy < vy.Len(); iy++ { | ||||||
|  | 		ignored := s.statelessCompare(withIndexes(-1, iy)).NumDiff == 0 | ||||||
|  | 		if !ignored { | ||||||
|  | 			indexesY = append(indexesY, iy) | ||||||
|  | 		} | ||||||
|  | 		ignoredY = append(ignoredY, ignored) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Compute an edit-script for slices vx and vy (excluding ignored elements). | ||||||
|  | 	edits := diff.Difference(len(indexesX), len(indexesY), func(ix, iy int) diff.Result { | ||||||
|  | 		return s.statelessCompare(withIndexes(indexesX[ix], indexesY[iy])) | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	// Replay the ignore-scripts and the edit-script. | ||||||
|  | 	var ix, iy int | ||||||
|  | 	for ix < vx.Len() || iy < vy.Len() { | ||||||
|  | 		var e diff.EditType | ||||||
|  | 		switch { | ||||||
|  | 		case ix < len(ignoredX) && ignoredX[ix]: | ||||||
|  | 			e = diff.UniqueX | ||||||
|  | 		case iy < len(ignoredY) && ignoredY[iy]: | ||||||
|  | 			e = diff.UniqueY | ||||||
|  | 		default: | ||||||
|  | 			e, edits = edits[0], edits[1:] | ||||||
|  | 		} | ||||||
|  | 		switch e { | ||||||
|  | 		case diff.UniqueX: | ||||||
|  | 			s.compareAny(withIndexes(ix, -1)) | ||||||
|  | 			ix++ | ||||||
|  | 		case diff.UniqueY: | ||||||
|  | 			s.compareAny(withIndexes(-1, iy)) | ||||||
|  | 			iy++ | ||||||
|  | 		default: | ||||||
|  | 			s.compareAny(withIndexes(ix, iy)) | ||||||
|  | 			ix++ | ||||||
|  | 			iy++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) compareMap(t reflect.Type, vx, vy reflect.Value) { | ||||||
|  | 	if vx.IsNil() || vy.IsNil() { | ||||||
|  | 		s.report(vx.IsNil() && vy.IsNil(), 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Cycle-detection for maps. | ||||||
|  | 	if eq, visited := s.curPtrs.Push(vx, vy); visited { | ||||||
|  | 		s.report(eq, reportByCycle) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	defer s.curPtrs.Pop(vx, vy) | ||||||
|  | 
 | ||||||
|  | 	// We combine and sort the two map keys so that we can perform the | ||||||
|  | 	// comparisons in a deterministic order. | ||||||
|  | 	step := MapIndex{&mapIndex{pathStep: pathStep{typ: t.Elem()}}} | ||||||
|  | 	for _, k := range value.SortKeys(append(vx.MapKeys(), vy.MapKeys()...)) { | ||||||
|  | 		step.vx = vx.MapIndex(k) | ||||||
|  | 		step.vy = vy.MapIndex(k) | ||||||
|  | 		step.key = k | ||||||
|  | 		if !step.vx.IsValid() && !step.vy.IsValid() { | ||||||
|  | 			// It is possible for both vx and vy to be invalid if the | ||||||
|  | 			// key contained a NaN value in it. | ||||||
|  | 			// | ||||||
|  | 			// Even with the ability to retrieve NaN keys in Go 1.12, | ||||||
|  | 			// there still isn't a sensible way to compare the values since | ||||||
|  | 			// a NaN key may map to multiple unordered values. | ||||||
|  | 			// The most reasonable way to compare NaNs would be to compare the | ||||||
|  | 			// set of values. However, this is impossible to do efficiently | ||||||
|  | 			// since set equality is provably an O(n^2) operation given only | ||||||
|  | 			// an Equal function. If we had a Less function or Hash function, | ||||||
|  | 			// this could be done in O(n*log(n)) or O(n), respectively. | ||||||
|  | 			// | ||||||
|  | 			// Rather than adding complex logic to deal with NaNs, make it | ||||||
|  | 			// the user's responsibility to compare such obscure maps. | ||||||
|  | 			const help = "consider providing a Comparer to compare the map" | ||||||
|  | 			panic(fmt.Sprintf("%#v has map key with NaNs\n%s", s.curPath, help)) | ||||||
|  | 		} | ||||||
|  | 		s.compareAny(step) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) comparePtr(t reflect.Type, vx, vy reflect.Value) { | ||||||
|  | 	if vx.IsNil() || vy.IsNil() { | ||||||
|  | 		s.report(vx.IsNil() && vy.IsNil(), 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Cycle-detection for pointers. | ||||||
|  | 	if eq, visited := s.curPtrs.Push(vx, vy); visited { | ||||||
|  | 		s.report(eq, reportByCycle) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	defer s.curPtrs.Pop(vx, vy) | ||||||
|  | 
 | ||||||
|  | 	vx, vy = vx.Elem(), vy.Elem() | ||||||
|  | 	s.compareAny(Indirect{&indirect{pathStep{t.Elem(), vx, vy}}}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) compareInterface(t reflect.Type, vx, vy reflect.Value) { | ||||||
|  | 	if vx.IsNil() || vy.IsNil() { | ||||||
|  | 		s.report(vx.IsNil() && vy.IsNil(), 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	vx, vy = vx.Elem(), vy.Elem() | ||||||
|  | 	if vx.Type() != vy.Type() { | ||||||
|  | 		s.report(false, 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	s.compareAny(TypeAssertion{&typeAssertion{pathStep{vx.Type(), vx, vy}}}) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *state) report(eq bool, rf resultFlags) { | ||||||
|  | 	if rf&reportByIgnore == 0 { | ||||||
|  | 		if eq { | ||||||
|  | 			s.result.NumSame++ | ||||||
|  | 			rf |= reportEqual | ||||||
|  | 		} else { | ||||||
|  | 			s.result.NumDiff++ | ||||||
|  | 			rf |= reportUnequal | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for _, r := range s.reporters { | ||||||
|  | 		r.Report(Result{flags: rf}) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // recChecker tracks the state needed to periodically perform checks that | ||||||
|  | // user provided transformers are not stuck in an infinitely recursive cycle. | ||||||
|  | type recChecker struct{ next int } | ||||||
|  | 
 | ||||||
|  | // Check scans the Path for any recursive transformers and panics when any | ||||||
|  | // recursive transformers are detected. Note that the presence of a | ||||||
|  | // recursive Transformer does not necessarily imply an infinite cycle. | ||||||
|  | // As such, this check only activates after some minimal number of path steps. | ||||||
|  | func (rc *recChecker) Check(p Path) { | ||||||
|  | 	const minLen = 1 << 16 | ||||||
|  | 	if rc.next == 0 { | ||||||
|  | 		rc.next = minLen | ||||||
|  | 	} | ||||||
|  | 	if len(p) < rc.next { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	rc.next <<= 1 | ||||||
|  | 
 | ||||||
|  | 	// Check whether the same transformer has appeared at least twice. | ||||||
|  | 	var ss []string | ||||||
|  | 	m := map[Option]int{} | ||||||
|  | 	for _, ps := range p { | ||||||
|  | 		if t, ok := ps.(Transform); ok { | ||||||
|  | 			t := t.Option() | ||||||
|  | 			if m[t] == 1 { // Transformer was used exactly once before | ||||||
|  | 				tf := t.(*transformer).fnc.Type() | ||||||
|  | 				ss = append(ss, fmt.Sprintf("%v: %v => %v", t, tf.In(0), tf.Out(0))) | ||||||
|  | 			} | ||||||
|  | 			m[t]++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if len(ss) > 0 { | ||||||
|  | 		const warning = "recursive set of Transformers detected" | ||||||
|  | 		const help = "consider using cmpopts.AcyclicTransformer" | ||||||
|  | 		set := strings.Join(ss, "\n\t") | ||||||
|  | 		panic(fmt.Sprintf("%s:\n\t%s\n%s", warning, set, help)) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // dynChecker tracks the state needed to periodically perform checks that | ||||||
|  | // user provided functions are symmetric and deterministic. | ||||||
|  | // The zero value is safe for immediate use. | ||||||
|  | type dynChecker struct{ curr, next int } | ||||||
|  | 
 | ||||||
|  | // Next increments the state and reports whether a check should be performed. | ||||||
|  | // | ||||||
|  | // Checks occur every Nth function call, where N is a triangular number: | ||||||
|  | // | ||||||
|  | //	0 1 3 6 10 15 21 28 36 45 55 66 78 91 105 120 136 153 171 190 ... | ||||||
|  | // | ||||||
|  | // See https://en.wikipedia.org/wiki/Triangular_number | ||||||
|  | // | ||||||
|  | // This sequence ensures that the cost of checks drops significantly as | ||||||
|  | // the number of functions calls grows larger. | ||||||
|  | func (dc *dynChecker) Next() bool { | ||||||
|  | 	ok := dc.curr == dc.next | ||||||
|  | 	if ok { | ||||||
|  | 		dc.curr = 0 | ||||||
|  | 		dc.next++ | ||||||
|  | 	} | ||||||
|  | 	dc.curr++ | ||||||
|  | 	return ok | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // makeAddressable returns a value that is always addressable. | ||||||
|  | // It returns the input verbatim if it is already addressable, | ||||||
|  | // otherwise it creates a new value and returns an addressable copy. | ||||||
|  | func makeAddressable(v reflect.Value) reflect.Value { | ||||||
|  | 	if v.CanAddr() { | ||||||
|  | 		return v | ||||||
|  | 	} | ||||||
|  | 	vc := reflect.New(v.Type()).Elem() | ||||||
|  | 	vc.Set(v) | ||||||
|  | 	return vc | ||||||
|  | } | ||||||
							
								
								
									
										31
									
								
								vendor/github.com/google/go-cmp/cmp/export.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								vendor/github.com/google/go-cmp/cmp/export.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,31 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  | 	"unsafe" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // retrieveUnexportedField uses unsafe to forcibly retrieve any field from | ||||||
|  | // a struct such that the value has read-write permissions. | ||||||
|  | // | ||||||
|  | // The parent struct, v, must be addressable, while f must be a StructField | ||||||
|  | // describing the field to retrieve. If addr is false, | ||||||
|  | // then the returned value will be shallowed copied to be non-addressable. | ||||||
|  | func retrieveUnexportedField(v reflect.Value, f reflect.StructField, addr bool) reflect.Value { | ||||||
|  | 	ve := reflect.NewAt(f.Type, unsafe.Pointer(uintptr(unsafe.Pointer(v.UnsafeAddr()))+f.Offset)).Elem() | ||||||
|  | 	if !addr { | ||||||
|  | 		// A field is addressable if and only if the struct is addressable. | ||||||
|  | 		// If the original parent value was not addressable, shallow copy the | ||||||
|  | 		// value to make it non-addressable to avoid leaking an implementation | ||||||
|  | 		// detail of how forcibly exporting a field works. | ||||||
|  | 		if ve.Kind() == reflect.Interface && ve.IsNil() { | ||||||
|  | 			return reflect.Zero(f.Type) | ||||||
|  | 		} | ||||||
|  | 		return reflect.ValueOf(ve.Interface()).Convert(f.Type) | ||||||
|  | 	} | ||||||
|  | 	return ve | ||||||
|  | } | ||||||
							
								
								
									
										18
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/debug_disable.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/debug_disable.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,18 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | //go:build !cmp_debug | ||||||
|  | // +build !cmp_debug | ||||||
|  | 
 | ||||||
|  | package diff | ||||||
|  | 
 | ||||||
|  | var debug debugger | ||||||
|  | 
 | ||||||
|  | type debugger struct{} | ||||||
|  | 
 | ||||||
|  | func (debugger) Begin(_, _ int, f EqualFunc, _, _ *EditScript) EqualFunc { | ||||||
|  | 	return f | ||||||
|  | } | ||||||
|  | func (debugger) Update() {} | ||||||
|  | func (debugger) Finish() {} | ||||||
							
								
								
									
										123
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/debug_enable.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										123
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/debug_enable.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,123 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | //go:build cmp_debug | ||||||
|  | // +build cmp_debug | ||||||
|  | 
 | ||||||
|  | package diff | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"strings" | ||||||
|  | 	"sync" | ||||||
|  | 	"time" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // The algorithm can be seen running in real-time by enabling debugging: | ||||||
|  | //	go test -tags=cmp_debug -v | ||||||
|  | // | ||||||
|  | // Example output: | ||||||
|  | //	=== RUN   TestDifference/#34 | ||||||
|  | //	┌───────────────────────────────┐ | ||||||
|  | //	│ \ · · · · · · · · · · · · · · │ | ||||||
|  | //	│ · # · · · · · · · · · · · · · │ | ||||||
|  | //	│ · \ · · · · · · · · · · · · · │ | ||||||
|  | //	│ · · \ · · · · · · · · · · · · │ | ||||||
|  | //	│ · · · X # · · · · · · · · · · │ | ||||||
|  | //	│ · · · # \ · · · · · · · · · · │ | ||||||
|  | //	│ · · · · · # # · · · · · · · · │ | ||||||
|  | //	│ · · · · · # \ · · · · · · · · │ | ||||||
|  | //	│ · · · · · · · \ · · · · · · · │ | ||||||
|  | //	│ · · · · · · · · \ · · · · · · │ | ||||||
|  | //	│ · · · · · · · · · \ · · · · · │ | ||||||
|  | //	│ · · · · · · · · · · \ · · # · │ | ||||||
|  | //	│ · · · · · · · · · · · \ # # · │ | ||||||
|  | //	│ · · · · · · · · · · · # # # · │ | ||||||
|  | //	│ · · · · · · · · · · # # # # · │ | ||||||
|  | //	│ · · · · · · · · · # # # # # · │ | ||||||
|  | //	│ · · · · · · · · · · · · · · \ │ | ||||||
|  | //	└───────────────────────────────┘ | ||||||
|  | //	[.Y..M.XY......YXYXY.|] | ||||||
|  | // | ||||||
|  | // The grid represents the edit-graph where the horizontal axis represents | ||||||
|  | // list X and the vertical axis represents list Y. The start of the two lists | ||||||
|  | // is the top-left, while the ends are the bottom-right. The '·' represents | ||||||
|  | // an unexplored node in the graph. The '\' indicates that the two symbols | ||||||
|  | // from list X and Y are equal. The 'X' indicates that two symbols are similar | ||||||
|  | // (but not exactly equal) to each other. The '#' indicates that the two symbols | ||||||
|  | // are different (and not similar). The algorithm traverses this graph trying to | ||||||
|  | // make the paths starting in the top-left and the bottom-right connect. | ||||||
|  | // | ||||||
|  | // The series of '.', 'X', 'Y', and 'M' characters at the bottom represents | ||||||
|  | // the currently established path from the forward and reverse searches, | ||||||
|  | // separated by a '|' character. | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	updateDelay  = 100 * time.Millisecond | ||||||
|  | 	finishDelay  = 500 * time.Millisecond | ||||||
|  | 	ansiTerminal = true // ANSI escape codes used to move terminal cursor | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var debug debugger | ||||||
|  | 
 | ||||||
|  | type debugger struct { | ||||||
|  | 	sync.Mutex | ||||||
|  | 	p1, p2           EditScript | ||||||
|  | 	fwdPath, revPath *EditScript | ||||||
|  | 	grid             []byte | ||||||
|  | 	lines            int | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (dbg *debugger) Begin(nx, ny int, f EqualFunc, p1, p2 *EditScript) EqualFunc { | ||||||
|  | 	dbg.Lock() | ||||||
|  | 	dbg.fwdPath, dbg.revPath = p1, p2 | ||||||
|  | 	top := "┌─" + strings.Repeat("──", nx) + "┐\n" | ||||||
|  | 	row := "│ " + strings.Repeat("· ", nx) + "│\n" | ||||||
|  | 	btm := "└─" + strings.Repeat("──", nx) + "┘\n" | ||||||
|  | 	dbg.grid = []byte(top + strings.Repeat(row, ny) + btm) | ||||||
|  | 	dbg.lines = strings.Count(dbg.String(), "\n") | ||||||
|  | 	fmt.Print(dbg) | ||||||
|  | 
 | ||||||
|  | 	// Wrap the EqualFunc so that we can intercept each result. | ||||||
|  | 	return func(ix, iy int) (r Result) { | ||||||
|  | 		cell := dbg.grid[len(top)+iy*len(row):][len("│ ")+len("· ")*ix:][:len("·")] | ||||||
|  | 		for i := range cell { | ||||||
|  | 			cell[i] = 0 // Zero out the multiple bytes of UTF-8 middle-dot | ||||||
|  | 		} | ||||||
|  | 		switch r = f(ix, iy); { | ||||||
|  | 		case r.Equal(): | ||||||
|  | 			cell[0] = '\\' | ||||||
|  | 		case r.Similar(): | ||||||
|  | 			cell[0] = 'X' | ||||||
|  | 		default: | ||||||
|  | 			cell[0] = '#' | ||||||
|  | 		} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (dbg *debugger) Update() { | ||||||
|  | 	dbg.print(updateDelay) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (dbg *debugger) Finish() { | ||||||
|  | 	dbg.print(finishDelay) | ||||||
|  | 	dbg.Unlock() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (dbg *debugger) String() string { | ||||||
|  | 	dbg.p1, dbg.p2 = *dbg.fwdPath, dbg.p2[:0] | ||||||
|  | 	for i := len(*dbg.revPath) - 1; i >= 0; i-- { | ||||||
|  | 		dbg.p2 = append(dbg.p2, (*dbg.revPath)[i]) | ||||||
|  | 	} | ||||||
|  | 	return fmt.Sprintf("%s[%v|%v]\n\n", dbg.grid, dbg.p1, dbg.p2) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (dbg *debugger) print(d time.Duration) { | ||||||
|  | 	if ansiTerminal { | ||||||
|  | 		fmt.Printf("\x1b[%dA", dbg.lines) // Reset terminal cursor | ||||||
|  | 	} | ||||||
|  | 	fmt.Print(dbg) | ||||||
|  | 	time.Sleep(d) | ||||||
|  | } | ||||||
							
								
								
									
										402
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/diff.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										402
									
								
								vendor/github.com/google/go-cmp/cmp/internal/diff/diff.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,402 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | // Package diff implements an algorithm for producing edit-scripts. | ||||||
|  | // The edit-script is a sequence of operations needed to transform one list | ||||||
|  | // of symbols into another (or vice-versa). The edits allowed are insertions, | ||||||
|  | // deletions, and modifications. The summation of all edits is called the | ||||||
|  | // Levenshtein distance as this problem is well-known in computer science. | ||||||
|  | // | ||||||
|  | // This package prioritizes performance over accuracy. That is, the run time | ||||||
|  | // is more important than obtaining a minimal Levenshtein distance. | ||||||
|  | package diff | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"math/rand" | ||||||
|  | 	"time" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/flags" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // EditType represents a single operation within an edit-script. | ||||||
|  | type EditType uint8 | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	// Identity indicates that a symbol pair is identical in both list X and Y. | ||||||
|  | 	Identity EditType = iota | ||||||
|  | 	// UniqueX indicates that a symbol only exists in X and not Y. | ||||||
|  | 	UniqueX | ||||||
|  | 	// UniqueY indicates that a symbol only exists in Y and not X. | ||||||
|  | 	UniqueY | ||||||
|  | 	// Modified indicates that a symbol pair is a modification of each other. | ||||||
|  | 	Modified | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // EditScript represents the series of differences between two lists. | ||||||
|  | type EditScript []EditType | ||||||
|  | 
 | ||||||
|  | // String returns a human-readable string representing the edit-script where | ||||||
|  | // Identity, UniqueX, UniqueY, and Modified are represented by the | ||||||
|  | // '.', 'X', 'Y', and 'M' characters, respectively. | ||||||
|  | func (es EditScript) String() string { | ||||||
|  | 	b := make([]byte, len(es)) | ||||||
|  | 	for i, e := range es { | ||||||
|  | 		switch e { | ||||||
|  | 		case Identity: | ||||||
|  | 			b[i] = '.' | ||||||
|  | 		case UniqueX: | ||||||
|  | 			b[i] = 'X' | ||||||
|  | 		case UniqueY: | ||||||
|  | 			b[i] = 'Y' | ||||||
|  | 		case Modified: | ||||||
|  | 			b[i] = 'M' | ||||||
|  | 		default: | ||||||
|  | 			panic("invalid edit-type") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return string(b) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // stats returns a histogram of the number of each type of edit operation. | ||||||
|  | func (es EditScript) stats() (s struct{ NI, NX, NY, NM int }) { | ||||||
|  | 	for _, e := range es { | ||||||
|  | 		switch e { | ||||||
|  | 		case Identity: | ||||||
|  | 			s.NI++ | ||||||
|  | 		case UniqueX: | ||||||
|  | 			s.NX++ | ||||||
|  | 		case UniqueY: | ||||||
|  | 			s.NY++ | ||||||
|  | 		case Modified: | ||||||
|  | 			s.NM++ | ||||||
|  | 		default: | ||||||
|  | 			panic("invalid edit-type") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Dist is the Levenshtein distance and is guaranteed to be 0 if and only if | ||||||
|  | // lists X and Y are equal. | ||||||
|  | func (es EditScript) Dist() int { return len(es) - es.stats().NI } | ||||||
|  | 
 | ||||||
|  | // LenX is the length of the X list. | ||||||
|  | func (es EditScript) LenX() int { return len(es) - es.stats().NY } | ||||||
|  | 
 | ||||||
|  | // LenY is the length of the Y list. | ||||||
|  | func (es EditScript) LenY() int { return len(es) - es.stats().NX } | ||||||
|  | 
 | ||||||
|  | // EqualFunc reports whether the symbols at indexes ix and iy are equal. | ||||||
|  | // When called by Difference, the index is guaranteed to be within nx and ny. | ||||||
|  | type EqualFunc func(ix int, iy int) Result | ||||||
|  | 
 | ||||||
|  | // Result is the result of comparison. | ||||||
|  | // NumSame is the number of sub-elements that are equal. | ||||||
|  | // NumDiff is the number of sub-elements that are not equal. | ||||||
|  | type Result struct{ NumSame, NumDiff int } | ||||||
|  | 
 | ||||||
|  | // BoolResult returns a Result that is either Equal or not Equal. | ||||||
|  | func BoolResult(b bool) Result { | ||||||
|  | 	if b { | ||||||
|  | 		return Result{NumSame: 1} // Equal, Similar | ||||||
|  | 	} else { | ||||||
|  | 		return Result{NumDiff: 2} // Not Equal, not Similar | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Equal indicates whether the symbols are equal. Two symbols are equal | ||||||
|  | // if and only if NumDiff == 0. If Equal, then they are also Similar. | ||||||
|  | func (r Result) Equal() bool { return r.NumDiff == 0 } | ||||||
|  | 
 | ||||||
|  | // Similar indicates whether two symbols are similar and may be represented | ||||||
|  | // by using the Modified type. As a special case, we consider binary comparisons | ||||||
|  | // (i.e., those that return Result{1, 0} or Result{0, 1}) to be similar. | ||||||
|  | // | ||||||
|  | // The exact ratio of NumSame to NumDiff to determine similarity may change. | ||||||
|  | func (r Result) Similar() bool { | ||||||
|  | 	// Use NumSame+1 to offset NumSame so that binary comparisons are similar. | ||||||
|  | 	return r.NumSame+1 >= r.NumDiff | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | var randBool = rand.New(rand.NewSource(time.Now().Unix())).Intn(2) == 0 | ||||||
|  | 
 | ||||||
|  | // Difference reports whether two lists of lengths nx and ny are equal | ||||||
|  | // given the definition of equality provided as f. | ||||||
|  | // | ||||||
|  | // This function returns an edit-script, which is a sequence of operations | ||||||
|  | // needed to convert one list into the other. The following invariants for | ||||||
|  | // the edit-script are maintained: | ||||||
|  | //   - eq == (es.Dist()==0) | ||||||
|  | //   - nx == es.LenX() | ||||||
|  | //   - ny == es.LenY() | ||||||
|  | // | ||||||
|  | // This algorithm is not guaranteed to be an optimal solution (i.e., one that | ||||||
|  | // produces an edit-script with a minimal Levenshtein distance). This algorithm | ||||||
|  | // favors performance over optimality. The exact output is not guaranteed to | ||||||
|  | // be stable and may change over time. | ||||||
|  | func Difference(nx, ny int, f EqualFunc) (es EditScript) { | ||||||
|  | 	// This algorithm is based on traversing what is known as an "edit-graph". | ||||||
|  | 	// See Figure 1 from "An O(ND) Difference Algorithm and Its Variations" | ||||||
|  | 	// by Eugene W. Myers. Since D can be as large as N itself, this is | ||||||
|  | 	// effectively O(N^2). Unlike the algorithm from that paper, we are not | ||||||
|  | 	// interested in the optimal path, but at least some "decent" path. | ||||||
|  | 	// | ||||||
|  | 	// For example, let X and Y be lists of symbols: | ||||||
|  | 	//	X = [A B C A B B A] | ||||||
|  | 	//	Y = [C B A B A C] | ||||||
|  | 	// | ||||||
|  | 	// The edit-graph can be drawn as the following: | ||||||
|  | 	//	   A B C A B B A | ||||||
|  | 	//	  ┌─────────────┐ | ||||||
|  | 	//	C │_|_|\|_|_|_|_│ 0 | ||||||
|  | 	//	B │_|\|_|_|\|\|_│ 1 | ||||||
|  | 	//	A │\|_|_|\|_|_|\│ 2 | ||||||
|  | 	//	B │_|\|_|_|\|\|_│ 3 | ||||||
|  | 	//	A │\|_|_|\|_|_|\│ 4 | ||||||
|  | 	//	C │ | |\| | | | │ 5 | ||||||
|  | 	//	  └─────────────┘ 6 | ||||||
|  | 	//	   0 1 2 3 4 5 6 7 | ||||||
|  | 	// | ||||||
|  | 	// List X is written along the horizontal axis, while list Y is written | ||||||
|  | 	// along the vertical axis. At any point on this grid, if the symbol in | ||||||
|  | 	// list X matches the corresponding symbol in list Y, then a '\' is drawn. | ||||||
|  | 	// The goal of any minimal edit-script algorithm is to find a path from the | ||||||
|  | 	// top-left corner to the bottom-right corner, while traveling through the | ||||||
|  | 	// fewest horizontal or vertical edges. | ||||||
|  | 	// A horizontal edge is equivalent to inserting a symbol from list X. | ||||||
|  | 	// A vertical edge is equivalent to inserting a symbol from list Y. | ||||||
|  | 	// A diagonal edge is equivalent to a matching symbol between both X and Y. | ||||||
|  | 
 | ||||||
|  | 	// Invariants: | ||||||
|  | 	//   - 0 ≤ fwdPath.X ≤ (fwdFrontier.X, revFrontier.X) ≤ revPath.X ≤ nx | ||||||
|  | 	//   - 0 ≤ fwdPath.Y ≤ (fwdFrontier.Y, revFrontier.Y) ≤ revPath.Y ≤ ny | ||||||
|  | 	// | ||||||
|  | 	// In general: | ||||||
|  | 	//   - fwdFrontier.X < revFrontier.X | ||||||
|  | 	//   - fwdFrontier.Y < revFrontier.Y | ||||||
|  | 	// | ||||||
|  | 	// Unless, it is time for the algorithm to terminate. | ||||||
|  | 	fwdPath := path{+1, point{0, 0}, make(EditScript, 0, (nx+ny)/2)} | ||||||
|  | 	revPath := path{-1, point{nx, ny}, make(EditScript, 0)} | ||||||
|  | 	fwdFrontier := fwdPath.point // Forward search frontier | ||||||
|  | 	revFrontier := revPath.point // Reverse search frontier | ||||||
|  | 
 | ||||||
|  | 	// Search budget bounds the cost of searching for better paths. | ||||||
|  | 	// The longest sequence of non-matching symbols that can be tolerated is | ||||||
|  | 	// approximately the square-root of the search budget. | ||||||
|  | 	searchBudget := 4 * (nx + ny) // O(n) | ||||||
|  | 
 | ||||||
|  | 	// Running the tests with the "cmp_debug" build tag prints a visualization | ||||||
|  | 	// of the algorithm running in real-time. This is educational for | ||||||
|  | 	// understanding how the algorithm works. See debug_enable.go. | ||||||
|  | 	f = debug.Begin(nx, ny, f, &fwdPath.es, &revPath.es) | ||||||
|  | 
 | ||||||
|  | 	// The algorithm below is a greedy, meet-in-the-middle algorithm for | ||||||
|  | 	// computing sub-optimal edit-scripts between two lists. | ||||||
|  | 	// | ||||||
|  | 	// The algorithm is approximately as follows: | ||||||
|  | 	//   - Searching for differences switches back-and-forth between | ||||||
|  | 	//     a search that starts at the beginning (the top-left corner), and | ||||||
|  | 	//     a search that starts at the end (the bottom-right corner). | ||||||
|  | 	//     The goal of the search is connect with the search | ||||||
|  | 	//     from the opposite corner. | ||||||
|  | 	//   - As we search, we build a path in a greedy manner, | ||||||
|  | 	//     where the first match seen is added to the path (this is sub-optimal, | ||||||
|  | 	//     but provides a decent result in practice). When matches are found, | ||||||
|  | 	//     we try the next pair of symbols in the lists and follow all matches | ||||||
|  | 	//     as far as possible. | ||||||
|  | 	//   - When searching for matches, we search along a diagonal going through | ||||||
|  | 	//     through the "frontier" point. If no matches are found, | ||||||
|  | 	//     we advance the frontier towards the opposite corner. | ||||||
|  | 	//   - This algorithm terminates when either the X coordinates or the | ||||||
|  | 	//     Y coordinates of the forward and reverse frontier points ever intersect. | ||||||
|  | 
 | ||||||
|  | 	// This algorithm is correct even if searching only in the forward direction | ||||||
|  | 	// or in the reverse direction. We do both because it is commonly observed | ||||||
|  | 	// that two lists commonly differ because elements were added to the front | ||||||
|  | 	// or end of the other list. | ||||||
|  | 	// | ||||||
|  | 	// Non-deterministically start with either the forward or reverse direction | ||||||
|  | 	// to introduce some deliberate instability so that we have the flexibility | ||||||
|  | 	// to change this algorithm in the future. | ||||||
|  | 	if flags.Deterministic || randBool { | ||||||
|  | 		goto forwardSearch | ||||||
|  | 	} else { | ||||||
|  | 		goto reverseSearch | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | forwardSearch: | ||||||
|  | 	{ | ||||||
|  | 		// Forward search from the beginning. | ||||||
|  | 		if fwdFrontier.X >= revFrontier.X || fwdFrontier.Y >= revFrontier.Y || searchBudget == 0 { | ||||||
|  | 			goto finishSearch | ||||||
|  | 		} | ||||||
|  | 		for stop1, stop2, i := false, false, 0; !(stop1 && stop2) && searchBudget > 0; i++ { | ||||||
|  | 			// Search in a diagonal pattern for a match. | ||||||
|  | 			z := zigzag(i) | ||||||
|  | 			p := point{fwdFrontier.X + z, fwdFrontier.Y - z} | ||||||
|  | 			switch { | ||||||
|  | 			case p.X >= revPath.X || p.Y < fwdPath.Y: | ||||||
|  | 				stop1 = true // Hit top-right corner | ||||||
|  | 			case p.Y >= revPath.Y || p.X < fwdPath.X: | ||||||
|  | 				stop2 = true // Hit bottom-left corner | ||||||
|  | 			case f(p.X, p.Y).Equal(): | ||||||
|  | 				// Match found, so connect the path to this point. | ||||||
|  | 				fwdPath.connect(p, f) | ||||||
|  | 				fwdPath.append(Identity) | ||||||
|  | 				// Follow sequence of matches as far as possible. | ||||||
|  | 				for fwdPath.X < revPath.X && fwdPath.Y < revPath.Y { | ||||||
|  | 					if !f(fwdPath.X, fwdPath.Y).Equal() { | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 					fwdPath.append(Identity) | ||||||
|  | 				} | ||||||
|  | 				fwdFrontier = fwdPath.point | ||||||
|  | 				stop1, stop2 = true, true | ||||||
|  | 			default: | ||||||
|  | 				searchBudget-- // Match not found | ||||||
|  | 			} | ||||||
|  | 			debug.Update() | ||||||
|  | 		} | ||||||
|  | 		// Advance the frontier towards reverse point. | ||||||
|  | 		if revPath.X-fwdFrontier.X >= revPath.Y-fwdFrontier.Y { | ||||||
|  | 			fwdFrontier.X++ | ||||||
|  | 		} else { | ||||||
|  | 			fwdFrontier.Y++ | ||||||
|  | 		} | ||||||
|  | 		goto reverseSearch | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | reverseSearch: | ||||||
|  | 	{ | ||||||
|  | 		// Reverse search from the end. | ||||||
|  | 		if fwdFrontier.X >= revFrontier.X || fwdFrontier.Y >= revFrontier.Y || searchBudget == 0 { | ||||||
|  | 			goto finishSearch | ||||||
|  | 		} | ||||||
|  | 		for stop1, stop2, i := false, false, 0; !(stop1 && stop2) && searchBudget > 0; i++ { | ||||||
|  | 			// Search in a diagonal pattern for a match. | ||||||
|  | 			z := zigzag(i) | ||||||
|  | 			p := point{revFrontier.X - z, revFrontier.Y + z} | ||||||
|  | 			switch { | ||||||
|  | 			case fwdPath.X >= p.X || revPath.Y < p.Y: | ||||||
|  | 				stop1 = true // Hit bottom-left corner | ||||||
|  | 			case fwdPath.Y >= p.Y || revPath.X < p.X: | ||||||
|  | 				stop2 = true // Hit top-right corner | ||||||
|  | 			case f(p.X-1, p.Y-1).Equal(): | ||||||
|  | 				// Match found, so connect the path to this point. | ||||||
|  | 				revPath.connect(p, f) | ||||||
|  | 				revPath.append(Identity) | ||||||
|  | 				// Follow sequence of matches as far as possible. | ||||||
|  | 				for fwdPath.X < revPath.X && fwdPath.Y < revPath.Y { | ||||||
|  | 					if !f(revPath.X-1, revPath.Y-1).Equal() { | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 					revPath.append(Identity) | ||||||
|  | 				} | ||||||
|  | 				revFrontier = revPath.point | ||||||
|  | 				stop1, stop2 = true, true | ||||||
|  | 			default: | ||||||
|  | 				searchBudget-- // Match not found | ||||||
|  | 			} | ||||||
|  | 			debug.Update() | ||||||
|  | 		} | ||||||
|  | 		// Advance the frontier towards forward point. | ||||||
|  | 		if revFrontier.X-fwdPath.X >= revFrontier.Y-fwdPath.Y { | ||||||
|  | 			revFrontier.X-- | ||||||
|  | 		} else { | ||||||
|  | 			revFrontier.Y-- | ||||||
|  | 		} | ||||||
|  | 		goto forwardSearch | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | finishSearch: | ||||||
|  | 	// Join the forward and reverse paths and then append the reverse path. | ||||||
|  | 	fwdPath.connect(revPath.point, f) | ||||||
|  | 	for i := len(revPath.es) - 1; i >= 0; i-- { | ||||||
|  | 		t := revPath.es[i] | ||||||
|  | 		revPath.es = revPath.es[:i] | ||||||
|  | 		fwdPath.append(t) | ||||||
|  | 	} | ||||||
|  | 	debug.Finish() | ||||||
|  | 	return fwdPath.es | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type path struct { | ||||||
|  | 	dir   int // +1 if forward, -1 if reverse | ||||||
|  | 	point     // Leading point of the EditScript path | ||||||
|  | 	es    EditScript | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // connect appends any necessary Identity, Modified, UniqueX, or UniqueY types | ||||||
|  | // to the edit-script to connect p.point to dst. | ||||||
|  | func (p *path) connect(dst point, f EqualFunc) { | ||||||
|  | 	if p.dir > 0 { | ||||||
|  | 		// Connect in forward direction. | ||||||
|  | 		for dst.X > p.X && dst.Y > p.Y { | ||||||
|  | 			switch r := f(p.X, p.Y); { | ||||||
|  | 			case r.Equal(): | ||||||
|  | 				p.append(Identity) | ||||||
|  | 			case r.Similar(): | ||||||
|  | 				p.append(Modified) | ||||||
|  | 			case dst.X-p.X >= dst.Y-p.Y: | ||||||
|  | 				p.append(UniqueX) | ||||||
|  | 			default: | ||||||
|  | 				p.append(UniqueY) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for dst.X > p.X { | ||||||
|  | 			p.append(UniqueX) | ||||||
|  | 		} | ||||||
|  | 		for dst.Y > p.Y { | ||||||
|  | 			p.append(UniqueY) | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		// Connect in reverse direction. | ||||||
|  | 		for p.X > dst.X && p.Y > dst.Y { | ||||||
|  | 			switch r := f(p.X-1, p.Y-1); { | ||||||
|  | 			case r.Equal(): | ||||||
|  | 				p.append(Identity) | ||||||
|  | 			case r.Similar(): | ||||||
|  | 				p.append(Modified) | ||||||
|  | 			case p.Y-dst.Y >= p.X-dst.X: | ||||||
|  | 				p.append(UniqueY) | ||||||
|  | 			default: | ||||||
|  | 				p.append(UniqueX) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for p.X > dst.X { | ||||||
|  | 			p.append(UniqueX) | ||||||
|  | 		} | ||||||
|  | 		for p.Y > dst.Y { | ||||||
|  | 			p.append(UniqueY) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (p *path) append(t EditType) { | ||||||
|  | 	p.es = append(p.es, t) | ||||||
|  | 	switch t { | ||||||
|  | 	case Identity, Modified: | ||||||
|  | 		p.add(p.dir, p.dir) | ||||||
|  | 	case UniqueX: | ||||||
|  | 		p.add(p.dir, 0) | ||||||
|  | 	case UniqueY: | ||||||
|  | 		p.add(0, p.dir) | ||||||
|  | 	} | ||||||
|  | 	debug.Update() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type point struct{ X, Y int } | ||||||
|  | 
 | ||||||
|  | func (p *point) add(dx, dy int) { p.X += dx; p.Y += dy } | ||||||
|  | 
 | ||||||
|  | // zigzag maps a consecutive sequence of integers to a zig-zag sequence. | ||||||
|  | // | ||||||
|  | //	[0 1 2 3 4 5 ...] => [0 -1 +1 -2 +2 ...] | ||||||
|  | func zigzag(x int) int { | ||||||
|  | 	if x&1 != 0 { | ||||||
|  | 		x = ^x | ||||||
|  | 	} | ||||||
|  | 	return x >> 1 | ||||||
|  | } | ||||||
							
								
								
									
										9
									
								
								vendor/github.com/google/go-cmp/cmp/internal/flags/flags.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								vendor/github.com/google/go-cmp/cmp/internal/flags/flags.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,9 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package flags | ||||||
|  | 
 | ||||||
|  | // Deterministic controls whether the output of Diff should be deterministic. | ||||||
|  | // This is only used for testing. | ||||||
|  | var Deterministic bool | ||||||
							
								
								
									
										99
									
								
								vendor/github.com/google/go-cmp/cmp/internal/function/func.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										99
									
								
								vendor/github.com/google/go-cmp/cmp/internal/function/func.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,99 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | // Package function provides functionality for identifying function types. | ||||||
|  | package function | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  | 	"regexp" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strings" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type funcType int | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	_ funcType = iota | ||||||
|  | 
 | ||||||
|  | 	tbFunc  // func(T) bool | ||||||
|  | 	ttbFunc // func(T, T) bool | ||||||
|  | 	trbFunc // func(T, R) bool | ||||||
|  | 	tibFunc // func(T, I) bool | ||||||
|  | 	trFunc  // func(T) R | ||||||
|  | 
 | ||||||
|  | 	Equal             = ttbFunc // func(T, T) bool | ||||||
|  | 	EqualAssignable   = tibFunc // func(T, I) bool; encapsulates func(T, T) bool | ||||||
|  | 	Transformer       = trFunc  // func(T) R | ||||||
|  | 	ValueFilter       = ttbFunc // func(T, T) bool | ||||||
|  | 	Less              = ttbFunc // func(T, T) bool | ||||||
|  | 	ValuePredicate    = tbFunc  // func(T) bool | ||||||
|  | 	KeyValuePredicate = trbFunc // func(T, R) bool | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var boolType = reflect.TypeOf(true) | ||||||
|  | 
 | ||||||
|  | // IsType reports whether the reflect.Type is of the specified function type. | ||||||
|  | func IsType(t reflect.Type, ft funcType) bool { | ||||||
|  | 	if t == nil || t.Kind() != reflect.Func || t.IsVariadic() { | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 	ni, no := t.NumIn(), t.NumOut() | ||||||
|  | 	switch ft { | ||||||
|  | 	case tbFunc: // func(T) bool | ||||||
|  | 		if ni == 1 && no == 1 && t.Out(0) == boolType { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	case ttbFunc: // func(T, T) bool | ||||||
|  | 		if ni == 2 && no == 1 && t.In(0) == t.In(1) && t.Out(0) == boolType { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	case trbFunc: // func(T, R) bool | ||||||
|  | 		if ni == 2 && no == 1 && t.Out(0) == boolType { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	case tibFunc: // func(T, I) bool | ||||||
|  | 		if ni == 2 && no == 1 && t.In(0).AssignableTo(t.In(1)) && t.Out(0) == boolType { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	case trFunc: // func(T) R | ||||||
|  | 		if ni == 1 && no == 1 { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | var lastIdentRx = regexp.MustCompile(`[_\p{L}][_\p{L}\p{N}]*$`) | ||||||
|  | 
 | ||||||
|  | // NameOf returns the name of the function value. | ||||||
|  | func NameOf(v reflect.Value) string { | ||||||
|  | 	fnc := runtime.FuncForPC(v.Pointer()) | ||||||
|  | 	if fnc == nil { | ||||||
|  | 		return "<unknown>" | ||||||
|  | 	} | ||||||
|  | 	fullName := fnc.Name() // e.g., "long/path/name/mypkg.(*MyType).(long/path/name/mypkg.myMethod)-fm" | ||||||
|  | 
 | ||||||
|  | 	// Method closures have a "-fm" suffix. | ||||||
|  | 	fullName = strings.TrimSuffix(fullName, "-fm") | ||||||
|  | 
 | ||||||
|  | 	var name string | ||||||
|  | 	for len(fullName) > 0 { | ||||||
|  | 		inParen := strings.HasSuffix(fullName, ")") | ||||||
|  | 		fullName = strings.TrimSuffix(fullName, ")") | ||||||
|  | 
 | ||||||
|  | 		s := lastIdentRx.FindString(fullName) | ||||||
|  | 		if s == "" { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 		name = s + "." + name | ||||||
|  | 		fullName = strings.TrimSuffix(fullName, s) | ||||||
|  | 
 | ||||||
|  | 		if i := strings.LastIndexByte(fullName, '('); inParen && i >= 0 { | ||||||
|  | 			fullName = fullName[:i] | ||||||
|  | 		} | ||||||
|  | 		fullName = strings.TrimSuffix(fullName, ".") | ||||||
|  | 	} | ||||||
|  | 	return strings.TrimSuffix(name, ".") | ||||||
|  | } | ||||||
							
								
								
									
										164
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/name.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										164
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/name.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,164 @@ | ||||||
|  | // Copyright 2020, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package value | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  | 	"strconv" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var anyType = reflect.TypeOf((*interface{})(nil)).Elem() | ||||||
|  | 
 | ||||||
|  | // TypeString is nearly identical to reflect.Type.String, | ||||||
|  | // but has an additional option to specify that full type names be used. | ||||||
|  | func TypeString(t reflect.Type, qualified bool) string { | ||||||
|  | 	return string(appendTypeName(nil, t, qualified, false)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func appendTypeName(b []byte, t reflect.Type, qualified, elideFunc bool) []byte { | ||||||
|  | 	// BUG: Go reflection provides no way to disambiguate two named types | ||||||
|  | 	// of the same name and within the same package, | ||||||
|  | 	// but declared within the namespace of different functions. | ||||||
|  | 
 | ||||||
|  | 	// Use the "any" alias instead of "interface{}" for better readability. | ||||||
|  | 	if t == anyType { | ||||||
|  | 		return append(b, "any"...) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Named type. | ||||||
|  | 	if t.Name() != "" { | ||||||
|  | 		if qualified && t.PkgPath() != "" { | ||||||
|  | 			b = append(b, '"') | ||||||
|  | 			b = append(b, t.PkgPath()...) | ||||||
|  | 			b = append(b, '"') | ||||||
|  | 			b = append(b, '.') | ||||||
|  | 			b = append(b, t.Name()...) | ||||||
|  | 		} else { | ||||||
|  | 			b = append(b, t.String()...) | ||||||
|  | 		} | ||||||
|  | 		return b | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Unnamed type. | ||||||
|  | 	switch k := t.Kind(); k { | ||||||
|  | 	case reflect.Bool, reflect.String, reflect.UnsafePointer, | ||||||
|  | 		reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, | ||||||
|  | 		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr, | ||||||
|  | 		reflect.Float32, reflect.Float64, reflect.Complex64, reflect.Complex128: | ||||||
|  | 		b = append(b, k.String()...) | ||||||
|  | 	case reflect.Chan: | ||||||
|  | 		if t.ChanDir() == reflect.RecvDir { | ||||||
|  | 			b = append(b, "<-"...) | ||||||
|  | 		} | ||||||
|  | 		b = append(b, "chan"...) | ||||||
|  | 		if t.ChanDir() == reflect.SendDir { | ||||||
|  | 			b = append(b, "<-"...) | ||||||
|  | 		} | ||||||
|  | 		b = append(b, ' ') | ||||||
|  | 		b = appendTypeName(b, t.Elem(), qualified, false) | ||||||
|  | 	case reflect.Func: | ||||||
|  | 		if !elideFunc { | ||||||
|  | 			b = append(b, "func"...) | ||||||
|  | 		} | ||||||
|  | 		b = append(b, '(') | ||||||
|  | 		for i := 0; i < t.NumIn(); i++ { | ||||||
|  | 			if i > 0 { | ||||||
|  | 				b = append(b, ", "...) | ||||||
|  | 			} | ||||||
|  | 			if i == t.NumIn()-1 && t.IsVariadic() { | ||||||
|  | 				b = append(b, "..."...) | ||||||
|  | 				b = appendTypeName(b, t.In(i).Elem(), qualified, false) | ||||||
|  | 			} else { | ||||||
|  | 				b = appendTypeName(b, t.In(i), qualified, false) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		b = append(b, ')') | ||||||
|  | 		switch t.NumOut() { | ||||||
|  | 		case 0: | ||||||
|  | 			// Do nothing | ||||||
|  | 		case 1: | ||||||
|  | 			b = append(b, ' ') | ||||||
|  | 			b = appendTypeName(b, t.Out(0), qualified, false) | ||||||
|  | 		default: | ||||||
|  | 			b = append(b, " ("...) | ||||||
|  | 			for i := 0; i < t.NumOut(); i++ { | ||||||
|  | 				if i > 0 { | ||||||
|  | 					b = append(b, ", "...) | ||||||
|  | 				} | ||||||
|  | 				b = appendTypeName(b, t.Out(i), qualified, false) | ||||||
|  | 			} | ||||||
|  | 			b = append(b, ')') | ||||||
|  | 		} | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		b = append(b, "struct{ "...) | ||||||
|  | 		for i := 0; i < t.NumField(); i++ { | ||||||
|  | 			if i > 0 { | ||||||
|  | 				b = append(b, "; "...) | ||||||
|  | 			} | ||||||
|  | 			sf := t.Field(i) | ||||||
|  | 			if !sf.Anonymous { | ||||||
|  | 				if qualified && sf.PkgPath != "" { | ||||||
|  | 					b = append(b, '"') | ||||||
|  | 					b = append(b, sf.PkgPath...) | ||||||
|  | 					b = append(b, '"') | ||||||
|  | 					b = append(b, '.') | ||||||
|  | 				} | ||||||
|  | 				b = append(b, sf.Name...) | ||||||
|  | 				b = append(b, ' ') | ||||||
|  | 			} | ||||||
|  | 			b = appendTypeName(b, sf.Type, qualified, false) | ||||||
|  | 			if sf.Tag != "" { | ||||||
|  | 				b = append(b, ' ') | ||||||
|  | 				b = strconv.AppendQuote(b, string(sf.Tag)) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if b[len(b)-1] == ' ' { | ||||||
|  | 			b = b[:len(b)-1] | ||||||
|  | 		} else { | ||||||
|  | 			b = append(b, ' ') | ||||||
|  | 		} | ||||||
|  | 		b = append(b, '}') | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		b = append(b, '[') | ||||||
|  | 		if k == reflect.Array { | ||||||
|  | 			b = strconv.AppendUint(b, uint64(t.Len()), 10) | ||||||
|  | 		} | ||||||
|  | 		b = append(b, ']') | ||||||
|  | 		b = appendTypeName(b, t.Elem(), qualified, false) | ||||||
|  | 	case reflect.Map: | ||||||
|  | 		b = append(b, "map["...) | ||||||
|  | 		b = appendTypeName(b, t.Key(), qualified, false) | ||||||
|  | 		b = append(b, ']') | ||||||
|  | 		b = appendTypeName(b, t.Elem(), qualified, false) | ||||||
|  | 	case reflect.Ptr: | ||||||
|  | 		b = append(b, '*') | ||||||
|  | 		b = appendTypeName(b, t.Elem(), qualified, false) | ||||||
|  | 	case reflect.Interface: | ||||||
|  | 		b = append(b, "interface{ "...) | ||||||
|  | 		for i := 0; i < t.NumMethod(); i++ { | ||||||
|  | 			if i > 0 { | ||||||
|  | 				b = append(b, "; "...) | ||||||
|  | 			} | ||||||
|  | 			m := t.Method(i) | ||||||
|  | 			if qualified && m.PkgPath != "" { | ||||||
|  | 				b = append(b, '"') | ||||||
|  | 				b = append(b, m.PkgPath...) | ||||||
|  | 				b = append(b, '"') | ||||||
|  | 				b = append(b, '.') | ||||||
|  | 			} | ||||||
|  | 			b = append(b, m.Name...) | ||||||
|  | 			b = appendTypeName(b, m.Type, qualified, true) | ||||||
|  | 		} | ||||||
|  | 		if b[len(b)-1] == ' ' { | ||||||
|  | 			b = b[:len(b)-1] | ||||||
|  | 		} else { | ||||||
|  | 			b = append(b, ' ') | ||||||
|  | 		} | ||||||
|  | 		b = append(b, '}') | ||||||
|  | 	default: | ||||||
|  | 		panic("invalid kind: " + k.String()) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
							
								
								
									
										34
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/pointer.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										34
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/pointer.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,34 @@ | ||||||
|  | // Copyright 2018, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package value | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"reflect" | ||||||
|  | 	"unsafe" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // Pointer is an opaque typed pointer and is guaranteed to be comparable. | ||||||
|  | type Pointer struct { | ||||||
|  | 	p unsafe.Pointer | ||||||
|  | 	t reflect.Type | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // PointerOf returns a Pointer from v, which must be a | ||||||
|  | // reflect.Ptr, reflect.Slice, or reflect.Map. | ||||||
|  | func PointerOf(v reflect.Value) Pointer { | ||||||
|  | 	// The proper representation of a pointer is unsafe.Pointer, | ||||||
|  | 	// which is necessary if the GC ever uses a moving collector. | ||||||
|  | 	return Pointer{unsafe.Pointer(v.Pointer()), v.Type()} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // IsNil reports whether the pointer is nil. | ||||||
|  | func (p Pointer) IsNil() bool { | ||||||
|  | 	return p.p == nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Uintptr returns the pointer as a uintptr. | ||||||
|  | func (p Pointer) Uintptr() uintptr { | ||||||
|  | 	return uintptr(p.p) | ||||||
|  | } | ||||||
							
								
								
									
										106
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/sort.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										106
									
								
								vendor/github.com/google/go-cmp/cmp/internal/value/sort.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,106 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package value | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"math" | ||||||
|  | 	"reflect" | ||||||
|  | 	"sort" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // SortKeys sorts a list of map keys, deduplicating keys if necessary. | ||||||
|  | // The type of each value must be comparable. | ||||||
|  | func SortKeys(vs []reflect.Value) []reflect.Value { | ||||||
|  | 	if len(vs) == 0 { | ||||||
|  | 		return vs | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Sort the map keys. | ||||||
|  | 	sort.SliceStable(vs, func(i, j int) bool { return isLess(vs[i], vs[j]) }) | ||||||
|  | 
 | ||||||
|  | 	// Deduplicate keys (fails for NaNs). | ||||||
|  | 	vs2 := vs[:1] | ||||||
|  | 	for _, v := range vs[1:] { | ||||||
|  | 		if isLess(vs2[len(vs2)-1], v) { | ||||||
|  | 			vs2 = append(vs2, v) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return vs2 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // isLess is a generic function for sorting arbitrary map keys. | ||||||
|  | // The inputs must be of the same type and must be comparable. | ||||||
|  | func isLess(x, y reflect.Value) bool { | ||||||
|  | 	switch x.Type().Kind() { | ||||||
|  | 	case reflect.Bool: | ||||||
|  | 		return !x.Bool() && y.Bool() | ||||||
|  | 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: | ||||||
|  | 		return x.Int() < y.Int() | ||||||
|  | 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: | ||||||
|  | 		return x.Uint() < y.Uint() | ||||||
|  | 	case reflect.Float32, reflect.Float64: | ||||||
|  | 		// NOTE: This does not sort -0 as less than +0 | ||||||
|  | 		// since Go maps treat -0 and +0 as equal keys. | ||||||
|  | 		fx, fy := x.Float(), y.Float() | ||||||
|  | 		return fx < fy || math.IsNaN(fx) && !math.IsNaN(fy) | ||||||
|  | 	case reflect.Complex64, reflect.Complex128: | ||||||
|  | 		cx, cy := x.Complex(), y.Complex() | ||||||
|  | 		rx, ix, ry, iy := real(cx), imag(cx), real(cy), imag(cy) | ||||||
|  | 		if rx == ry || (math.IsNaN(rx) && math.IsNaN(ry)) { | ||||||
|  | 			return ix < iy || math.IsNaN(ix) && !math.IsNaN(iy) | ||||||
|  | 		} | ||||||
|  | 		return rx < ry || math.IsNaN(rx) && !math.IsNaN(ry) | ||||||
|  | 	case reflect.Ptr, reflect.UnsafePointer, reflect.Chan: | ||||||
|  | 		return x.Pointer() < y.Pointer() | ||||||
|  | 	case reflect.String: | ||||||
|  | 		return x.String() < y.String() | ||||||
|  | 	case reflect.Array: | ||||||
|  | 		for i := 0; i < x.Len(); i++ { | ||||||
|  | 			if isLess(x.Index(i), y.Index(i)) { | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			if isLess(y.Index(i), x.Index(i)) { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return false | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		for i := 0; i < x.NumField(); i++ { | ||||||
|  | 			if isLess(x.Field(i), y.Field(i)) { | ||||||
|  | 				return true | ||||||
|  | 			} | ||||||
|  | 			if isLess(y.Field(i), x.Field(i)) { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return false | ||||||
|  | 	case reflect.Interface: | ||||||
|  | 		vx, vy := x.Elem(), y.Elem() | ||||||
|  | 		if !vx.IsValid() || !vy.IsValid() { | ||||||
|  | 			return !vx.IsValid() && vy.IsValid() | ||||||
|  | 		} | ||||||
|  | 		tx, ty := vx.Type(), vy.Type() | ||||||
|  | 		if tx == ty { | ||||||
|  | 			return isLess(x.Elem(), y.Elem()) | ||||||
|  | 		} | ||||||
|  | 		if tx.Kind() != ty.Kind() { | ||||||
|  | 			return vx.Kind() < vy.Kind() | ||||||
|  | 		} | ||||||
|  | 		if tx.String() != ty.String() { | ||||||
|  | 			return tx.String() < ty.String() | ||||||
|  | 		} | ||||||
|  | 		if tx.PkgPath() != ty.PkgPath() { | ||||||
|  | 			return tx.PkgPath() < ty.PkgPath() | ||||||
|  | 		} | ||||||
|  | 		// This can happen in rare situations, so we fallback to just comparing | ||||||
|  | 		// the unique pointer for a reflect.Type. This guarantees deterministic | ||||||
|  | 		// ordering within a program, but it is obviously not stable. | ||||||
|  | 		return reflect.ValueOf(vx.Type()).Pointer() < reflect.ValueOf(vy.Type()).Pointer() | ||||||
|  | 	default: | ||||||
|  | 		// Must be Func, Map, or Slice; which are not comparable. | ||||||
|  | 		panic(fmt.Sprintf("%T is not comparable", x.Type())) | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										554
									
								
								vendor/github.com/google/go-cmp/cmp/options.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										554
									
								
								vendor/github.com/google/go-cmp/cmp/options.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,554 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"regexp" | ||||||
|  | 	"strings" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/function" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // Option configures for specific behavior of [Equal] and [Diff]. In particular, | ||||||
|  | // the fundamental Option functions ([Ignore], [Transformer], and [Comparer]), | ||||||
|  | // configure how equality is determined. | ||||||
|  | // | ||||||
|  | // The fundamental options may be composed with filters ([FilterPath] and | ||||||
|  | // [FilterValues]) to control the scope over which they are applied. | ||||||
|  | // | ||||||
|  | // The [github.com/google/go-cmp/cmp/cmpopts] package provides helper functions | ||||||
|  | // for creating options that may be used with [Equal] and [Diff]. | ||||||
|  | type Option interface { | ||||||
|  | 	// filter applies all filters and returns the option that remains. | ||||||
|  | 	// Each option may only read s.curPath and call s.callTTBFunc. | ||||||
|  | 	// | ||||||
|  | 	// An Options is returned only if multiple comparers or transformers | ||||||
|  | 	// can apply simultaneously and will only contain values of those types | ||||||
|  | 	// or sub-Options containing values of those types. | ||||||
|  | 	filter(s *state, t reflect.Type, vx, vy reflect.Value) applicableOption | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // applicableOption represents the following types: | ||||||
|  | // | ||||||
|  | //	Fundamental: ignore | validator | *comparer | *transformer | ||||||
|  | //	Grouping:    Options | ||||||
|  | type applicableOption interface { | ||||||
|  | 	Option | ||||||
|  | 
 | ||||||
|  | 	// apply executes the option, which may mutate s or panic. | ||||||
|  | 	apply(s *state, vx, vy reflect.Value) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // coreOption represents the following types: | ||||||
|  | // | ||||||
|  | //	Fundamental: ignore | validator | *comparer | *transformer | ||||||
|  | //	Filters:     *pathFilter | *valuesFilter | ||||||
|  | type coreOption interface { | ||||||
|  | 	Option | ||||||
|  | 	isCore() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type core struct{} | ||||||
|  | 
 | ||||||
|  | func (core) isCore() {} | ||||||
|  | 
 | ||||||
|  | // Options is a list of [Option] values that also satisfies the [Option] interface. | ||||||
|  | // Helper comparison packages may return an Options value when packing multiple | ||||||
|  | // [Option] values into a single [Option]. When this package processes an Options, | ||||||
|  | // it will be implicitly expanded into a flat list. | ||||||
|  | // | ||||||
|  | // Applying a filter on an Options is equivalent to applying that same filter | ||||||
|  | // on all individual options held within. | ||||||
|  | type Options []Option | ||||||
|  | 
 | ||||||
|  | func (opts Options) filter(s *state, t reflect.Type, vx, vy reflect.Value) (out applicableOption) { | ||||||
|  | 	for _, opt := range opts { | ||||||
|  | 		switch opt := opt.filter(s, t, vx, vy); opt.(type) { | ||||||
|  | 		case ignore: | ||||||
|  | 			return ignore{} // Only ignore can short-circuit evaluation | ||||||
|  | 		case validator: | ||||||
|  | 			out = validator{} // Takes precedence over comparer or transformer | ||||||
|  | 		case *comparer, *transformer, Options: | ||||||
|  | 			switch out.(type) { | ||||||
|  | 			case nil: | ||||||
|  | 				out = opt | ||||||
|  | 			case validator: | ||||||
|  | 				// Keep validator | ||||||
|  | 			case *comparer, *transformer, Options: | ||||||
|  | 				out = Options{out, opt} // Conflicting comparers or transformers | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts Options) apply(s *state, _, _ reflect.Value) { | ||||||
|  | 	const warning = "ambiguous set of applicable options" | ||||||
|  | 	const help = "consider using filters to ensure at most one Comparer or Transformer may apply" | ||||||
|  | 	var ss []string | ||||||
|  | 	for _, opt := range flattenOptions(nil, opts) { | ||||||
|  | 		ss = append(ss, fmt.Sprint(opt)) | ||||||
|  | 	} | ||||||
|  | 	set := strings.Join(ss, "\n\t") | ||||||
|  | 	panic(fmt.Sprintf("%s at %#v:\n\t%s\n%s", warning, s.curPath, set, help)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts Options) String() string { | ||||||
|  | 	var ss []string | ||||||
|  | 	for _, opt := range opts { | ||||||
|  | 		ss = append(ss, fmt.Sprint(opt)) | ||||||
|  | 	} | ||||||
|  | 	return fmt.Sprintf("Options{%s}", strings.Join(ss, ", ")) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FilterPath returns a new [Option] where opt is only evaluated if filter f | ||||||
|  | // returns true for the current [Path] in the value tree. | ||||||
|  | // | ||||||
|  | // This filter is called even if a slice element or map entry is missing and | ||||||
|  | // provides an opportunity to ignore such cases. The filter function must be | ||||||
|  | // symmetric such that the filter result is identical regardless of whether the | ||||||
|  | // missing value is from x or y. | ||||||
|  | // | ||||||
|  | // The option passed in may be an [Ignore], [Transformer], [Comparer], [Options], or | ||||||
|  | // a previously filtered [Option]. | ||||||
|  | func FilterPath(f func(Path) bool, opt Option) Option { | ||||||
|  | 	if f == nil { | ||||||
|  | 		panic("invalid path filter function") | ||||||
|  | 	} | ||||||
|  | 	if opt := normalizeOption(opt); opt != nil { | ||||||
|  | 		return &pathFilter{fnc: f, opt: opt} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type pathFilter struct { | ||||||
|  | 	core | ||||||
|  | 	fnc func(Path) bool | ||||||
|  | 	opt Option | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (f pathFilter) filter(s *state, t reflect.Type, vx, vy reflect.Value) applicableOption { | ||||||
|  | 	if f.fnc(s.curPath) { | ||||||
|  | 		return f.opt.filter(s, t, vx, vy) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (f pathFilter) String() string { | ||||||
|  | 	return fmt.Sprintf("FilterPath(%s, %v)", function.NameOf(reflect.ValueOf(f.fnc)), f.opt) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FilterValues returns a new [Option] where opt is only evaluated if filter f, | ||||||
|  | // which is a function of the form "func(T, T) bool", returns true for the | ||||||
|  | // current pair of values being compared. If either value is invalid or | ||||||
|  | // the type of the values is not assignable to T, then this filter implicitly | ||||||
|  | // returns false. | ||||||
|  | // | ||||||
|  | // The filter function must be | ||||||
|  | // symmetric (i.e., agnostic to the order of the inputs) and | ||||||
|  | // deterministic (i.e., produces the same result when given the same inputs). | ||||||
|  | // If T is an interface, it is possible that f is called with two values with | ||||||
|  | // different concrete types that both implement T. | ||||||
|  | // | ||||||
|  | // The option passed in may be an [Ignore], [Transformer], [Comparer], [Options], or | ||||||
|  | // a previously filtered [Option]. | ||||||
|  | func FilterValues(f interface{}, opt Option) Option { | ||||||
|  | 	v := reflect.ValueOf(f) | ||||||
|  | 	if !function.IsType(v.Type(), function.ValueFilter) || v.IsNil() { | ||||||
|  | 		panic(fmt.Sprintf("invalid values filter function: %T", f)) | ||||||
|  | 	} | ||||||
|  | 	if opt := normalizeOption(opt); opt != nil { | ||||||
|  | 		vf := &valuesFilter{fnc: v, opt: opt} | ||||||
|  | 		if ti := v.Type().In(0); ti.Kind() != reflect.Interface || ti.NumMethod() > 0 { | ||||||
|  | 			vf.typ = ti | ||||||
|  | 		} | ||||||
|  | 		return vf | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type valuesFilter struct { | ||||||
|  | 	core | ||||||
|  | 	typ reflect.Type  // T | ||||||
|  | 	fnc reflect.Value // func(T, T) bool | ||||||
|  | 	opt Option | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (f valuesFilter) filter(s *state, t reflect.Type, vx, vy reflect.Value) applicableOption { | ||||||
|  | 	if !vx.IsValid() || !vx.CanInterface() || !vy.IsValid() || !vy.CanInterface() { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	if (f.typ == nil || t.AssignableTo(f.typ)) && s.callTTBFunc(f.fnc, vx, vy) { | ||||||
|  | 		return f.opt.filter(s, t, vx, vy) | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (f valuesFilter) String() string { | ||||||
|  | 	return fmt.Sprintf("FilterValues(%s, %v)", function.NameOf(f.fnc), f.opt) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Ignore is an [Option] that causes all comparisons to be ignored. | ||||||
|  | // This value is intended to be combined with [FilterPath] or [FilterValues]. | ||||||
|  | // It is an error to pass an unfiltered Ignore option to [Equal]. | ||||||
|  | func Ignore() Option { return ignore{} } | ||||||
|  | 
 | ||||||
|  | type ignore struct{ core } | ||||||
|  | 
 | ||||||
|  | func (ignore) isFiltered() bool                                                     { return false } | ||||||
|  | func (ignore) filter(_ *state, _ reflect.Type, _, _ reflect.Value) applicableOption { return ignore{} } | ||||||
|  | func (ignore) apply(s *state, _, _ reflect.Value)                                   { s.report(true, reportByIgnore) } | ||||||
|  | func (ignore) String() string                                                       { return "Ignore()" } | ||||||
|  | 
 | ||||||
|  | // validator is a sentinel Option type to indicate that some options could not | ||||||
|  | // be evaluated due to unexported fields, missing slice elements, or | ||||||
|  | // missing map entries. Both values are validator only for unexported fields. | ||||||
|  | type validator struct{ core } | ||||||
|  | 
 | ||||||
|  | func (validator) filter(_ *state, _ reflect.Type, vx, vy reflect.Value) applicableOption { | ||||||
|  | 	if !vx.IsValid() || !vy.IsValid() { | ||||||
|  | 		return validator{} | ||||||
|  | 	} | ||||||
|  | 	if !vx.CanInterface() || !vy.CanInterface() { | ||||||
|  | 		return validator{} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | func (validator) apply(s *state, vx, vy reflect.Value) { | ||||||
|  | 	// Implies missing slice element or map entry. | ||||||
|  | 	if !vx.IsValid() || !vy.IsValid() { | ||||||
|  | 		s.report(vx.IsValid() == vy.IsValid(), 0) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Unable to Interface implies unexported field without visibility access. | ||||||
|  | 	if !vx.CanInterface() || !vy.CanInterface() { | ||||||
|  | 		help := "consider using a custom Comparer; if you control the implementation of type, you can also consider using an Exporter, AllowUnexported, or cmpopts.IgnoreUnexported" | ||||||
|  | 		var name string | ||||||
|  | 		if t := s.curPath.Index(-2).Type(); t.Name() != "" { | ||||||
|  | 			// Named type with unexported fields. | ||||||
|  | 			name = fmt.Sprintf("%q.%v", t.PkgPath(), t.Name()) // e.g., "path/to/package".MyType | ||||||
|  | 			if _, ok := reflect.New(t).Interface().(error); ok { | ||||||
|  | 				help = "consider using cmpopts.EquateErrors to compare error values" | ||||||
|  | 			} else if t.Comparable() { | ||||||
|  | 				help = "consider using cmpopts.EquateComparable to compare comparable Go types" | ||||||
|  | 			} | ||||||
|  | 		} else { | ||||||
|  | 			// Unnamed type with unexported fields. Derive PkgPath from field. | ||||||
|  | 			var pkgPath string | ||||||
|  | 			for i := 0; i < t.NumField() && pkgPath == ""; i++ { | ||||||
|  | 				pkgPath = t.Field(i).PkgPath | ||||||
|  | 			} | ||||||
|  | 			name = fmt.Sprintf("%q.(%v)", pkgPath, t.String()) // e.g., "path/to/package".(struct { a int }) | ||||||
|  | 		} | ||||||
|  | 		panic(fmt.Sprintf("cannot handle unexported field at %#v:\n\t%v\n%s", s.curPath, name, help)) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	panic("not reachable") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // identRx represents a valid identifier according to the Go specification. | ||||||
|  | const identRx = `[_\p{L}][_\p{L}\p{N}]*` | ||||||
|  | 
 | ||||||
|  | var identsRx = regexp.MustCompile(`^` + identRx + `(\.` + identRx + `)*$`) | ||||||
|  | 
 | ||||||
|  | // Transformer returns an [Option] that applies a transformation function that | ||||||
|  | // converts values of a certain type into that of another. | ||||||
|  | // | ||||||
|  | // The transformer f must be a function "func(T) R" that converts values of | ||||||
|  | // type T to those of type R and is implicitly filtered to input values | ||||||
|  | // assignable to T. The transformer must not mutate T in any way. | ||||||
|  | // | ||||||
|  | // To help prevent some cases of infinite recursive cycles applying the | ||||||
|  | // same transform to the output of itself (e.g., in the case where the | ||||||
|  | // input and output types are the same), an implicit filter is added such that | ||||||
|  | // a transformer is applicable only if that exact transformer is not already | ||||||
|  | // in the tail of the [Path] since the last non-[Transform] step. | ||||||
|  | // For situations where the implicit filter is still insufficient, | ||||||
|  | // consider using [github.com/google/go-cmp/cmp/cmpopts.AcyclicTransformer], | ||||||
|  | // which adds a filter to prevent the transformer from | ||||||
|  | // being recursively applied upon itself. | ||||||
|  | // | ||||||
|  | // The name is a user provided label that is used as the [Transform.Name] in the | ||||||
|  | // transformation [PathStep] (and eventually shown in the [Diff] output). | ||||||
|  | // The name must be a valid identifier or qualified identifier in Go syntax. | ||||||
|  | // If empty, an arbitrary name is used. | ||||||
|  | func Transformer(name string, f interface{}) Option { | ||||||
|  | 	v := reflect.ValueOf(f) | ||||||
|  | 	if !function.IsType(v.Type(), function.Transformer) || v.IsNil() { | ||||||
|  | 		panic(fmt.Sprintf("invalid transformer function: %T", f)) | ||||||
|  | 	} | ||||||
|  | 	if name == "" { | ||||||
|  | 		name = function.NameOf(v) | ||||||
|  | 		if !identsRx.MatchString(name) { | ||||||
|  | 			name = "λ" // Lambda-symbol as placeholder name | ||||||
|  | 		} | ||||||
|  | 	} else if !identsRx.MatchString(name) { | ||||||
|  | 		panic(fmt.Sprintf("invalid name: %q", name)) | ||||||
|  | 	} | ||||||
|  | 	tr := &transformer{name: name, fnc: reflect.ValueOf(f)} | ||||||
|  | 	if ti := v.Type().In(0); ti.Kind() != reflect.Interface || ti.NumMethod() > 0 { | ||||||
|  | 		tr.typ = ti | ||||||
|  | 	} | ||||||
|  | 	return tr | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type transformer struct { | ||||||
|  | 	core | ||||||
|  | 	name string | ||||||
|  | 	typ  reflect.Type  // T | ||||||
|  | 	fnc  reflect.Value // func(T) R | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (tr *transformer) isFiltered() bool { return tr.typ != nil } | ||||||
|  | 
 | ||||||
|  | func (tr *transformer) filter(s *state, t reflect.Type, _, _ reflect.Value) applicableOption { | ||||||
|  | 	for i := len(s.curPath) - 1; i >= 0; i-- { | ||||||
|  | 		if t, ok := s.curPath[i].(Transform); !ok { | ||||||
|  | 			break // Hit most recent non-Transform step | ||||||
|  | 		} else if tr == t.trans { | ||||||
|  | 			return nil // Cannot directly use same Transform | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if tr.typ == nil || t.AssignableTo(tr.typ) { | ||||||
|  | 		return tr | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (tr *transformer) apply(s *state, vx, vy reflect.Value) { | ||||||
|  | 	step := Transform{&transform{pathStep{typ: tr.fnc.Type().Out(0)}, tr}} | ||||||
|  | 	vvx := s.callTRFunc(tr.fnc, vx, step) | ||||||
|  | 	vvy := s.callTRFunc(tr.fnc, vy, step) | ||||||
|  | 	step.vx, step.vy = vvx, vvy | ||||||
|  | 	s.compareAny(step) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (tr transformer) String() string { | ||||||
|  | 	return fmt.Sprintf("Transformer(%s, %s)", tr.name, function.NameOf(tr.fnc)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Comparer returns an [Option] that determines whether two values are equal | ||||||
|  | // to each other. | ||||||
|  | // | ||||||
|  | // The comparer f must be a function "func(T, T) bool" and is implicitly | ||||||
|  | // filtered to input values assignable to T. If T is an interface, it is | ||||||
|  | // possible that f is called with two values of different concrete types that | ||||||
|  | // both implement T. | ||||||
|  | // | ||||||
|  | // The equality function must be: | ||||||
|  | //   - Symmetric: equal(x, y) == equal(y, x) | ||||||
|  | //   - Deterministic: equal(x, y) == equal(x, y) | ||||||
|  | //   - Pure: equal(x, y) does not modify x or y | ||||||
|  | func Comparer(f interface{}) Option { | ||||||
|  | 	v := reflect.ValueOf(f) | ||||||
|  | 	if !function.IsType(v.Type(), function.Equal) || v.IsNil() { | ||||||
|  | 		panic(fmt.Sprintf("invalid comparer function: %T", f)) | ||||||
|  | 	} | ||||||
|  | 	cm := &comparer{fnc: v} | ||||||
|  | 	if ti := v.Type().In(0); ti.Kind() != reflect.Interface || ti.NumMethod() > 0 { | ||||||
|  | 		cm.typ = ti | ||||||
|  | 	} | ||||||
|  | 	return cm | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type comparer struct { | ||||||
|  | 	core | ||||||
|  | 	typ reflect.Type  // T | ||||||
|  | 	fnc reflect.Value // func(T, T) bool | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (cm *comparer) isFiltered() bool { return cm.typ != nil } | ||||||
|  | 
 | ||||||
|  | func (cm *comparer) filter(_ *state, t reflect.Type, _, _ reflect.Value) applicableOption { | ||||||
|  | 	if cm.typ == nil || t.AssignableTo(cm.typ) { | ||||||
|  | 		return cm | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (cm *comparer) apply(s *state, vx, vy reflect.Value) { | ||||||
|  | 	eq := s.callTTBFunc(cm.fnc, vx, vy) | ||||||
|  | 	s.report(eq, reportByFunc) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (cm comparer) String() string { | ||||||
|  | 	return fmt.Sprintf("Comparer(%s)", function.NameOf(cm.fnc)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Exporter returns an [Option] that specifies whether [Equal] is allowed to | ||||||
|  | // introspect into the unexported fields of certain struct types. | ||||||
|  | // | ||||||
|  | // Users of this option must understand that comparing on unexported fields | ||||||
|  | // from external packages is not safe since changes in the internal | ||||||
|  | // implementation of some external package may cause the result of [Equal] | ||||||
|  | // to unexpectedly change. However, it may be valid to use this option on types | ||||||
|  | // defined in an internal package where the semantic meaning of an unexported | ||||||
|  | // field is in the control of the user. | ||||||
|  | // | ||||||
|  | // In many cases, a custom [Comparer] should be used instead that defines | ||||||
|  | // equality as a function of the public API of a type rather than the underlying | ||||||
|  | // unexported implementation. | ||||||
|  | // | ||||||
|  | // For example, the [reflect.Type] documentation defines equality to be determined | ||||||
|  | // by the == operator on the interface (essentially performing a shallow pointer | ||||||
|  | // comparison) and most attempts to compare *[regexp.Regexp] types are interested | ||||||
|  | // in only checking that the regular expression strings are equal. | ||||||
|  | // Both of these are accomplished using [Comparer] options: | ||||||
|  | // | ||||||
|  | //	Comparer(func(x, y reflect.Type) bool { return x == y }) | ||||||
|  | //	Comparer(func(x, y *regexp.Regexp) bool { return x.String() == y.String() }) | ||||||
|  | // | ||||||
|  | // In other cases, the [github.com/google/go-cmp/cmp/cmpopts.IgnoreUnexported] | ||||||
|  | // option can be used to ignore all unexported fields on specified struct types. | ||||||
|  | func Exporter(f func(reflect.Type) bool) Option { | ||||||
|  | 	return exporter(f) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type exporter func(reflect.Type) bool | ||||||
|  | 
 | ||||||
|  | func (exporter) filter(_ *state, _ reflect.Type, _, _ reflect.Value) applicableOption { | ||||||
|  | 	panic("not implemented") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // AllowUnexported returns an [Option] that allows [Equal] to forcibly introspect | ||||||
|  | // unexported fields of the specified struct types. | ||||||
|  | // | ||||||
|  | // See [Exporter] for the proper use of this option. | ||||||
|  | func AllowUnexported(types ...interface{}) Option { | ||||||
|  | 	m := make(map[reflect.Type]bool) | ||||||
|  | 	for _, typ := range types { | ||||||
|  | 		t := reflect.TypeOf(typ) | ||||||
|  | 		if t.Kind() != reflect.Struct { | ||||||
|  | 			panic(fmt.Sprintf("invalid struct type: %T", typ)) | ||||||
|  | 		} | ||||||
|  | 		m[t] = true | ||||||
|  | 	} | ||||||
|  | 	return exporter(func(t reflect.Type) bool { return m[t] }) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Result represents the comparison result for a single node and | ||||||
|  | // is provided by cmp when calling Report (see [Reporter]). | ||||||
|  | type Result struct { | ||||||
|  | 	_     [0]func() // Make Result incomparable | ||||||
|  | 	flags resultFlags | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Equal reports whether the node was determined to be equal or not. | ||||||
|  | // As a special case, ignored nodes are considered equal. | ||||||
|  | func (r Result) Equal() bool { | ||||||
|  | 	return r.flags&(reportEqual|reportByIgnore) != 0 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // ByIgnore reports whether the node is equal because it was ignored. | ||||||
|  | // This never reports true if [Result.Equal] reports false. | ||||||
|  | func (r Result) ByIgnore() bool { | ||||||
|  | 	return r.flags&reportByIgnore != 0 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // ByMethod reports whether the Equal method determined equality. | ||||||
|  | func (r Result) ByMethod() bool { | ||||||
|  | 	return r.flags&reportByMethod != 0 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // ByFunc reports whether a [Comparer] function determined equality. | ||||||
|  | func (r Result) ByFunc() bool { | ||||||
|  | 	return r.flags&reportByFunc != 0 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // ByCycle reports whether a reference cycle was detected. | ||||||
|  | func (r Result) ByCycle() bool { | ||||||
|  | 	return r.flags&reportByCycle != 0 | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type resultFlags uint | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	_ resultFlags = (1 << iota) / 2 | ||||||
|  | 
 | ||||||
|  | 	reportEqual | ||||||
|  | 	reportUnequal | ||||||
|  | 	reportByIgnore | ||||||
|  | 	reportByMethod | ||||||
|  | 	reportByFunc | ||||||
|  | 	reportByCycle | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // Reporter is an [Option] that can be passed to [Equal]. When [Equal] traverses | ||||||
|  | // the value trees, it calls PushStep as it descends into each node in the | ||||||
|  | // tree and PopStep as it ascend out of the node. The leaves of the tree are | ||||||
|  | // either compared (determined to be equal or not equal) or ignored and reported | ||||||
|  | // as such by calling the Report method. | ||||||
|  | func Reporter(r interface { | ||||||
|  | 	// PushStep is called when a tree-traversal operation is performed. | ||||||
|  | 	// The PathStep itself is only valid until the step is popped. | ||||||
|  | 	// The PathStep.Values are valid for the duration of the entire traversal | ||||||
|  | 	// and must not be mutated. | ||||||
|  | 	// | ||||||
|  | 	// Equal always calls PushStep at the start to provide an operation-less | ||||||
|  | 	// PathStep used to report the root values. | ||||||
|  | 	// | ||||||
|  | 	// Within a slice, the exact set of inserted, removed, or modified elements | ||||||
|  | 	// is unspecified and may change in future implementations. | ||||||
|  | 	// The entries of a map are iterated through in an unspecified order. | ||||||
|  | 	PushStep(PathStep) | ||||||
|  | 
 | ||||||
|  | 	// Report is called exactly once on leaf nodes to report whether the | ||||||
|  | 	// comparison identified the node as equal, unequal, or ignored. | ||||||
|  | 	// A leaf node is one that is immediately preceded by and followed by | ||||||
|  | 	// a pair of PushStep and PopStep calls. | ||||||
|  | 	Report(Result) | ||||||
|  | 
 | ||||||
|  | 	// PopStep ascends back up the value tree. | ||||||
|  | 	// There is always a matching pop call for every push call. | ||||||
|  | 	PopStep() | ||||||
|  | }) Option { | ||||||
|  | 	return reporter{r} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type reporter struct{ reporterIface } | ||||||
|  | type reporterIface interface { | ||||||
|  | 	PushStep(PathStep) | ||||||
|  | 	Report(Result) | ||||||
|  | 	PopStep() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (reporter) filter(_ *state, _ reflect.Type, _, _ reflect.Value) applicableOption { | ||||||
|  | 	panic("not implemented") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // normalizeOption normalizes the input options such that all Options groups | ||||||
|  | // are flattened and groups with a single element are reduced to that element. | ||||||
|  | // Only coreOptions and Options containing coreOptions are allowed. | ||||||
|  | func normalizeOption(src Option) Option { | ||||||
|  | 	switch opts := flattenOptions(nil, Options{src}); len(opts) { | ||||||
|  | 	case 0: | ||||||
|  | 		return nil | ||||||
|  | 	case 1: | ||||||
|  | 		return opts[0] | ||||||
|  | 	default: | ||||||
|  | 		return opts | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // flattenOptions copies all options in src to dst as a flat list. | ||||||
|  | // Only coreOptions and Options containing coreOptions are allowed. | ||||||
|  | func flattenOptions(dst, src Options) Options { | ||||||
|  | 	for _, opt := range src { | ||||||
|  | 		switch opt := opt.(type) { | ||||||
|  | 		case nil: | ||||||
|  | 			continue | ||||||
|  | 		case Options: | ||||||
|  | 			dst = flattenOptions(dst, opt) | ||||||
|  | 		case coreOption: | ||||||
|  | 			dst = append(dst, opt) | ||||||
|  | 		default: | ||||||
|  | 			panic(fmt.Sprintf("invalid option type: %T", opt)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return dst | ||||||
|  | } | ||||||
							
								
								
									
										390
									
								
								vendor/github.com/google/go-cmp/cmp/path.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										390
									
								
								vendor/github.com/google/go-cmp/cmp/path.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,390 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strings" | ||||||
|  | 	"unicode" | ||||||
|  | 	"unicode/utf8" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/value" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // Path is a list of [PathStep] describing the sequence of operations to get | ||||||
|  | // from some root type to the current position in the value tree. | ||||||
|  | // The first Path element is always an operation-less [PathStep] that exists | ||||||
|  | // simply to identify the initial type. | ||||||
|  | // | ||||||
|  | // When traversing structs with embedded structs, the embedded struct will | ||||||
|  | // always be accessed as a field before traversing the fields of the | ||||||
|  | // embedded struct themselves. That is, an exported field from the | ||||||
|  | // embedded struct will never be accessed directly from the parent struct. | ||||||
|  | type Path []PathStep | ||||||
|  | 
 | ||||||
|  | // PathStep is a union-type for specific operations to traverse | ||||||
|  | // a value's tree structure. Users of this package never need to implement | ||||||
|  | // these types as values of this type will be returned by this package. | ||||||
|  | // | ||||||
|  | // Implementations of this interface: | ||||||
|  | //   - [StructField] | ||||||
|  | //   - [SliceIndex] | ||||||
|  | //   - [MapIndex] | ||||||
|  | //   - [Indirect] | ||||||
|  | //   - [TypeAssertion] | ||||||
|  | //   - [Transform] | ||||||
|  | type PathStep interface { | ||||||
|  | 	String() string | ||||||
|  | 
 | ||||||
|  | 	// Type is the resulting type after performing the path step. | ||||||
|  | 	Type() reflect.Type | ||||||
|  | 
 | ||||||
|  | 	// Values is the resulting values after performing the path step. | ||||||
|  | 	// The type of each valid value is guaranteed to be identical to Type. | ||||||
|  | 	// | ||||||
|  | 	// In some cases, one or both may be invalid or have restrictions: | ||||||
|  | 	//   - For StructField, both are not interface-able if the current field | ||||||
|  | 	//     is unexported and the struct type is not explicitly permitted by | ||||||
|  | 	//     an Exporter to traverse unexported fields. | ||||||
|  | 	//   - For SliceIndex, one may be invalid if an element is missing from | ||||||
|  | 	//     either the x or y slice. | ||||||
|  | 	//   - For MapIndex, one may be invalid if an entry is missing from | ||||||
|  | 	//     either the x or y map. | ||||||
|  | 	// | ||||||
|  | 	// The provided values must not be mutated. | ||||||
|  | 	Values() (vx, vy reflect.Value) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | var ( | ||||||
|  | 	_ PathStep = StructField{} | ||||||
|  | 	_ PathStep = SliceIndex{} | ||||||
|  | 	_ PathStep = MapIndex{} | ||||||
|  | 	_ PathStep = Indirect{} | ||||||
|  | 	_ PathStep = TypeAssertion{} | ||||||
|  | 	_ PathStep = Transform{} | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func (pa *Path) push(s PathStep) { | ||||||
|  | 	*pa = append(*pa, s) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (pa *Path) pop() { | ||||||
|  | 	*pa = (*pa)[:len(*pa)-1] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Last returns the last [PathStep] in the Path. | ||||||
|  | // If the path is empty, this returns a non-nil [PathStep] | ||||||
|  | // that reports a nil [PathStep.Type]. | ||||||
|  | func (pa Path) Last() PathStep { | ||||||
|  | 	return pa.Index(-1) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Index returns the ith step in the Path and supports negative indexing. | ||||||
|  | // A negative index starts counting from the tail of the Path such that -1 | ||||||
|  | // refers to the last step, -2 refers to the second-to-last step, and so on. | ||||||
|  | // If index is invalid, this returns a non-nil [PathStep] | ||||||
|  | // that reports a nil [PathStep.Type]. | ||||||
|  | func (pa Path) Index(i int) PathStep { | ||||||
|  | 	if i < 0 { | ||||||
|  | 		i = len(pa) + i | ||||||
|  | 	} | ||||||
|  | 	if i < 0 || i >= len(pa) { | ||||||
|  | 		return pathStep{} | ||||||
|  | 	} | ||||||
|  | 	return pa[i] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // String returns the simplified path to a node. | ||||||
|  | // The simplified path only contains struct field accesses. | ||||||
|  | // | ||||||
|  | // For example: | ||||||
|  | // | ||||||
|  | //	MyMap.MySlices.MyField | ||||||
|  | func (pa Path) String() string { | ||||||
|  | 	var ss []string | ||||||
|  | 	for _, s := range pa { | ||||||
|  | 		if _, ok := s.(StructField); ok { | ||||||
|  | 			ss = append(ss, s.String()) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return strings.TrimPrefix(strings.Join(ss, ""), ".") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // GoString returns the path to a specific node using Go syntax. | ||||||
|  | // | ||||||
|  | // For example: | ||||||
|  | // | ||||||
|  | //	(*root.MyMap["key"].(*mypkg.MyStruct).MySlices)[2][3].MyField | ||||||
|  | func (pa Path) GoString() string { | ||||||
|  | 	var ssPre, ssPost []string | ||||||
|  | 	var numIndirect int | ||||||
|  | 	for i, s := range pa { | ||||||
|  | 		var nextStep PathStep | ||||||
|  | 		if i+1 < len(pa) { | ||||||
|  | 			nextStep = pa[i+1] | ||||||
|  | 		} | ||||||
|  | 		switch s := s.(type) { | ||||||
|  | 		case Indirect: | ||||||
|  | 			numIndirect++ | ||||||
|  | 			pPre, pPost := "(", ")" | ||||||
|  | 			switch nextStep.(type) { | ||||||
|  | 			case Indirect: | ||||||
|  | 				continue // Next step is indirection, so let them batch up | ||||||
|  | 			case StructField: | ||||||
|  | 				numIndirect-- // Automatic indirection on struct fields | ||||||
|  | 			case nil: | ||||||
|  | 				pPre, pPost = "", "" // Last step; no need for parenthesis | ||||||
|  | 			} | ||||||
|  | 			if numIndirect > 0 { | ||||||
|  | 				ssPre = append(ssPre, pPre+strings.Repeat("*", numIndirect)) | ||||||
|  | 				ssPost = append(ssPost, pPost) | ||||||
|  | 			} | ||||||
|  | 			numIndirect = 0 | ||||||
|  | 			continue | ||||||
|  | 		case Transform: | ||||||
|  | 			ssPre = append(ssPre, s.trans.name+"(") | ||||||
|  | 			ssPost = append(ssPost, ")") | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 		ssPost = append(ssPost, s.String()) | ||||||
|  | 	} | ||||||
|  | 	for i, j := 0, len(ssPre)-1; i < j; i, j = i+1, j-1 { | ||||||
|  | 		ssPre[i], ssPre[j] = ssPre[j], ssPre[i] | ||||||
|  | 	} | ||||||
|  | 	return strings.Join(ssPre, "") + strings.Join(ssPost, "") | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type pathStep struct { | ||||||
|  | 	typ    reflect.Type | ||||||
|  | 	vx, vy reflect.Value | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (ps pathStep) Type() reflect.Type             { return ps.typ } | ||||||
|  | func (ps pathStep) Values() (vx, vy reflect.Value) { return ps.vx, ps.vy } | ||||||
|  | func (ps pathStep) String() string { | ||||||
|  | 	if ps.typ == nil { | ||||||
|  | 		return "<nil>" | ||||||
|  | 	} | ||||||
|  | 	s := value.TypeString(ps.typ, false) | ||||||
|  | 	if s == "" || strings.ContainsAny(s, "{}\n") { | ||||||
|  | 		return "root" // Type too simple or complex to print | ||||||
|  | 	} | ||||||
|  | 	return fmt.Sprintf("{%s}", s) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // StructField is a [PathStep] that represents a struct field access | ||||||
|  | // on a field called [StructField.Name]. | ||||||
|  | type StructField struct{ *structField } | ||||||
|  | type structField struct { | ||||||
|  | 	pathStep | ||||||
|  | 	name string | ||||||
|  | 	idx  int | ||||||
|  | 
 | ||||||
|  | 	// These fields are used for forcibly accessing an unexported field. | ||||||
|  | 	// pvx, pvy, and field are only valid if unexported is true. | ||||||
|  | 	unexported bool | ||||||
|  | 	mayForce   bool                // Forcibly allow visibility | ||||||
|  | 	paddr      bool                // Was parent addressable? | ||||||
|  | 	pvx, pvy   reflect.Value       // Parent values (always addressable) | ||||||
|  | 	field      reflect.StructField // Field information | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (sf StructField) Type() reflect.Type { return sf.typ } | ||||||
|  | func (sf StructField) Values() (vx, vy reflect.Value) { | ||||||
|  | 	if !sf.unexported { | ||||||
|  | 		return sf.vx, sf.vy // CanInterface reports true | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Forcibly obtain read-write access to an unexported struct field. | ||||||
|  | 	if sf.mayForce { | ||||||
|  | 		vx = retrieveUnexportedField(sf.pvx, sf.field, sf.paddr) | ||||||
|  | 		vy = retrieveUnexportedField(sf.pvy, sf.field, sf.paddr) | ||||||
|  | 		return vx, vy // CanInterface reports true | ||||||
|  | 	} | ||||||
|  | 	return sf.vx, sf.vy // CanInterface reports false | ||||||
|  | } | ||||||
|  | func (sf StructField) String() string { return fmt.Sprintf(".%s", sf.name) } | ||||||
|  | 
 | ||||||
|  | // Name is the field name. | ||||||
|  | func (sf StructField) Name() string { return sf.name } | ||||||
|  | 
 | ||||||
|  | // Index is the index of the field in the parent struct type. | ||||||
|  | // See [reflect.Type.Field]. | ||||||
|  | func (sf StructField) Index() int { return sf.idx } | ||||||
|  | 
 | ||||||
|  | // SliceIndex is a [PathStep] that represents an index operation on | ||||||
|  | // a slice or array at some index [SliceIndex.Key]. | ||||||
|  | type SliceIndex struct{ *sliceIndex } | ||||||
|  | type sliceIndex struct { | ||||||
|  | 	pathStep | ||||||
|  | 	xkey, ykey int | ||||||
|  | 	isSlice    bool // False for reflect.Array | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (si SliceIndex) Type() reflect.Type             { return si.typ } | ||||||
|  | func (si SliceIndex) Values() (vx, vy reflect.Value) { return si.vx, si.vy } | ||||||
|  | func (si SliceIndex) String() string { | ||||||
|  | 	switch { | ||||||
|  | 	case si.xkey == si.ykey: | ||||||
|  | 		return fmt.Sprintf("[%d]", si.xkey) | ||||||
|  | 	case si.ykey == -1: | ||||||
|  | 		// [5->?] means "I don't know where X[5] went" | ||||||
|  | 		return fmt.Sprintf("[%d->?]", si.xkey) | ||||||
|  | 	case si.xkey == -1: | ||||||
|  | 		// [?->3] means "I don't know where Y[3] came from" | ||||||
|  | 		return fmt.Sprintf("[?->%d]", si.ykey) | ||||||
|  | 	default: | ||||||
|  | 		// [5->3] means "X[5] moved to Y[3]" | ||||||
|  | 		return fmt.Sprintf("[%d->%d]", si.xkey, si.ykey) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Key is the index key; it may return -1 if in a split state | ||||||
|  | func (si SliceIndex) Key() int { | ||||||
|  | 	if si.xkey != si.ykey { | ||||||
|  | 		return -1 | ||||||
|  | 	} | ||||||
|  | 	return si.xkey | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // SplitKeys are the indexes for indexing into slices in the | ||||||
|  | // x and y values, respectively. These indexes may differ due to the | ||||||
|  | // insertion or removal of an element in one of the slices, causing | ||||||
|  | // all of the indexes to be shifted. If an index is -1, then that | ||||||
|  | // indicates that the element does not exist in the associated slice. | ||||||
|  | // | ||||||
|  | // [SliceIndex.Key] is guaranteed to return -1 if and only if the indexes | ||||||
|  | // returned by SplitKeys are not the same. SplitKeys will never return -1 for | ||||||
|  | // both indexes. | ||||||
|  | func (si SliceIndex) SplitKeys() (ix, iy int) { return si.xkey, si.ykey } | ||||||
|  | 
 | ||||||
|  | // MapIndex is a [PathStep] that represents an index operation on a map at some index Key. | ||||||
|  | type MapIndex struct{ *mapIndex } | ||||||
|  | type mapIndex struct { | ||||||
|  | 	pathStep | ||||||
|  | 	key reflect.Value | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (mi MapIndex) Type() reflect.Type             { return mi.typ } | ||||||
|  | func (mi MapIndex) Values() (vx, vy reflect.Value) { return mi.vx, mi.vy } | ||||||
|  | func (mi MapIndex) String() string                 { return fmt.Sprintf("[%#v]", mi.key) } | ||||||
|  | 
 | ||||||
|  | // Key is the value of the map key. | ||||||
|  | func (mi MapIndex) Key() reflect.Value { return mi.key } | ||||||
|  | 
 | ||||||
|  | // Indirect is a [PathStep] that represents pointer indirection on the parent type. | ||||||
|  | type Indirect struct{ *indirect } | ||||||
|  | type indirect struct { | ||||||
|  | 	pathStep | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (in Indirect) Type() reflect.Type             { return in.typ } | ||||||
|  | func (in Indirect) Values() (vx, vy reflect.Value) { return in.vx, in.vy } | ||||||
|  | func (in Indirect) String() string                 { return "*" } | ||||||
|  | 
 | ||||||
|  | // TypeAssertion is a [PathStep] that represents a type assertion on an interface. | ||||||
|  | type TypeAssertion struct{ *typeAssertion } | ||||||
|  | type typeAssertion struct { | ||||||
|  | 	pathStep | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (ta TypeAssertion) Type() reflect.Type             { return ta.typ } | ||||||
|  | func (ta TypeAssertion) Values() (vx, vy reflect.Value) { return ta.vx, ta.vy } | ||||||
|  | func (ta TypeAssertion) String() string                 { return fmt.Sprintf(".(%v)", value.TypeString(ta.typ, false)) } | ||||||
|  | 
 | ||||||
|  | // Transform is a [PathStep] that represents a transformation | ||||||
|  | // from the parent type to the current type. | ||||||
|  | type Transform struct{ *transform } | ||||||
|  | type transform struct { | ||||||
|  | 	pathStep | ||||||
|  | 	trans *transformer | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (tf Transform) Type() reflect.Type             { return tf.typ } | ||||||
|  | func (tf Transform) Values() (vx, vy reflect.Value) { return tf.vx, tf.vy } | ||||||
|  | func (tf Transform) String() string                 { return fmt.Sprintf("%s()", tf.trans.name) } | ||||||
|  | 
 | ||||||
|  | // Name is the name of the [Transformer]. | ||||||
|  | func (tf Transform) Name() string { return tf.trans.name } | ||||||
|  | 
 | ||||||
|  | // Func is the function pointer to the transformer function. | ||||||
|  | func (tf Transform) Func() reflect.Value { return tf.trans.fnc } | ||||||
|  | 
 | ||||||
|  | // Option returns the originally constructed [Transformer] option. | ||||||
|  | // The == operator can be used to detect the exact option used. | ||||||
|  | func (tf Transform) Option() Option { return tf.trans } | ||||||
|  | 
 | ||||||
|  | // pointerPath represents a dual-stack of pointers encountered when | ||||||
|  | // recursively traversing the x and y values. This data structure supports | ||||||
|  | // detection of cycles and determining whether the cycles are equal. | ||||||
|  | // In Go, cycles can occur via pointers, slices, and maps. | ||||||
|  | // | ||||||
|  | // The pointerPath uses a map to represent a stack; where descension into a | ||||||
|  | // pointer pushes the address onto the stack, and ascension from a pointer | ||||||
|  | // pops the address from the stack. Thus, when traversing into a pointer from | ||||||
|  | // reflect.Ptr, reflect.Slice element, or reflect.Map, we can detect cycles | ||||||
|  | // by checking whether the pointer has already been visited. The cycle detection | ||||||
|  | // uses a separate stack for the x and y values. | ||||||
|  | // | ||||||
|  | // If a cycle is detected we need to determine whether the two pointers | ||||||
|  | // should be considered equal. The definition of equality chosen by Equal | ||||||
|  | // requires two graphs to have the same structure. To determine this, both the | ||||||
|  | // x and y values must have a cycle where the previous pointers were also | ||||||
|  | // encountered together as a pair. | ||||||
|  | // | ||||||
|  | // Semantically, this is equivalent to augmenting Indirect, SliceIndex, and | ||||||
|  | // MapIndex with pointer information for the x and y values. | ||||||
|  | // Suppose px and py are two pointers to compare, we then search the | ||||||
|  | // Path for whether px was ever encountered in the Path history of x, and | ||||||
|  | // similarly so with py. If either side has a cycle, the comparison is only | ||||||
|  | // equal if both px and py have a cycle resulting from the same PathStep. | ||||||
|  | // | ||||||
|  | // Using a map as a stack is more performant as we can perform cycle detection | ||||||
|  | // in O(1) instead of O(N) where N is len(Path). | ||||||
|  | type pointerPath struct { | ||||||
|  | 	// mx is keyed by x pointers, where the value is the associated y pointer. | ||||||
|  | 	mx map[value.Pointer]value.Pointer | ||||||
|  | 	// my is keyed by y pointers, where the value is the associated x pointer. | ||||||
|  | 	my map[value.Pointer]value.Pointer | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (p *pointerPath) Init() { | ||||||
|  | 	p.mx = make(map[value.Pointer]value.Pointer) | ||||||
|  | 	p.my = make(map[value.Pointer]value.Pointer) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Push indicates intent to descend into pointers vx and vy where | ||||||
|  | // visited reports whether either has been seen before. If visited before, | ||||||
|  | // equal reports whether both pointers were encountered together. | ||||||
|  | // Pop must be called if and only if the pointers were never visited. | ||||||
|  | // | ||||||
|  | // The pointers vx and vy must be a reflect.Ptr, reflect.Slice, or reflect.Map | ||||||
|  | // and be non-nil. | ||||||
|  | func (p pointerPath) Push(vx, vy reflect.Value) (equal, visited bool) { | ||||||
|  | 	px := value.PointerOf(vx) | ||||||
|  | 	py := value.PointerOf(vy) | ||||||
|  | 	_, ok1 := p.mx[px] | ||||||
|  | 	_, ok2 := p.my[py] | ||||||
|  | 	if ok1 || ok2 { | ||||||
|  | 		equal = p.mx[px] == py && p.my[py] == px // Pointers paired together | ||||||
|  | 		return equal, true | ||||||
|  | 	} | ||||||
|  | 	p.mx[px] = py | ||||||
|  | 	p.my[py] = px | ||||||
|  | 	return false, false | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // Pop ascends from pointers vx and vy. | ||||||
|  | func (p pointerPath) Pop(vx, vy reflect.Value) { | ||||||
|  | 	delete(p.mx, value.PointerOf(vx)) | ||||||
|  | 	delete(p.my, value.PointerOf(vy)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // isExported reports whether the identifier is exported. | ||||||
|  | func isExported(id string) bool { | ||||||
|  | 	r, _ := utf8.DecodeRuneInString(id) | ||||||
|  | 	return unicode.IsUpper(r) | ||||||
|  | } | ||||||
							
								
								
									
										54
									
								
								vendor/github.com/google/go-cmp/cmp/report.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								vendor/github.com/google/go-cmp/cmp/report.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,54 @@ | ||||||
|  | // Copyright 2017, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | // defaultReporter implements the reporter interface. | ||||||
|  | // | ||||||
|  | // As Equal serially calls the PushStep, Report, and PopStep methods, the | ||||||
|  | // defaultReporter constructs a tree-based representation of the compared value | ||||||
|  | // and the result of each comparison (see valueNode). | ||||||
|  | // | ||||||
|  | // When the String method is called, the FormatDiff method transforms the | ||||||
|  | // valueNode tree into a textNode tree, which is a tree-based representation | ||||||
|  | // of the textual output (see textNode). | ||||||
|  | // | ||||||
|  | // Lastly, the textNode.String method produces the final report as a string. | ||||||
|  | type defaultReporter struct { | ||||||
|  | 	root *valueNode | ||||||
|  | 	curr *valueNode | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (r *defaultReporter) PushStep(ps PathStep) { | ||||||
|  | 	r.curr = r.curr.PushStep(ps) | ||||||
|  | 	if r.root == nil { | ||||||
|  | 		r.root = r.curr | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | func (r *defaultReporter) Report(rs Result) { | ||||||
|  | 	r.curr.Report(rs) | ||||||
|  | } | ||||||
|  | func (r *defaultReporter) PopStep() { | ||||||
|  | 	r.curr = r.curr.PopStep() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // String provides a full report of the differences detected as a structured | ||||||
|  | // literal in pseudo-Go syntax. String may only be called after the entire tree | ||||||
|  | // has been traversed. | ||||||
|  | func (r *defaultReporter) String() string { | ||||||
|  | 	assert(r.root != nil && r.curr == nil) | ||||||
|  | 	if r.root.NumDiff == 0 { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	ptrs := new(pointerReferences) | ||||||
|  | 	text := formatOptions{}.FormatDiff(r.root, ptrs) | ||||||
|  | 	resolveReferences(text) | ||||||
|  | 	return text.String() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func assert(ok bool) { | ||||||
|  | 	if !ok { | ||||||
|  | 		panic("assertion failure") | ||||||
|  | 	} | ||||||
|  | } | ||||||
							
								
								
									
										433
									
								
								vendor/github.com/google/go-cmp/cmp/report_compare.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										433
									
								
								vendor/github.com/google/go-cmp/cmp/report_compare.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,433 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // numContextRecords is the number of surrounding equal records to print. | ||||||
|  | const numContextRecords = 2 | ||||||
|  | 
 | ||||||
|  | type diffMode byte | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	diffUnknown   diffMode = 0 | ||||||
|  | 	diffIdentical diffMode = ' ' | ||||||
|  | 	diffRemoved   diffMode = '-' | ||||||
|  | 	diffInserted  diffMode = '+' | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type typeMode int | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	// emitType always prints the type. | ||||||
|  | 	emitType typeMode = iota | ||||||
|  | 	// elideType never prints the type. | ||||||
|  | 	elideType | ||||||
|  | 	// autoType prints the type only for composite kinds | ||||||
|  | 	// (i.e., structs, slices, arrays, and maps). | ||||||
|  | 	autoType | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type formatOptions struct { | ||||||
|  | 	// DiffMode controls the output mode of FormatDiff. | ||||||
|  | 	// | ||||||
|  | 	// If diffUnknown,   then produce a diff of the x and y values. | ||||||
|  | 	// If diffIdentical, then emit values as if they were equal. | ||||||
|  | 	// If diffRemoved,   then only emit x values (ignoring y values). | ||||||
|  | 	// If diffInserted,  then only emit y values (ignoring x values). | ||||||
|  | 	DiffMode diffMode | ||||||
|  | 
 | ||||||
|  | 	// TypeMode controls whether to print the type for the current node. | ||||||
|  | 	// | ||||||
|  | 	// As a general rule of thumb, we always print the type of the next node | ||||||
|  | 	// after an interface, and always elide the type of the next node after | ||||||
|  | 	// a slice or map node. | ||||||
|  | 	TypeMode typeMode | ||||||
|  | 
 | ||||||
|  | 	// formatValueOptions are options specific to printing reflect.Values. | ||||||
|  | 	formatValueOptions | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts formatOptions) WithDiffMode(d diffMode) formatOptions { | ||||||
|  | 	opts.DiffMode = d | ||||||
|  | 	return opts | ||||||
|  | } | ||||||
|  | func (opts formatOptions) WithTypeMode(t typeMode) formatOptions { | ||||||
|  | 	opts.TypeMode = t | ||||||
|  | 	return opts | ||||||
|  | } | ||||||
|  | func (opts formatOptions) WithVerbosity(level int) formatOptions { | ||||||
|  | 	opts.VerbosityLevel = level | ||||||
|  | 	opts.LimitVerbosity = true | ||||||
|  | 	return opts | ||||||
|  | } | ||||||
|  | func (opts formatOptions) verbosity() uint { | ||||||
|  | 	switch { | ||||||
|  | 	case opts.VerbosityLevel < 0: | ||||||
|  | 		return 0 | ||||||
|  | 	case opts.VerbosityLevel > 16: | ||||||
|  | 		return 16 // some reasonable maximum to avoid shift overflow | ||||||
|  | 	default: | ||||||
|  | 		return uint(opts.VerbosityLevel) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | const maxVerbosityPreset = 6 | ||||||
|  | 
 | ||||||
|  | // verbosityPreset modifies the verbosity settings given an index | ||||||
|  | // between 0 and maxVerbosityPreset, inclusive. | ||||||
|  | func verbosityPreset(opts formatOptions, i int) formatOptions { | ||||||
|  | 	opts.VerbosityLevel = int(opts.verbosity()) + 2*i | ||||||
|  | 	if i > 0 { | ||||||
|  | 		opts.AvoidStringer = true | ||||||
|  | 	} | ||||||
|  | 	if i >= maxVerbosityPreset { | ||||||
|  | 		opts.PrintAddresses = true | ||||||
|  | 		opts.QualifiedNames = true | ||||||
|  | 	} | ||||||
|  | 	return opts | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FormatDiff converts a valueNode tree into a textNode tree, where the later | ||||||
|  | // is a textual representation of the differences detected in the former. | ||||||
|  | func (opts formatOptions) FormatDiff(v *valueNode, ptrs *pointerReferences) (out textNode) { | ||||||
|  | 	if opts.DiffMode == diffIdentical { | ||||||
|  | 		opts = opts.WithVerbosity(1) | ||||||
|  | 	} else if opts.verbosity() < 3 { | ||||||
|  | 		opts = opts.WithVerbosity(3) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Check whether we have specialized formatting for this node. | ||||||
|  | 	// This is not necessary, but helpful for producing more readable outputs. | ||||||
|  | 	if opts.CanFormatDiffSlice(v) { | ||||||
|  | 		return opts.FormatDiffSlice(v) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	var parentKind reflect.Kind | ||||||
|  | 	if v.parent != nil && v.parent.TransformerName == "" { | ||||||
|  | 		parentKind = v.parent.Type.Kind() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// For leaf nodes, format the value based on the reflect.Values alone. | ||||||
|  | 	// As a special case, treat equal []byte as a leaf nodes. | ||||||
|  | 	isBytes := v.Type.Kind() == reflect.Slice && v.Type.Elem() == byteType | ||||||
|  | 	isEqualBytes := isBytes && v.NumDiff+v.NumIgnored+v.NumTransformed == 0 | ||||||
|  | 	if v.MaxDepth == 0 || isEqualBytes { | ||||||
|  | 		switch opts.DiffMode { | ||||||
|  | 		case diffUnknown, diffIdentical: | ||||||
|  | 			// Format Equal. | ||||||
|  | 			if v.NumDiff == 0 { | ||||||
|  | 				outx := opts.FormatValue(v.ValueX, parentKind, ptrs) | ||||||
|  | 				outy := opts.FormatValue(v.ValueY, parentKind, ptrs) | ||||||
|  | 				if v.NumIgnored > 0 && v.NumSame == 0 { | ||||||
|  | 					return textEllipsis | ||||||
|  | 				} else if outx.Len() < outy.Len() { | ||||||
|  | 					return outx | ||||||
|  | 				} else { | ||||||
|  | 					return outy | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Format unequal. | ||||||
|  | 			assert(opts.DiffMode == diffUnknown) | ||||||
|  | 			var list textList | ||||||
|  | 			outx := opts.WithTypeMode(elideType).FormatValue(v.ValueX, parentKind, ptrs) | ||||||
|  | 			outy := opts.WithTypeMode(elideType).FormatValue(v.ValueY, parentKind, ptrs) | ||||||
|  | 			for i := 0; i <= maxVerbosityPreset && outx != nil && outy != nil && outx.Equal(outy); i++ { | ||||||
|  | 				opts2 := verbosityPreset(opts, i).WithTypeMode(elideType) | ||||||
|  | 				outx = opts2.FormatValue(v.ValueX, parentKind, ptrs) | ||||||
|  | 				outy = opts2.FormatValue(v.ValueY, parentKind, ptrs) | ||||||
|  | 			} | ||||||
|  | 			if outx != nil { | ||||||
|  | 				list = append(list, textRecord{Diff: '-', Value: outx}) | ||||||
|  | 			} | ||||||
|  | 			if outy != nil { | ||||||
|  | 				list = append(list, textRecord{Diff: '+', Value: outy}) | ||||||
|  | 			} | ||||||
|  | 			return opts.WithTypeMode(emitType).FormatType(v.Type, list) | ||||||
|  | 		case diffRemoved: | ||||||
|  | 			return opts.FormatValue(v.ValueX, parentKind, ptrs) | ||||||
|  | 		case diffInserted: | ||||||
|  | 			return opts.FormatValue(v.ValueY, parentKind, ptrs) | ||||||
|  | 		default: | ||||||
|  | 			panic("invalid diff mode") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Register slice element to support cycle detection. | ||||||
|  | 	if parentKind == reflect.Slice { | ||||||
|  | 		ptrRefs := ptrs.PushPair(v.ValueX, v.ValueY, opts.DiffMode, true) | ||||||
|  | 		defer ptrs.Pop() | ||||||
|  | 		defer func() { out = wrapTrunkReferences(ptrRefs, out) }() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Descend into the child value node. | ||||||
|  | 	if v.TransformerName != "" { | ||||||
|  | 		out := opts.WithTypeMode(emitType).FormatDiff(v.Value, ptrs) | ||||||
|  | 		out = &textWrap{Prefix: "Inverse(" + v.TransformerName + ", ", Value: out, Suffix: ")"} | ||||||
|  | 		return opts.FormatType(v.Type, out) | ||||||
|  | 	} else { | ||||||
|  | 		switch k := v.Type.Kind(); k { | ||||||
|  | 		case reflect.Struct, reflect.Array, reflect.Slice: | ||||||
|  | 			out = opts.formatDiffList(v.Records, k, ptrs) | ||||||
|  | 			out = opts.FormatType(v.Type, out) | ||||||
|  | 		case reflect.Map: | ||||||
|  | 			// Register map to support cycle detection. | ||||||
|  | 			ptrRefs := ptrs.PushPair(v.ValueX, v.ValueY, opts.DiffMode, false) | ||||||
|  | 			defer ptrs.Pop() | ||||||
|  | 
 | ||||||
|  | 			out = opts.formatDiffList(v.Records, k, ptrs) | ||||||
|  | 			out = wrapTrunkReferences(ptrRefs, out) | ||||||
|  | 			out = opts.FormatType(v.Type, out) | ||||||
|  | 		case reflect.Ptr: | ||||||
|  | 			// Register pointer to support cycle detection. | ||||||
|  | 			ptrRefs := ptrs.PushPair(v.ValueX, v.ValueY, opts.DiffMode, false) | ||||||
|  | 			defer ptrs.Pop() | ||||||
|  | 
 | ||||||
|  | 			out = opts.FormatDiff(v.Value, ptrs) | ||||||
|  | 			out = wrapTrunkReferences(ptrRefs, out) | ||||||
|  | 			out = &textWrap{Prefix: "&", Value: out} | ||||||
|  | 		case reflect.Interface: | ||||||
|  | 			out = opts.WithTypeMode(emitType).FormatDiff(v.Value, ptrs) | ||||||
|  | 		default: | ||||||
|  | 			panic(fmt.Sprintf("%v cannot have children", k)) | ||||||
|  | 		} | ||||||
|  | 		return out | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts formatOptions) formatDiffList(recs []reportRecord, k reflect.Kind, ptrs *pointerReferences) textNode { | ||||||
|  | 	// Derive record name based on the data structure kind. | ||||||
|  | 	var name string | ||||||
|  | 	var formatKey func(reflect.Value) string | ||||||
|  | 	switch k { | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		name = "field" | ||||||
|  | 		opts = opts.WithTypeMode(autoType) | ||||||
|  | 		formatKey = func(v reflect.Value) string { return v.String() } | ||||||
|  | 	case reflect.Slice, reflect.Array: | ||||||
|  | 		name = "element" | ||||||
|  | 		opts = opts.WithTypeMode(elideType) | ||||||
|  | 		formatKey = func(reflect.Value) string { return "" } | ||||||
|  | 	case reflect.Map: | ||||||
|  | 		name = "entry" | ||||||
|  | 		opts = opts.WithTypeMode(elideType) | ||||||
|  | 		formatKey = func(v reflect.Value) string { return formatMapKey(v, false, ptrs) } | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	maxLen := -1 | ||||||
|  | 	if opts.LimitVerbosity { | ||||||
|  | 		if opts.DiffMode == diffIdentical { | ||||||
|  | 			maxLen = ((1 << opts.verbosity()) >> 1) << 2 // 0, 4, 8, 16, 32, etc... | ||||||
|  | 		} else { | ||||||
|  | 			maxLen = (1 << opts.verbosity()) << 1 // 2, 4, 8, 16, 32, 64, etc... | ||||||
|  | 		} | ||||||
|  | 		opts.VerbosityLevel-- | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Handle unification. | ||||||
|  | 	switch opts.DiffMode { | ||||||
|  | 	case diffIdentical, diffRemoved, diffInserted: | ||||||
|  | 		var list textList | ||||||
|  | 		var deferredEllipsis bool // Add final "..." to indicate records were dropped | ||||||
|  | 		for _, r := range recs { | ||||||
|  | 			if len(list) == maxLen { | ||||||
|  | 				deferredEllipsis = true | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Elide struct fields that are zero value. | ||||||
|  | 			if k == reflect.Struct { | ||||||
|  | 				var isZero bool | ||||||
|  | 				switch opts.DiffMode { | ||||||
|  | 				case diffIdentical: | ||||||
|  | 					isZero = r.Value.ValueX.IsZero() || r.Value.ValueY.IsZero() | ||||||
|  | 				case diffRemoved: | ||||||
|  | 					isZero = r.Value.ValueX.IsZero() | ||||||
|  | 				case diffInserted: | ||||||
|  | 					isZero = r.Value.ValueY.IsZero() | ||||||
|  | 				} | ||||||
|  | 				if isZero { | ||||||
|  | 					continue | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			// Elide ignored nodes. | ||||||
|  | 			if r.Value.NumIgnored > 0 && r.Value.NumSame+r.Value.NumDiff == 0 { | ||||||
|  | 				deferredEllipsis = !(k == reflect.Slice || k == reflect.Array) | ||||||
|  | 				if !deferredEllipsis { | ||||||
|  | 					list.AppendEllipsis(diffStats{}) | ||||||
|  | 				} | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 			if out := opts.FormatDiff(r.Value, ptrs); out != nil { | ||||||
|  | 				list = append(list, textRecord{Key: formatKey(r.Key), Value: out}) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if deferredEllipsis { | ||||||
|  | 			list.AppendEllipsis(diffStats{}) | ||||||
|  | 		} | ||||||
|  | 		return &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | 	case diffUnknown: | ||||||
|  | 	default: | ||||||
|  | 		panic("invalid diff mode") | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Handle differencing. | ||||||
|  | 	var numDiffs int | ||||||
|  | 	var list textList | ||||||
|  | 	var keys []reflect.Value // invariant: len(list) == len(keys) | ||||||
|  | 	groups := coalesceAdjacentRecords(name, recs) | ||||||
|  | 	maxGroup := diffStats{Name: name} | ||||||
|  | 	for i, ds := range groups { | ||||||
|  | 		if maxLen >= 0 && numDiffs >= maxLen { | ||||||
|  | 			maxGroup = maxGroup.Append(ds) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Handle equal records. | ||||||
|  | 		if ds.NumDiff() == 0 { | ||||||
|  | 			// Compute the number of leading and trailing records to print. | ||||||
|  | 			var numLo, numHi int | ||||||
|  | 			numEqual := ds.NumIgnored + ds.NumIdentical | ||||||
|  | 			for numLo < numContextRecords && numLo+numHi < numEqual && i != 0 { | ||||||
|  | 				if r := recs[numLo].Value; r.NumIgnored > 0 && r.NumSame+r.NumDiff == 0 { | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 				numLo++ | ||||||
|  | 			} | ||||||
|  | 			for numHi < numContextRecords && numLo+numHi < numEqual && i != len(groups)-1 { | ||||||
|  | 				if r := recs[numEqual-numHi-1].Value; r.NumIgnored > 0 && r.NumSame+r.NumDiff == 0 { | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 				numHi++ | ||||||
|  | 			} | ||||||
|  | 			if numEqual-(numLo+numHi) == 1 && ds.NumIgnored == 0 { | ||||||
|  | 				numHi++ // Avoid pointless coalescing of a single equal record | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Format the equal values. | ||||||
|  | 			for _, r := range recs[:numLo] { | ||||||
|  | 				out := opts.WithDiffMode(diffIdentical).FormatDiff(r.Value, ptrs) | ||||||
|  | 				list = append(list, textRecord{Key: formatKey(r.Key), Value: out}) | ||||||
|  | 				keys = append(keys, r.Key) | ||||||
|  | 			} | ||||||
|  | 			if numEqual > numLo+numHi { | ||||||
|  | 				ds.NumIdentical -= numLo + numHi | ||||||
|  | 				list.AppendEllipsis(ds) | ||||||
|  | 				for len(keys) < len(list) { | ||||||
|  | 					keys = append(keys, reflect.Value{}) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			for _, r := range recs[numEqual-numHi : numEqual] { | ||||||
|  | 				out := opts.WithDiffMode(diffIdentical).FormatDiff(r.Value, ptrs) | ||||||
|  | 				list = append(list, textRecord{Key: formatKey(r.Key), Value: out}) | ||||||
|  | 				keys = append(keys, r.Key) | ||||||
|  | 			} | ||||||
|  | 			recs = recs[numEqual:] | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Handle unequal records. | ||||||
|  | 		for _, r := range recs[:ds.NumDiff()] { | ||||||
|  | 			switch { | ||||||
|  | 			case opts.CanFormatDiffSlice(r.Value): | ||||||
|  | 				out := opts.FormatDiffSlice(r.Value) | ||||||
|  | 				list = append(list, textRecord{Key: formatKey(r.Key), Value: out}) | ||||||
|  | 				keys = append(keys, r.Key) | ||||||
|  | 			case r.Value.NumChildren == r.Value.MaxDepth: | ||||||
|  | 				outx := opts.WithDiffMode(diffRemoved).FormatDiff(r.Value, ptrs) | ||||||
|  | 				outy := opts.WithDiffMode(diffInserted).FormatDiff(r.Value, ptrs) | ||||||
|  | 				for i := 0; i <= maxVerbosityPreset && outx != nil && outy != nil && outx.Equal(outy); i++ { | ||||||
|  | 					opts2 := verbosityPreset(opts, i) | ||||||
|  | 					outx = opts2.WithDiffMode(diffRemoved).FormatDiff(r.Value, ptrs) | ||||||
|  | 					outy = opts2.WithDiffMode(diffInserted).FormatDiff(r.Value, ptrs) | ||||||
|  | 				} | ||||||
|  | 				if outx != nil { | ||||||
|  | 					list = append(list, textRecord{Diff: diffRemoved, Key: formatKey(r.Key), Value: outx}) | ||||||
|  | 					keys = append(keys, r.Key) | ||||||
|  | 				} | ||||||
|  | 				if outy != nil { | ||||||
|  | 					list = append(list, textRecord{Diff: diffInserted, Key: formatKey(r.Key), Value: outy}) | ||||||
|  | 					keys = append(keys, r.Key) | ||||||
|  | 				} | ||||||
|  | 			default: | ||||||
|  | 				out := opts.FormatDiff(r.Value, ptrs) | ||||||
|  | 				list = append(list, textRecord{Key: formatKey(r.Key), Value: out}) | ||||||
|  | 				keys = append(keys, r.Key) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		recs = recs[ds.NumDiff():] | ||||||
|  | 		numDiffs += ds.NumDiff() | ||||||
|  | 	} | ||||||
|  | 	if maxGroup.IsZero() { | ||||||
|  | 		assert(len(recs) == 0) | ||||||
|  | 	} else { | ||||||
|  | 		list.AppendEllipsis(maxGroup) | ||||||
|  | 		for len(keys) < len(list) { | ||||||
|  | 			keys = append(keys, reflect.Value{}) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	assert(len(list) == len(keys)) | ||||||
|  | 
 | ||||||
|  | 	// For maps, the default formatting logic uses fmt.Stringer which may | ||||||
|  | 	// produce ambiguous output. Avoid calling String to disambiguate. | ||||||
|  | 	if k == reflect.Map { | ||||||
|  | 		var ambiguous bool | ||||||
|  | 		seenKeys := map[string]reflect.Value{} | ||||||
|  | 		for i, currKey := range keys { | ||||||
|  | 			if currKey.IsValid() { | ||||||
|  | 				strKey := list[i].Key | ||||||
|  | 				prevKey, seen := seenKeys[strKey] | ||||||
|  | 				if seen && prevKey.CanInterface() && currKey.CanInterface() { | ||||||
|  | 					ambiguous = prevKey.Interface() != currKey.Interface() | ||||||
|  | 					if ambiguous { | ||||||
|  | 						break | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				seenKeys[strKey] = currKey | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		if ambiguous { | ||||||
|  | 			for i, k := range keys { | ||||||
|  | 				if k.IsValid() { | ||||||
|  | 					list[i].Key = formatMapKey(k, true, ptrs) | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // coalesceAdjacentRecords coalesces the list of records into groups of | ||||||
|  | // adjacent equal, or unequal counts. | ||||||
|  | func coalesceAdjacentRecords(name string, recs []reportRecord) (groups []diffStats) { | ||||||
|  | 	var prevCase int // Arbitrary index into which case last occurred | ||||||
|  | 	lastStats := func(i int) *diffStats { | ||||||
|  | 		if prevCase != i { | ||||||
|  | 			groups = append(groups, diffStats{Name: name}) | ||||||
|  | 			prevCase = i | ||||||
|  | 		} | ||||||
|  | 		return &groups[len(groups)-1] | ||||||
|  | 	} | ||||||
|  | 	for _, r := range recs { | ||||||
|  | 		switch rv := r.Value; { | ||||||
|  | 		case rv.NumIgnored > 0 && rv.NumSame+rv.NumDiff == 0: | ||||||
|  | 			lastStats(1).NumIgnored++ | ||||||
|  | 		case rv.NumDiff == 0: | ||||||
|  | 			lastStats(1).NumIdentical++ | ||||||
|  | 		case rv.NumDiff > 0 && !rv.ValueY.IsValid(): | ||||||
|  | 			lastStats(2).NumRemoved++ | ||||||
|  | 		case rv.NumDiff > 0 && !rv.ValueX.IsValid(): | ||||||
|  | 			lastStats(2).NumInserted++ | ||||||
|  | 		default: | ||||||
|  | 			lastStats(2).NumModified++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return groups | ||||||
|  | } | ||||||
							
								
								
									
										264
									
								
								vendor/github.com/google/go-cmp/cmp/report_references.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										264
									
								
								vendor/github.com/google/go-cmp/cmp/report_references.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,264 @@ | ||||||
|  | // Copyright 2020, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strings" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/flags" | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/value" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | const ( | ||||||
|  | 	pointerDelimPrefix = "⟪" | ||||||
|  | 	pointerDelimSuffix = "⟫" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // formatPointer prints the address of the pointer. | ||||||
|  | func formatPointer(p value.Pointer, withDelims bool) string { | ||||||
|  | 	v := p.Uintptr() | ||||||
|  | 	if flags.Deterministic { | ||||||
|  | 		v = 0xdeadf00f // Only used for stable testing purposes | ||||||
|  | 	} | ||||||
|  | 	if withDelims { | ||||||
|  | 		return pointerDelimPrefix + formatHex(uint64(v)) + pointerDelimSuffix | ||||||
|  | 	} | ||||||
|  | 	return formatHex(uint64(v)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // pointerReferences is a stack of pointers visited so far. | ||||||
|  | type pointerReferences [][2]value.Pointer | ||||||
|  | 
 | ||||||
|  | func (ps *pointerReferences) PushPair(vx, vy reflect.Value, d diffMode, deref bool) (pp [2]value.Pointer) { | ||||||
|  | 	if deref && vx.IsValid() { | ||||||
|  | 		vx = vx.Addr() | ||||||
|  | 	} | ||||||
|  | 	if deref && vy.IsValid() { | ||||||
|  | 		vy = vy.Addr() | ||||||
|  | 	} | ||||||
|  | 	switch d { | ||||||
|  | 	case diffUnknown, diffIdentical: | ||||||
|  | 		pp = [2]value.Pointer{value.PointerOf(vx), value.PointerOf(vy)} | ||||||
|  | 	case diffRemoved: | ||||||
|  | 		pp = [2]value.Pointer{value.PointerOf(vx), value.Pointer{}} | ||||||
|  | 	case diffInserted: | ||||||
|  | 		pp = [2]value.Pointer{value.Pointer{}, value.PointerOf(vy)} | ||||||
|  | 	} | ||||||
|  | 	*ps = append(*ps, pp) | ||||||
|  | 	return pp | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (ps *pointerReferences) Push(v reflect.Value) (p value.Pointer, seen bool) { | ||||||
|  | 	p = value.PointerOf(v) | ||||||
|  | 	for _, pp := range *ps { | ||||||
|  | 		if p == pp[0] || p == pp[1] { | ||||||
|  | 			return p, true | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	*ps = append(*ps, [2]value.Pointer{p, p}) | ||||||
|  | 	return p, false | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (ps *pointerReferences) Pop() { | ||||||
|  | 	*ps = (*ps)[:len(*ps)-1] | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // trunkReferences is metadata for a textNode indicating that the sub-tree | ||||||
|  | // represents the value for either pointer in a pair of references. | ||||||
|  | type trunkReferences struct{ pp [2]value.Pointer } | ||||||
|  | 
 | ||||||
|  | // trunkReference is metadata for a textNode indicating that the sub-tree | ||||||
|  | // represents the value for the given pointer reference. | ||||||
|  | type trunkReference struct{ p value.Pointer } | ||||||
|  | 
 | ||||||
|  | // leafReference is metadata for a textNode indicating that the value is | ||||||
|  | // truncated as it refers to another part of the tree (i.e., a trunk). | ||||||
|  | type leafReference struct{ p value.Pointer } | ||||||
|  | 
 | ||||||
|  | func wrapTrunkReferences(pp [2]value.Pointer, s textNode) textNode { | ||||||
|  | 	switch { | ||||||
|  | 	case pp[0].IsNil(): | ||||||
|  | 		return &textWrap{Value: s, Metadata: trunkReference{pp[1]}} | ||||||
|  | 	case pp[1].IsNil(): | ||||||
|  | 		return &textWrap{Value: s, Metadata: trunkReference{pp[0]}} | ||||||
|  | 	case pp[0] == pp[1]: | ||||||
|  | 		return &textWrap{Value: s, Metadata: trunkReference{pp[0]}} | ||||||
|  | 	default: | ||||||
|  | 		return &textWrap{Value: s, Metadata: trunkReferences{pp}} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | func wrapTrunkReference(p value.Pointer, printAddress bool, s textNode) textNode { | ||||||
|  | 	var prefix string | ||||||
|  | 	if printAddress { | ||||||
|  | 		prefix = formatPointer(p, true) | ||||||
|  | 	} | ||||||
|  | 	return &textWrap{Prefix: prefix, Value: s, Metadata: trunkReference{p}} | ||||||
|  | } | ||||||
|  | func makeLeafReference(p value.Pointer, printAddress bool) textNode { | ||||||
|  | 	out := &textWrap{Prefix: "(", Value: textEllipsis, Suffix: ")"} | ||||||
|  | 	var prefix string | ||||||
|  | 	if printAddress { | ||||||
|  | 		prefix = formatPointer(p, true) | ||||||
|  | 	} | ||||||
|  | 	return &textWrap{Prefix: prefix, Value: out, Metadata: leafReference{p}} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // resolveReferences walks the textNode tree searching for any leaf reference | ||||||
|  | // metadata and resolves each against the corresponding trunk references. | ||||||
|  | // Since pointer addresses in memory are not particularly readable to the user, | ||||||
|  | // it replaces each pointer value with an arbitrary and unique reference ID. | ||||||
|  | func resolveReferences(s textNode) { | ||||||
|  | 	var walkNodes func(textNode, func(textNode)) | ||||||
|  | 	walkNodes = func(s textNode, f func(textNode)) { | ||||||
|  | 		f(s) | ||||||
|  | 		switch s := s.(type) { | ||||||
|  | 		case *textWrap: | ||||||
|  | 			walkNodes(s.Value, f) | ||||||
|  | 		case textList: | ||||||
|  | 			for _, r := range s { | ||||||
|  | 				walkNodes(r.Value, f) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Collect all trunks and leaves with reference metadata. | ||||||
|  | 	var trunks, leaves []*textWrap | ||||||
|  | 	walkNodes(s, func(s textNode) { | ||||||
|  | 		if s, ok := s.(*textWrap); ok { | ||||||
|  | 			switch s.Metadata.(type) { | ||||||
|  | 			case leafReference: | ||||||
|  | 				leaves = append(leaves, s) | ||||||
|  | 			case trunkReference, trunkReferences: | ||||||
|  | 				trunks = append(trunks, s) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	// No leaf references to resolve. | ||||||
|  | 	if len(leaves) == 0 { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Collect the set of all leaf references to resolve. | ||||||
|  | 	leafPtrs := make(map[value.Pointer]bool) | ||||||
|  | 	for _, leaf := range leaves { | ||||||
|  | 		leafPtrs[leaf.Metadata.(leafReference).p] = true | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Collect the set of trunk pointers that are always paired together. | ||||||
|  | 	// This allows us to assign a single ID to both pointers for brevity. | ||||||
|  | 	// If a pointer in a pair ever occurs by itself or as a different pair, | ||||||
|  | 	// then the pair is broken. | ||||||
|  | 	pairedTrunkPtrs := make(map[value.Pointer]value.Pointer) | ||||||
|  | 	unpair := func(p value.Pointer) { | ||||||
|  | 		if !pairedTrunkPtrs[p].IsNil() { | ||||||
|  | 			pairedTrunkPtrs[pairedTrunkPtrs[p]] = value.Pointer{} // invalidate other half | ||||||
|  | 		} | ||||||
|  | 		pairedTrunkPtrs[p] = value.Pointer{} // invalidate this half | ||||||
|  | 	} | ||||||
|  | 	for _, trunk := range trunks { | ||||||
|  | 		switch p := trunk.Metadata.(type) { | ||||||
|  | 		case trunkReference: | ||||||
|  | 			unpair(p.p) // standalone pointer cannot be part of a pair | ||||||
|  | 		case trunkReferences: | ||||||
|  | 			p0, ok0 := pairedTrunkPtrs[p.pp[0]] | ||||||
|  | 			p1, ok1 := pairedTrunkPtrs[p.pp[1]] | ||||||
|  | 			switch { | ||||||
|  | 			case !ok0 && !ok1: | ||||||
|  | 				// Register the newly seen pair. | ||||||
|  | 				pairedTrunkPtrs[p.pp[0]] = p.pp[1] | ||||||
|  | 				pairedTrunkPtrs[p.pp[1]] = p.pp[0] | ||||||
|  | 			case ok0 && ok1 && p0 == p.pp[1] && p1 == p.pp[0]: | ||||||
|  | 				// Exact pair already seen; do nothing. | ||||||
|  | 			default: | ||||||
|  | 				// Pair conflicts with some other pair; break all pairs. | ||||||
|  | 				unpair(p.pp[0]) | ||||||
|  | 				unpair(p.pp[1]) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Correlate each pointer referenced by leaves to a unique identifier, | ||||||
|  | 	// and print the IDs for each trunk that matches those pointers. | ||||||
|  | 	var nextID uint | ||||||
|  | 	ptrIDs := make(map[value.Pointer]uint) | ||||||
|  | 	newID := func() uint { | ||||||
|  | 		id := nextID | ||||||
|  | 		nextID++ | ||||||
|  | 		return id | ||||||
|  | 	} | ||||||
|  | 	for _, trunk := range trunks { | ||||||
|  | 		switch p := trunk.Metadata.(type) { | ||||||
|  | 		case trunkReference: | ||||||
|  | 			if print := leafPtrs[p.p]; print { | ||||||
|  | 				id, ok := ptrIDs[p.p] | ||||||
|  | 				if !ok { | ||||||
|  | 					id = newID() | ||||||
|  | 					ptrIDs[p.p] = id | ||||||
|  | 				} | ||||||
|  | 				trunk.Prefix = updateReferencePrefix(trunk.Prefix, formatReference(id)) | ||||||
|  | 			} | ||||||
|  | 		case trunkReferences: | ||||||
|  | 			print0 := leafPtrs[p.pp[0]] | ||||||
|  | 			print1 := leafPtrs[p.pp[1]] | ||||||
|  | 			if print0 || print1 { | ||||||
|  | 				id0, ok0 := ptrIDs[p.pp[0]] | ||||||
|  | 				id1, ok1 := ptrIDs[p.pp[1]] | ||||||
|  | 				isPair := pairedTrunkPtrs[p.pp[0]] == p.pp[1] && pairedTrunkPtrs[p.pp[1]] == p.pp[0] | ||||||
|  | 				if isPair { | ||||||
|  | 					var id uint | ||||||
|  | 					assert(ok0 == ok1) // must be seen together or not at all | ||||||
|  | 					if ok0 { | ||||||
|  | 						assert(id0 == id1) // must have the same ID | ||||||
|  | 						id = id0 | ||||||
|  | 					} else { | ||||||
|  | 						id = newID() | ||||||
|  | 						ptrIDs[p.pp[0]] = id | ||||||
|  | 						ptrIDs[p.pp[1]] = id | ||||||
|  | 					} | ||||||
|  | 					trunk.Prefix = updateReferencePrefix(trunk.Prefix, formatReference(id)) | ||||||
|  | 				} else { | ||||||
|  | 					if print0 && !ok0 { | ||||||
|  | 						id0 = newID() | ||||||
|  | 						ptrIDs[p.pp[0]] = id0 | ||||||
|  | 					} | ||||||
|  | 					if print1 && !ok1 { | ||||||
|  | 						id1 = newID() | ||||||
|  | 						ptrIDs[p.pp[1]] = id1 | ||||||
|  | 					} | ||||||
|  | 					switch { | ||||||
|  | 					case print0 && print1: | ||||||
|  | 						trunk.Prefix = updateReferencePrefix(trunk.Prefix, formatReference(id0)+","+formatReference(id1)) | ||||||
|  | 					case print0: | ||||||
|  | 						trunk.Prefix = updateReferencePrefix(trunk.Prefix, formatReference(id0)) | ||||||
|  | 					case print1: | ||||||
|  | 						trunk.Prefix = updateReferencePrefix(trunk.Prefix, formatReference(id1)) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Update all leaf references with the unique identifier. | ||||||
|  | 	for _, leaf := range leaves { | ||||||
|  | 		if id, ok := ptrIDs[leaf.Metadata.(leafReference).p]; ok { | ||||||
|  | 			leaf.Prefix = updateReferencePrefix(leaf.Prefix, formatReference(id)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func formatReference(id uint) string { | ||||||
|  | 	return fmt.Sprintf("ref#%d", id) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func updateReferencePrefix(prefix, ref string) string { | ||||||
|  | 	if prefix == "" { | ||||||
|  | 		return pointerDelimPrefix + ref + pointerDelimSuffix | ||||||
|  | 	} | ||||||
|  | 	suffix := strings.TrimPrefix(prefix, pointerDelimPrefix) | ||||||
|  | 	return pointerDelimPrefix + ref + ": " + suffix | ||||||
|  | } | ||||||
							
								
								
									
										414
									
								
								vendor/github.com/google/go-cmp/cmp/report_reflect.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										414
									
								
								vendor/github.com/google/go-cmp/cmp/report_reflect.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,414 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  | 	"unicode" | ||||||
|  | 	"unicode/utf8" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/value" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var ( | ||||||
|  | 	anyType    = reflect.TypeOf((*interface{})(nil)).Elem() | ||||||
|  | 	stringType = reflect.TypeOf((*string)(nil)).Elem() | ||||||
|  | 	bytesType  = reflect.TypeOf((*[]byte)(nil)).Elem() | ||||||
|  | 	byteType   = reflect.TypeOf((*byte)(nil)).Elem() | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | type formatValueOptions struct { | ||||||
|  | 	// AvoidStringer controls whether to avoid calling custom stringer | ||||||
|  | 	// methods like error.Error or fmt.Stringer.String. | ||||||
|  | 	AvoidStringer bool | ||||||
|  | 
 | ||||||
|  | 	// PrintAddresses controls whether to print the address of all pointers, | ||||||
|  | 	// slice elements, and maps. | ||||||
|  | 	PrintAddresses bool | ||||||
|  | 
 | ||||||
|  | 	// QualifiedNames controls whether FormatType uses the fully qualified name | ||||||
|  | 	// (including the full package path as opposed to just the package name). | ||||||
|  | 	QualifiedNames bool | ||||||
|  | 
 | ||||||
|  | 	// VerbosityLevel controls the amount of output to produce. | ||||||
|  | 	// A higher value produces more output. A value of zero or lower produces | ||||||
|  | 	// no output (represented using an ellipsis). | ||||||
|  | 	// If LimitVerbosity is false, then the level is treated as infinite. | ||||||
|  | 	VerbosityLevel int | ||||||
|  | 
 | ||||||
|  | 	// LimitVerbosity specifies that formatting should respect VerbosityLevel. | ||||||
|  | 	LimitVerbosity bool | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FormatType prints the type as if it were wrapping s. | ||||||
|  | // This may return s as-is depending on the current type and TypeMode mode. | ||||||
|  | func (opts formatOptions) FormatType(t reflect.Type, s textNode) textNode { | ||||||
|  | 	// Check whether to emit the type or not. | ||||||
|  | 	switch opts.TypeMode { | ||||||
|  | 	case autoType: | ||||||
|  | 		switch t.Kind() { | ||||||
|  | 		case reflect.Struct, reflect.Slice, reflect.Array, reflect.Map: | ||||||
|  | 			if s.Equal(textNil) { | ||||||
|  | 				return s | ||||||
|  | 			} | ||||||
|  | 		default: | ||||||
|  | 			return s | ||||||
|  | 		} | ||||||
|  | 		if opts.DiffMode == diffIdentical { | ||||||
|  | 			return s // elide type for identical nodes | ||||||
|  | 		} | ||||||
|  | 	case elideType: | ||||||
|  | 		return s | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Determine the type label, applying special handling for unnamed types. | ||||||
|  | 	typeName := value.TypeString(t, opts.QualifiedNames) | ||||||
|  | 	if t.Name() == "" { | ||||||
|  | 		// According to Go grammar, certain type literals contain symbols that | ||||||
|  | 		// do not strongly bind to the next lexicographical token (e.g., *T). | ||||||
|  | 		switch t.Kind() { | ||||||
|  | 		case reflect.Chan, reflect.Func, reflect.Ptr: | ||||||
|  | 			typeName = "(" + typeName + ")" | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return &textWrap{Prefix: typeName, Value: wrapParens(s)} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // wrapParens wraps s with a set of parenthesis, but avoids it if the | ||||||
|  | // wrapped node itself is already surrounded by a pair of parenthesis or braces. | ||||||
|  | // It handles unwrapping one level of pointer-reference nodes. | ||||||
|  | func wrapParens(s textNode) textNode { | ||||||
|  | 	var refNode *textWrap | ||||||
|  | 	if s2, ok := s.(*textWrap); ok { | ||||||
|  | 		// Unwrap a single pointer reference node. | ||||||
|  | 		switch s2.Metadata.(type) { | ||||||
|  | 		case leafReference, trunkReference, trunkReferences: | ||||||
|  | 			refNode = s2 | ||||||
|  | 			if s3, ok := refNode.Value.(*textWrap); ok { | ||||||
|  | 				s2 = s3 | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Already has delimiters that make parenthesis unnecessary. | ||||||
|  | 		hasParens := strings.HasPrefix(s2.Prefix, "(") && strings.HasSuffix(s2.Suffix, ")") | ||||||
|  | 		hasBraces := strings.HasPrefix(s2.Prefix, "{") && strings.HasSuffix(s2.Suffix, "}") | ||||||
|  | 		if hasParens || hasBraces { | ||||||
|  | 			return s | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if refNode != nil { | ||||||
|  | 		refNode.Value = &textWrap{Prefix: "(", Value: refNode.Value, Suffix: ")"} | ||||||
|  | 		return s | ||||||
|  | 	} | ||||||
|  | 	return &textWrap{Prefix: "(", Value: s, Suffix: ")"} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FormatValue prints the reflect.Value, taking extra care to avoid descending | ||||||
|  | // into pointers already in ptrs. As pointers are visited, ptrs is also updated. | ||||||
|  | func (opts formatOptions) FormatValue(v reflect.Value, parentKind reflect.Kind, ptrs *pointerReferences) (out textNode) { | ||||||
|  | 	if !v.IsValid() { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	t := v.Type() | ||||||
|  | 
 | ||||||
|  | 	// Check slice element for cycles. | ||||||
|  | 	if parentKind == reflect.Slice { | ||||||
|  | 		ptrRef, visited := ptrs.Push(v.Addr()) | ||||||
|  | 		if visited { | ||||||
|  | 			return makeLeafReference(ptrRef, false) | ||||||
|  | 		} | ||||||
|  | 		defer ptrs.Pop() | ||||||
|  | 		defer func() { out = wrapTrunkReference(ptrRef, false, out) }() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Check whether there is an Error or String method to call. | ||||||
|  | 	if !opts.AvoidStringer && v.CanInterface() { | ||||||
|  | 		// Avoid calling Error or String methods on nil receivers since many | ||||||
|  | 		// implementations crash when doing so. | ||||||
|  | 		if (t.Kind() != reflect.Ptr && t.Kind() != reflect.Interface) || !v.IsNil() { | ||||||
|  | 			var prefix, strVal string | ||||||
|  | 			func() { | ||||||
|  | 				// Swallow and ignore any panics from String or Error. | ||||||
|  | 				defer func() { recover() }() | ||||||
|  | 				switch v := v.Interface().(type) { | ||||||
|  | 				case error: | ||||||
|  | 					strVal = v.Error() | ||||||
|  | 					prefix = "e" | ||||||
|  | 				case fmt.Stringer: | ||||||
|  | 					strVal = v.String() | ||||||
|  | 					prefix = "s" | ||||||
|  | 				} | ||||||
|  | 			}() | ||||||
|  | 			if prefix != "" { | ||||||
|  | 				return opts.formatString(prefix, strVal) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Check whether to explicitly wrap the result with the type. | ||||||
|  | 	var skipType bool | ||||||
|  | 	defer func() { | ||||||
|  | 		if !skipType { | ||||||
|  | 			out = opts.FormatType(t, out) | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  | 
 | ||||||
|  | 	switch t.Kind() { | ||||||
|  | 	case reflect.Bool: | ||||||
|  | 		return textLine(fmt.Sprint(v.Bool())) | ||||||
|  | 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: | ||||||
|  | 		return textLine(fmt.Sprint(v.Int())) | ||||||
|  | 	case reflect.Uint, reflect.Uint16, reflect.Uint32, reflect.Uint64: | ||||||
|  | 		return textLine(fmt.Sprint(v.Uint())) | ||||||
|  | 	case reflect.Uint8: | ||||||
|  | 		if parentKind == reflect.Slice || parentKind == reflect.Array { | ||||||
|  | 			return textLine(formatHex(v.Uint())) | ||||||
|  | 		} | ||||||
|  | 		return textLine(fmt.Sprint(v.Uint())) | ||||||
|  | 	case reflect.Uintptr: | ||||||
|  | 		return textLine(formatHex(v.Uint())) | ||||||
|  | 	case reflect.Float32, reflect.Float64: | ||||||
|  | 		return textLine(fmt.Sprint(v.Float())) | ||||||
|  | 	case reflect.Complex64, reflect.Complex128: | ||||||
|  | 		return textLine(fmt.Sprint(v.Complex())) | ||||||
|  | 	case reflect.String: | ||||||
|  | 		return opts.formatString("", v.String()) | ||||||
|  | 	case reflect.UnsafePointer, reflect.Chan, reflect.Func: | ||||||
|  | 		return textLine(formatPointer(value.PointerOf(v), true)) | ||||||
|  | 	case reflect.Struct: | ||||||
|  | 		var list textList | ||||||
|  | 		v := makeAddressable(v) // needed for retrieveUnexportedField | ||||||
|  | 		maxLen := v.NumField() | ||||||
|  | 		if opts.LimitVerbosity { | ||||||
|  | 			maxLen = ((1 << opts.verbosity()) >> 1) << 2 // 0, 4, 8, 16, 32, etc... | ||||||
|  | 			opts.VerbosityLevel-- | ||||||
|  | 		} | ||||||
|  | 		for i := 0; i < v.NumField(); i++ { | ||||||
|  | 			vv := v.Field(i) | ||||||
|  | 			if vv.IsZero() { | ||||||
|  | 				continue // Elide fields with zero values | ||||||
|  | 			} | ||||||
|  | 			if len(list) == maxLen { | ||||||
|  | 				list.AppendEllipsis(diffStats{}) | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 			sf := t.Field(i) | ||||||
|  | 			if !isExported(sf.Name) { | ||||||
|  | 				vv = retrieveUnexportedField(v, sf, true) | ||||||
|  | 			} | ||||||
|  | 			s := opts.WithTypeMode(autoType).FormatValue(vv, t.Kind(), ptrs) | ||||||
|  | 			list = append(list, textRecord{Key: sf.Name, Value: s}) | ||||||
|  | 		} | ||||||
|  | 		return &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | 	case reflect.Slice: | ||||||
|  | 		if v.IsNil() { | ||||||
|  | 			return textNil | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check whether this is a []byte of text data. | ||||||
|  | 		if t.Elem() == byteType { | ||||||
|  | 			b := v.Bytes() | ||||||
|  | 			isPrintSpace := func(r rune) bool { return unicode.IsPrint(r) || unicode.IsSpace(r) } | ||||||
|  | 			if len(b) > 0 && utf8.Valid(b) && len(bytes.TrimFunc(b, isPrintSpace)) == 0 { | ||||||
|  | 				out = opts.formatString("", string(b)) | ||||||
|  | 				skipType = true | ||||||
|  | 				return opts.FormatType(t, out) | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		fallthrough | ||||||
|  | 	case reflect.Array: | ||||||
|  | 		maxLen := v.Len() | ||||||
|  | 		if opts.LimitVerbosity { | ||||||
|  | 			maxLen = ((1 << opts.verbosity()) >> 1) << 2 // 0, 4, 8, 16, 32, etc... | ||||||
|  | 			opts.VerbosityLevel-- | ||||||
|  | 		} | ||||||
|  | 		var list textList | ||||||
|  | 		for i := 0; i < v.Len(); i++ { | ||||||
|  | 			if len(list) == maxLen { | ||||||
|  | 				list.AppendEllipsis(diffStats{}) | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 			s := opts.WithTypeMode(elideType).FormatValue(v.Index(i), t.Kind(), ptrs) | ||||||
|  | 			list = append(list, textRecord{Value: s}) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		out = &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | 		if t.Kind() == reflect.Slice && opts.PrintAddresses { | ||||||
|  | 			header := fmt.Sprintf("ptr:%v, len:%d, cap:%d", formatPointer(value.PointerOf(v), false), v.Len(), v.Cap()) | ||||||
|  | 			out = &textWrap{Prefix: pointerDelimPrefix + header + pointerDelimSuffix, Value: out} | ||||||
|  | 		} | ||||||
|  | 		return out | ||||||
|  | 	case reflect.Map: | ||||||
|  | 		if v.IsNil() { | ||||||
|  | 			return textNil | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check pointer for cycles. | ||||||
|  | 		ptrRef, visited := ptrs.Push(v) | ||||||
|  | 		if visited { | ||||||
|  | 			return makeLeafReference(ptrRef, opts.PrintAddresses) | ||||||
|  | 		} | ||||||
|  | 		defer ptrs.Pop() | ||||||
|  | 
 | ||||||
|  | 		maxLen := v.Len() | ||||||
|  | 		if opts.LimitVerbosity { | ||||||
|  | 			maxLen = ((1 << opts.verbosity()) >> 1) << 2 // 0, 4, 8, 16, 32, etc... | ||||||
|  | 			opts.VerbosityLevel-- | ||||||
|  | 		} | ||||||
|  | 		var list textList | ||||||
|  | 		for _, k := range value.SortKeys(v.MapKeys()) { | ||||||
|  | 			if len(list) == maxLen { | ||||||
|  | 				list.AppendEllipsis(diffStats{}) | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 			sk := formatMapKey(k, false, ptrs) | ||||||
|  | 			sv := opts.WithTypeMode(elideType).FormatValue(v.MapIndex(k), t.Kind(), ptrs) | ||||||
|  | 			list = append(list, textRecord{Key: sk, Value: sv}) | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		out = &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | 		out = wrapTrunkReference(ptrRef, opts.PrintAddresses, out) | ||||||
|  | 		return out | ||||||
|  | 	case reflect.Ptr: | ||||||
|  | 		if v.IsNil() { | ||||||
|  | 			return textNil | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Check pointer for cycles. | ||||||
|  | 		ptrRef, visited := ptrs.Push(v) | ||||||
|  | 		if visited { | ||||||
|  | 			out = makeLeafReference(ptrRef, opts.PrintAddresses) | ||||||
|  | 			return &textWrap{Prefix: "&", Value: out} | ||||||
|  | 		} | ||||||
|  | 		defer ptrs.Pop() | ||||||
|  | 
 | ||||||
|  | 		// Skip the name only if this is an unnamed pointer type. | ||||||
|  | 		// Otherwise taking the address of a value does not reproduce | ||||||
|  | 		// the named pointer type. | ||||||
|  | 		if v.Type().Name() == "" { | ||||||
|  | 			skipType = true // Let the underlying value print the type instead | ||||||
|  | 		} | ||||||
|  | 		out = opts.FormatValue(v.Elem(), t.Kind(), ptrs) | ||||||
|  | 		out = wrapTrunkReference(ptrRef, opts.PrintAddresses, out) | ||||||
|  | 		out = &textWrap{Prefix: "&", Value: out} | ||||||
|  | 		return out | ||||||
|  | 	case reflect.Interface: | ||||||
|  | 		if v.IsNil() { | ||||||
|  | 			return textNil | ||||||
|  | 		} | ||||||
|  | 		// Interfaces accept different concrete types, | ||||||
|  | 		// so configure the underlying value to explicitly print the type. | ||||||
|  | 		return opts.WithTypeMode(emitType).FormatValue(v.Elem(), t.Kind(), ptrs) | ||||||
|  | 	default: | ||||||
|  | 		panic(fmt.Sprintf("%v kind not handled", v.Kind())) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts formatOptions) formatString(prefix, s string) textNode { | ||||||
|  | 	maxLen := len(s) | ||||||
|  | 	maxLines := strings.Count(s, "\n") + 1 | ||||||
|  | 	if opts.LimitVerbosity { | ||||||
|  | 		maxLen = (1 << opts.verbosity()) << 5   // 32, 64, 128, 256, etc... | ||||||
|  | 		maxLines = (1 << opts.verbosity()) << 2 //  4, 8, 16, 32, 64, etc... | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// For multiline strings, use the triple-quote syntax, | ||||||
|  | 	// but only use it when printing removed or inserted nodes since | ||||||
|  | 	// we only want the extra verbosity for those cases. | ||||||
|  | 	lines := strings.Split(strings.TrimSuffix(s, "\n"), "\n") | ||||||
|  | 	isTripleQuoted := len(lines) >= 4 && (opts.DiffMode == '-' || opts.DiffMode == '+') | ||||||
|  | 	for i := 0; i < len(lines) && isTripleQuoted; i++ { | ||||||
|  | 		lines[i] = strings.TrimPrefix(strings.TrimSuffix(lines[i], "\r"), "\r") // trim leading/trailing carriage returns for legacy Windows endline support | ||||||
|  | 		isPrintable := func(r rune) bool { | ||||||
|  | 			return unicode.IsPrint(r) || r == '\t' // specially treat tab as printable | ||||||
|  | 		} | ||||||
|  | 		line := lines[i] | ||||||
|  | 		isTripleQuoted = !strings.HasPrefix(strings.TrimPrefix(line, prefix), `"""`) && !strings.HasPrefix(line, "...") && strings.TrimFunc(line, isPrintable) == "" && len(line) <= maxLen | ||||||
|  | 	} | ||||||
|  | 	if isTripleQuoted { | ||||||
|  | 		var list textList | ||||||
|  | 		list = append(list, textRecord{Diff: opts.DiffMode, Value: textLine(prefix + `"""`), ElideComma: true}) | ||||||
|  | 		for i, line := range lines { | ||||||
|  | 			if numElided := len(lines) - i; i == maxLines-1 && numElided > 1 { | ||||||
|  | 				comment := commentString(fmt.Sprintf("%d elided lines", numElided)) | ||||||
|  | 				list = append(list, textRecord{Diff: opts.DiffMode, Value: textEllipsis, ElideComma: true, Comment: comment}) | ||||||
|  | 				break | ||||||
|  | 			} | ||||||
|  | 			list = append(list, textRecord{Diff: opts.DiffMode, Value: textLine(line), ElideComma: true}) | ||||||
|  | 		} | ||||||
|  | 		list = append(list, textRecord{Diff: opts.DiffMode, Value: textLine(prefix + `"""`), ElideComma: true}) | ||||||
|  | 		return &textWrap{Prefix: "(", Value: list, Suffix: ")"} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Format the string as a single-line quoted string. | ||||||
|  | 	if len(s) > maxLen+len(textEllipsis) { | ||||||
|  | 		return textLine(prefix + formatString(s[:maxLen]) + string(textEllipsis)) | ||||||
|  | 	} | ||||||
|  | 	return textLine(prefix + formatString(s)) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // formatMapKey formats v as if it were a map key. | ||||||
|  | // The result is guaranteed to be a single line. | ||||||
|  | func formatMapKey(v reflect.Value, disambiguate bool, ptrs *pointerReferences) string { | ||||||
|  | 	var opts formatOptions | ||||||
|  | 	opts.DiffMode = diffIdentical | ||||||
|  | 	opts.TypeMode = elideType | ||||||
|  | 	opts.PrintAddresses = disambiguate | ||||||
|  | 	opts.AvoidStringer = disambiguate | ||||||
|  | 	opts.QualifiedNames = disambiguate | ||||||
|  | 	opts.VerbosityLevel = maxVerbosityPreset | ||||||
|  | 	opts.LimitVerbosity = true | ||||||
|  | 	s := opts.FormatValue(v, reflect.Map, ptrs).String() | ||||||
|  | 	return strings.TrimSpace(s) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // formatString prints s as a double-quoted or backtick-quoted string. | ||||||
|  | func formatString(s string) string { | ||||||
|  | 	// Use quoted string if it the same length as a raw string literal. | ||||||
|  | 	// Otherwise, attempt to use the raw string form. | ||||||
|  | 	qs := strconv.Quote(s) | ||||||
|  | 	if len(qs) == 1+len(s)+1 { | ||||||
|  | 		return qs | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Disallow newlines to ensure output is a single line. | ||||||
|  | 	// Only allow printable runes for readability purposes. | ||||||
|  | 	rawInvalid := func(r rune) bool { | ||||||
|  | 		return r == '`' || r == '\n' || !(unicode.IsPrint(r) || r == '\t') | ||||||
|  | 	} | ||||||
|  | 	if utf8.ValidString(s) && strings.IndexFunc(s, rawInvalid) < 0 { | ||||||
|  | 		return "`" + s + "`" | ||||||
|  | 	} | ||||||
|  | 	return qs | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // formatHex prints u as a hexadecimal integer in Go notation. | ||||||
|  | func formatHex(u uint64) string { | ||||||
|  | 	var f string | ||||||
|  | 	switch { | ||||||
|  | 	case u <= 0xff: | ||||||
|  | 		f = "0x%02x" | ||||||
|  | 	case u <= 0xffff: | ||||||
|  | 		f = "0x%04x" | ||||||
|  | 	case u <= 0xffffff: | ||||||
|  | 		f = "0x%06x" | ||||||
|  | 	case u <= 0xffffffff: | ||||||
|  | 		f = "0x%08x" | ||||||
|  | 	case u <= 0xffffffffff: | ||||||
|  | 		f = "0x%010x" | ||||||
|  | 	case u <= 0xffffffffffff: | ||||||
|  | 		f = "0x%012x" | ||||||
|  | 	case u <= 0xffffffffffffff: | ||||||
|  | 		f = "0x%014x" | ||||||
|  | 	case u <= 0xffffffffffffffff: | ||||||
|  | 		f = "0x%016x" | ||||||
|  | 	} | ||||||
|  | 	return fmt.Sprintf(f, u) | ||||||
|  | } | ||||||
							
								
								
									
										614
									
								
								vendor/github.com/google/go-cmp/cmp/report_slices.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										614
									
								
								vendor/github.com/google/go-cmp/cmp/report_slices.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,614 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"math" | ||||||
|  | 	"reflect" | ||||||
|  | 	"strconv" | ||||||
|  | 	"strings" | ||||||
|  | 	"unicode" | ||||||
|  | 	"unicode/utf8" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/diff" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // CanFormatDiffSlice reports whether we support custom formatting for nodes | ||||||
|  | // that are slices of primitive kinds or strings. | ||||||
|  | func (opts formatOptions) CanFormatDiffSlice(v *valueNode) bool { | ||||||
|  | 	switch { | ||||||
|  | 	case opts.DiffMode != diffUnknown: | ||||||
|  | 		return false // Must be formatting in diff mode | ||||||
|  | 	case v.NumDiff == 0: | ||||||
|  | 		return false // No differences detected | ||||||
|  | 	case !v.ValueX.IsValid() || !v.ValueY.IsValid(): | ||||||
|  | 		return false // Both values must be valid | ||||||
|  | 	case v.NumIgnored > 0: | ||||||
|  | 		return false // Some ignore option was used | ||||||
|  | 	case v.NumTransformed > 0: | ||||||
|  | 		return false // Some transform option was used | ||||||
|  | 	case v.NumCompared > 1: | ||||||
|  | 		return false // More than one comparison was used | ||||||
|  | 	case v.NumCompared == 1 && v.Type.Name() != "": | ||||||
|  | 		// The need for cmp to check applicability of options on every element | ||||||
|  | 		// in a slice is a significant performance detriment for large []byte. | ||||||
|  | 		// The workaround is to specify Comparer(bytes.Equal), | ||||||
|  | 		// which enables cmp to compare []byte more efficiently. | ||||||
|  | 		// If they differ, we still want to provide batched diffing. | ||||||
|  | 		// The logic disallows named types since they tend to have their own | ||||||
|  | 		// String method, with nicer formatting than what this provides. | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Check whether this is an interface with the same concrete types. | ||||||
|  | 	t := v.Type | ||||||
|  | 	vx, vy := v.ValueX, v.ValueY | ||||||
|  | 	if t.Kind() == reflect.Interface && !vx.IsNil() && !vy.IsNil() && vx.Elem().Type() == vy.Elem().Type() { | ||||||
|  | 		vx, vy = vx.Elem(), vy.Elem() | ||||||
|  | 		t = vx.Type() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Check whether we provide specialized diffing for this type. | ||||||
|  | 	switch t.Kind() { | ||||||
|  | 	case reflect.String: | ||||||
|  | 	case reflect.Array, reflect.Slice: | ||||||
|  | 		// Only slices of primitive types have specialized handling. | ||||||
|  | 		switch t.Elem().Kind() { | ||||||
|  | 		case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, | ||||||
|  | 			reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr, | ||||||
|  | 			reflect.Bool, reflect.Float32, reflect.Float64, reflect.Complex64, reflect.Complex128: | ||||||
|  | 		default: | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Both slice values have to be non-empty. | ||||||
|  | 		if t.Kind() == reflect.Slice && (vx.Len() == 0 || vy.Len() == 0) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// If a sufficient number of elements already differ, | ||||||
|  | 		// use specialized formatting even if length requirement is not met. | ||||||
|  | 		if v.NumDiff > v.NumSame { | ||||||
|  | 			return true | ||||||
|  | 		} | ||||||
|  | 	default: | ||||||
|  | 		return false | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Use specialized string diffing for longer slices or strings. | ||||||
|  | 	const minLength = 32 | ||||||
|  | 	return vx.Len() >= minLength && vy.Len() >= minLength | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // FormatDiffSlice prints a diff for the slices (or strings) represented by v. | ||||||
|  | // This provides custom-tailored logic to make printing of differences in | ||||||
|  | // textual strings and slices of primitive kinds more readable. | ||||||
|  | func (opts formatOptions) FormatDiffSlice(v *valueNode) textNode { | ||||||
|  | 	assert(opts.DiffMode == diffUnknown) | ||||||
|  | 	t, vx, vy := v.Type, v.ValueX, v.ValueY | ||||||
|  | 	if t.Kind() == reflect.Interface { | ||||||
|  | 		vx, vy = vx.Elem(), vy.Elem() | ||||||
|  | 		t = vx.Type() | ||||||
|  | 		opts = opts.WithTypeMode(emitType) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Auto-detect the type of the data. | ||||||
|  | 	var sx, sy string | ||||||
|  | 	var ssx, ssy []string | ||||||
|  | 	var isString, isMostlyText, isPureLinedText, isBinary bool | ||||||
|  | 	switch { | ||||||
|  | 	case t.Kind() == reflect.String: | ||||||
|  | 		sx, sy = vx.String(), vy.String() | ||||||
|  | 		isString = true | ||||||
|  | 	case t.Kind() == reflect.Slice && t.Elem() == byteType: | ||||||
|  | 		sx, sy = string(vx.Bytes()), string(vy.Bytes()) | ||||||
|  | 		isString = true | ||||||
|  | 	case t.Kind() == reflect.Array: | ||||||
|  | 		// Arrays need to be addressable for slice operations to work. | ||||||
|  | 		vx2, vy2 := reflect.New(t).Elem(), reflect.New(t).Elem() | ||||||
|  | 		vx2.Set(vx) | ||||||
|  | 		vy2.Set(vy) | ||||||
|  | 		vx, vy = vx2, vy2 | ||||||
|  | 	} | ||||||
|  | 	if isString { | ||||||
|  | 		var numTotalRunes, numValidRunes, numLines, lastLineIdx, maxLineLen int | ||||||
|  | 		for i, r := range sx + sy { | ||||||
|  | 			numTotalRunes++ | ||||||
|  | 			if (unicode.IsPrint(r) || unicode.IsSpace(r)) && r != utf8.RuneError { | ||||||
|  | 				numValidRunes++ | ||||||
|  | 			} | ||||||
|  | 			if r == '\n' { | ||||||
|  | 				if maxLineLen < i-lastLineIdx { | ||||||
|  | 					maxLineLen = i - lastLineIdx | ||||||
|  | 				} | ||||||
|  | 				lastLineIdx = i + 1 | ||||||
|  | 				numLines++ | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		isPureText := numValidRunes == numTotalRunes | ||||||
|  | 		isMostlyText = float64(numValidRunes) > math.Floor(0.90*float64(numTotalRunes)) | ||||||
|  | 		isPureLinedText = isPureText && numLines >= 4 && maxLineLen <= 1024 | ||||||
|  | 		isBinary = !isMostlyText | ||||||
|  | 
 | ||||||
|  | 		// Avoid diffing by lines if it produces a significantly more complex | ||||||
|  | 		// edit script than diffing by bytes. | ||||||
|  | 		if isPureLinedText { | ||||||
|  | 			ssx = strings.Split(sx, "\n") | ||||||
|  | 			ssy = strings.Split(sy, "\n") | ||||||
|  | 			esLines := diff.Difference(len(ssx), len(ssy), func(ix, iy int) diff.Result { | ||||||
|  | 				return diff.BoolResult(ssx[ix] == ssy[iy]) | ||||||
|  | 			}) | ||||||
|  | 			esBytes := diff.Difference(len(sx), len(sy), func(ix, iy int) diff.Result { | ||||||
|  | 				return diff.BoolResult(sx[ix] == sy[iy]) | ||||||
|  | 			}) | ||||||
|  | 			efficiencyLines := float64(esLines.Dist()) / float64(len(esLines)) | ||||||
|  | 			efficiencyBytes := float64(esBytes.Dist()) / float64(len(esBytes)) | ||||||
|  | 			quotedLength := len(strconv.Quote(sx + sy)) | ||||||
|  | 			unquotedLength := len(sx) + len(sy) | ||||||
|  | 			escapeExpansionRatio := float64(quotedLength) / float64(unquotedLength) | ||||||
|  | 			isPureLinedText = efficiencyLines < 4*efficiencyBytes || escapeExpansionRatio > 1.1 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Format the string into printable records. | ||||||
|  | 	var list textList | ||||||
|  | 	var delim string | ||||||
|  | 	switch { | ||||||
|  | 	// If the text appears to be multi-lined text, | ||||||
|  | 	// then perform differencing across individual lines. | ||||||
|  | 	case isPureLinedText: | ||||||
|  | 		list = opts.formatDiffSlice( | ||||||
|  | 			reflect.ValueOf(ssx), reflect.ValueOf(ssy), 1, "line", | ||||||
|  | 			func(v reflect.Value, d diffMode) textRecord { | ||||||
|  | 				s := formatString(v.Index(0).String()) | ||||||
|  | 				return textRecord{Diff: d, Value: textLine(s)} | ||||||
|  | 			}, | ||||||
|  | 		) | ||||||
|  | 		delim = "\n" | ||||||
|  | 
 | ||||||
|  | 		// If possible, use a custom triple-quote (""") syntax for printing | ||||||
|  | 		// differences in a string literal. This format is more readable, | ||||||
|  | 		// but has edge-cases where differences are visually indistinguishable. | ||||||
|  | 		// This format is avoided under the following conditions: | ||||||
|  | 		//   - A line starts with `"""` | ||||||
|  | 		//   - A line starts with "..." | ||||||
|  | 		//   - A line contains non-printable characters | ||||||
|  | 		//   - Adjacent different lines differ only by whitespace | ||||||
|  | 		// | ||||||
|  | 		// For example: | ||||||
|  | 		// | ||||||
|  | 		//		""" | ||||||
|  | 		//		... // 3 identical lines | ||||||
|  | 		//		foo | ||||||
|  | 		//		bar | ||||||
|  | 		//	-	baz | ||||||
|  | 		//	+	BAZ | ||||||
|  | 		//		""" | ||||||
|  | 		isTripleQuoted := true | ||||||
|  | 		prevRemoveLines := map[string]bool{} | ||||||
|  | 		prevInsertLines := map[string]bool{} | ||||||
|  | 		var list2 textList | ||||||
|  | 		list2 = append(list2, textRecord{Value: textLine(`"""`), ElideComma: true}) | ||||||
|  | 		for _, r := range list { | ||||||
|  | 			if !r.Value.Equal(textEllipsis) { | ||||||
|  | 				line, _ := strconv.Unquote(string(r.Value.(textLine))) | ||||||
|  | 				line = strings.TrimPrefix(strings.TrimSuffix(line, "\r"), "\r") // trim leading/trailing carriage returns for legacy Windows endline support | ||||||
|  | 				normLine := strings.Map(func(r rune) rune { | ||||||
|  | 					if unicode.IsSpace(r) { | ||||||
|  | 						return -1 // drop whitespace to avoid visually indistinguishable output | ||||||
|  | 					} | ||||||
|  | 					return r | ||||||
|  | 				}, line) | ||||||
|  | 				isPrintable := func(r rune) bool { | ||||||
|  | 					return unicode.IsPrint(r) || r == '\t' // specially treat tab as printable | ||||||
|  | 				} | ||||||
|  | 				isTripleQuoted = !strings.HasPrefix(line, `"""`) && !strings.HasPrefix(line, "...") && strings.TrimFunc(line, isPrintable) == "" | ||||||
|  | 				switch r.Diff { | ||||||
|  | 				case diffRemoved: | ||||||
|  | 					isTripleQuoted = isTripleQuoted && !prevInsertLines[normLine] | ||||||
|  | 					prevRemoveLines[normLine] = true | ||||||
|  | 				case diffInserted: | ||||||
|  | 					isTripleQuoted = isTripleQuoted && !prevRemoveLines[normLine] | ||||||
|  | 					prevInsertLines[normLine] = true | ||||||
|  | 				} | ||||||
|  | 				if !isTripleQuoted { | ||||||
|  | 					break | ||||||
|  | 				} | ||||||
|  | 				r.Value = textLine(line) | ||||||
|  | 				r.ElideComma = true | ||||||
|  | 			} | ||||||
|  | 			if !(r.Diff == diffRemoved || r.Diff == diffInserted) { // start a new non-adjacent difference group | ||||||
|  | 				prevRemoveLines = map[string]bool{} | ||||||
|  | 				prevInsertLines = map[string]bool{} | ||||||
|  | 			} | ||||||
|  | 			list2 = append(list2, r) | ||||||
|  | 		} | ||||||
|  | 		if r := list2[len(list2)-1]; r.Diff == diffIdentical && len(r.Value.(textLine)) == 0 { | ||||||
|  | 			list2 = list2[:len(list2)-1] // elide single empty line at the end | ||||||
|  | 		} | ||||||
|  | 		list2 = append(list2, textRecord{Value: textLine(`"""`), ElideComma: true}) | ||||||
|  | 		if isTripleQuoted { | ||||||
|  | 			var out textNode = &textWrap{Prefix: "(", Value: list2, Suffix: ")"} | ||||||
|  | 			switch t.Kind() { | ||||||
|  | 			case reflect.String: | ||||||
|  | 				if t != stringType { | ||||||
|  | 					out = opts.FormatType(t, out) | ||||||
|  | 				} | ||||||
|  | 			case reflect.Slice: | ||||||
|  | 				// Always emit type for slices since the triple-quote syntax | ||||||
|  | 				// looks like a string (not a slice). | ||||||
|  | 				opts = opts.WithTypeMode(emitType) | ||||||
|  | 				out = opts.FormatType(t, out) | ||||||
|  | 			} | ||||||
|  | 			return out | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 	// If the text appears to be single-lined text, | ||||||
|  | 	// then perform differencing in approximately fixed-sized chunks. | ||||||
|  | 	// The output is printed as quoted strings. | ||||||
|  | 	case isMostlyText: | ||||||
|  | 		list = opts.formatDiffSlice( | ||||||
|  | 			reflect.ValueOf(sx), reflect.ValueOf(sy), 64, "byte", | ||||||
|  | 			func(v reflect.Value, d diffMode) textRecord { | ||||||
|  | 				s := formatString(v.String()) | ||||||
|  | 				return textRecord{Diff: d, Value: textLine(s)} | ||||||
|  | 			}, | ||||||
|  | 		) | ||||||
|  | 
 | ||||||
|  | 	// If the text appears to be binary data, | ||||||
|  | 	// then perform differencing in approximately fixed-sized chunks. | ||||||
|  | 	// The output is inspired by hexdump. | ||||||
|  | 	case isBinary: | ||||||
|  | 		list = opts.formatDiffSlice( | ||||||
|  | 			reflect.ValueOf(sx), reflect.ValueOf(sy), 16, "byte", | ||||||
|  | 			func(v reflect.Value, d diffMode) textRecord { | ||||||
|  | 				var ss []string | ||||||
|  | 				for i := 0; i < v.Len(); i++ { | ||||||
|  | 					ss = append(ss, formatHex(v.Index(i).Uint())) | ||||||
|  | 				} | ||||||
|  | 				s := strings.Join(ss, ", ") | ||||||
|  | 				comment := commentString(fmt.Sprintf("%c|%v|", d, formatASCII(v.String()))) | ||||||
|  | 				return textRecord{Diff: d, Value: textLine(s), Comment: comment} | ||||||
|  | 			}, | ||||||
|  | 		) | ||||||
|  | 
 | ||||||
|  | 	// For all other slices of primitive types, | ||||||
|  | 	// then perform differencing in approximately fixed-sized chunks. | ||||||
|  | 	// The size of each chunk depends on the width of the element kind. | ||||||
|  | 	default: | ||||||
|  | 		var chunkSize int | ||||||
|  | 		if t.Elem().Kind() == reflect.Bool { | ||||||
|  | 			chunkSize = 16 | ||||||
|  | 		} else { | ||||||
|  | 			switch t.Elem().Bits() { | ||||||
|  | 			case 8: | ||||||
|  | 				chunkSize = 16 | ||||||
|  | 			case 16: | ||||||
|  | 				chunkSize = 12 | ||||||
|  | 			case 32: | ||||||
|  | 				chunkSize = 8 | ||||||
|  | 			default: | ||||||
|  | 				chunkSize = 8 | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		list = opts.formatDiffSlice( | ||||||
|  | 			vx, vy, chunkSize, t.Elem().Kind().String(), | ||||||
|  | 			func(v reflect.Value, d diffMode) textRecord { | ||||||
|  | 				var ss []string | ||||||
|  | 				for i := 0; i < v.Len(); i++ { | ||||||
|  | 					switch t.Elem().Kind() { | ||||||
|  | 					case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: | ||||||
|  | 						ss = append(ss, fmt.Sprint(v.Index(i).Int())) | ||||||
|  | 					case reflect.Uint, reflect.Uint16, reflect.Uint32, reflect.Uint64: | ||||||
|  | 						ss = append(ss, fmt.Sprint(v.Index(i).Uint())) | ||||||
|  | 					case reflect.Uint8, reflect.Uintptr: | ||||||
|  | 						ss = append(ss, formatHex(v.Index(i).Uint())) | ||||||
|  | 					case reflect.Bool, reflect.Float32, reflect.Float64, reflect.Complex64, reflect.Complex128: | ||||||
|  | 						ss = append(ss, fmt.Sprint(v.Index(i).Interface())) | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  | 				s := strings.Join(ss, ", ") | ||||||
|  | 				return textRecord{Diff: d, Value: textLine(s)} | ||||||
|  | 			}, | ||||||
|  | 		) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Wrap the output with appropriate type information. | ||||||
|  | 	var out textNode = &textWrap{Prefix: "{", Value: list, Suffix: "}"} | ||||||
|  | 	if !isMostlyText { | ||||||
|  | 		// The "{...}" byte-sequence literal is not valid Go syntax for strings. | ||||||
|  | 		// Emit the type for extra clarity (e.g. "string{...}"). | ||||||
|  | 		if t.Kind() == reflect.String { | ||||||
|  | 			opts = opts.WithTypeMode(emitType) | ||||||
|  | 		} | ||||||
|  | 		return opts.FormatType(t, out) | ||||||
|  | 	} | ||||||
|  | 	switch t.Kind() { | ||||||
|  | 	case reflect.String: | ||||||
|  | 		out = &textWrap{Prefix: "strings.Join(", Value: out, Suffix: fmt.Sprintf(", %q)", delim)} | ||||||
|  | 		if t != stringType { | ||||||
|  | 			out = opts.FormatType(t, out) | ||||||
|  | 		} | ||||||
|  | 	case reflect.Slice: | ||||||
|  | 		out = &textWrap{Prefix: "bytes.Join(", Value: out, Suffix: fmt.Sprintf(", %q)", delim)} | ||||||
|  | 		if t != bytesType { | ||||||
|  | 			out = opts.FormatType(t, out) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return out | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // formatASCII formats s as an ASCII string. | ||||||
|  | // This is useful for printing binary strings in a semi-legible way. | ||||||
|  | func formatASCII(s string) string { | ||||||
|  | 	b := bytes.Repeat([]byte{'.'}, len(s)) | ||||||
|  | 	for i := 0; i < len(s); i++ { | ||||||
|  | 		if ' ' <= s[i] && s[i] <= '~' { | ||||||
|  | 			b[i] = s[i] | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return string(b) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (opts formatOptions) formatDiffSlice( | ||||||
|  | 	vx, vy reflect.Value, chunkSize int, name string, | ||||||
|  | 	makeRec func(reflect.Value, diffMode) textRecord, | ||||||
|  | ) (list textList) { | ||||||
|  | 	eq := func(ix, iy int) bool { | ||||||
|  | 		return vx.Index(ix).Interface() == vy.Index(iy).Interface() | ||||||
|  | 	} | ||||||
|  | 	es := diff.Difference(vx.Len(), vy.Len(), func(ix, iy int) diff.Result { | ||||||
|  | 		return diff.BoolResult(eq(ix, iy)) | ||||||
|  | 	}) | ||||||
|  | 
 | ||||||
|  | 	appendChunks := func(v reflect.Value, d diffMode) int { | ||||||
|  | 		n0 := v.Len() | ||||||
|  | 		for v.Len() > 0 { | ||||||
|  | 			n := chunkSize | ||||||
|  | 			if n > v.Len() { | ||||||
|  | 				n = v.Len() | ||||||
|  | 			} | ||||||
|  | 			list = append(list, makeRec(v.Slice(0, n), d)) | ||||||
|  | 			v = v.Slice(n, v.Len()) | ||||||
|  | 		} | ||||||
|  | 		return n0 - v.Len() | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	var numDiffs int | ||||||
|  | 	maxLen := -1 | ||||||
|  | 	if opts.LimitVerbosity { | ||||||
|  | 		maxLen = (1 << opts.verbosity()) << 2 // 4, 8, 16, 32, 64, etc... | ||||||
|  | 		opts.VerbosityLevel-- | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	groups := coalesceAdjacentEdits(name, es) | ||||||
|  | 	groups = coalesceInterveningIdentical(groups, chunkSize/4) | ||||||
|  | 	groups = cleanupSurroundingIdentical(groups, eq) | ||||||
|  | 	maxGroup := diffStats{Name: name} | ||||||
|  | 	for i, ds := range groups { | ||||||
|  | 		if maxLen >= 0 && numDiffs >= maxLen { | ||||||
|  | 			maxGroup = maxGroup.Append(ds) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Print equal. | ||||||
|  | 		if ds.NumDiff() == 0 { | ||||||
|  | 			// Compute the number of leading and trailing equal bytes to print. | ||||||
|  | 			var numLo, numHi int | ||||||
|  | 			numEqual := ds.NumIgnored + ds.NumIdentical | ||||||
|  | 			for numLo < chunkSize*numContextRecords && numLo+numHi < numEqual && i != 0 { | ||||||
|  | 				numLo++ | ||||||
|  | 			} | ||||||
|  | 			for numHi < chunkSize*numContextRecords && numLo+numHi < numEqual && i != len(groups)-1 { | ||||||
|  | 				numHi++ | ||||||
|  | 			} | ||||||
|  | 			if numEqual-(numLo+numHi) <= chunkSize && ds.NumIgnored == 0 { | ||||||
|  | 				numHi = numEqual - numLo // Avoid pointless coalescing of single equal row | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Print the equal bytes. | ||||||
|  | 			appendChunks(vx.Slice(0, numLo), diffIdentical) | ||||||
|  | 			if numEqual > numLo+numHi { | ||||||
|  | 				ds.NumIdentical -= numLo + numHi | ||||||
|  | 				list.AppendEllipsis(ds) | ||||||
|  | 			} | ||||||
|  | 			appendChunks(vx.Slice(numEqual-numHi, numEqual), diffIdentical) | ||||||
|  | 			vx = vx.Slice(numEqual, vx.Len()) | ||||||
|  | 			vy = vy.Slice(numEqual, vy.Len()) | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Print unequal. | ||||||
|  | 		len0 := len(list) | ||||||
|  | 		nx := appendChunks(vx.Slice(0, ds.NumIdentical+ds.NumRemoved+ds.NumModified), diffRemoved) | ||||||
|  | 		vx = vx.Slice(nx, vx.Len()) | ||||||
|  | 		ny := appendChunks(vy.Slice(0, ds.NumIdentical+ds.NumInserted+ds.NumModified), diffInserted) | ||||||
|  | 		vy = vy.Slice(ny, vy.Len()) | ||||||
|  | 		numDiffs += len(list) - len0 | ||||||
|  | 	} | ||||||
|  | 	if maxGroup.IsZero() { | ||||||
|  | 		assert(vx.Len() == 0 && vy.Len() == 0) | ||||||
|  | 	} else { | ||||||
|  | 		list.AppendEllipsis(maxGroup) | ||||||
|  | 	} | ||||||
|  | 	return list | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // coalesceAdjacentEdits coalesces the list of edits into groups of adjacent | ||||||
|  | // equal or unequal counts. | ||||||
|  | // | ||||||
|  | // Example: | ||||||
|  | // | ||||||
|  | //	Input:  "..XXY...Y" | ||||||
|  | //	Output: [ | ||||||
|  | //		{NumIdentical: 2}, | ||||||
|  | //		{NumRemoved: 2, NumInserted 1}, | ||||||
|  | //		{NumIdentical: 3}, | ||||||
|  | //		{NumInserted: 1}, | ||||||
|  | //	] | ||||||
|  | func coalesceAdjacentEdits(name string, es diff.EditScript) (groups []diffStats) { | ||||||
|  | 	var prevMode byte | ||||||
|  | 	lastStats := func(mode byte) *diffStats { | ||||||
|  | 		if prevMode != mode { | ||||||
|  | 			groups = append(groups, diffStats{Name: name}) | ||||||
|  | 			prevMode = mode | ||||||
|  | 		} | ||||||
|  | 		return &groups[len(groups)-1] | ||||||
|  | 	} | ||||||
|  | 	for _, e := range es { | ||||||
|  | 		switch e { | ||||||
|  | 		case diff.Identity: | ||||||
|  | 			lastStats('=').NumIdentical++ | ||||||
|  | 		case diff.UniqueX: | ||||||
|  | 			lastStats('!').NumRemoved++ | ||||||
|  | 		case diff.UniqueY: | ||||||
|  | 			lastStats('!').NumInserted++ | ||||||
|  | 		case diff.Modified: | ||||||
|  | 			lastStats('!').NumModified++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return groups | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // coalesceInterveningIdentical coalesces sufficiently short (<= windowSize) | ||||||
|  | // equal groups into adjacent unequal groups that currently result in a | ||||||
|  | // dual inserted/removed printout. This acts as a high-pass filter to smooth | ||||||
|  | // out high-frequency changes within the windowSize. | ||||||
|  | // | ||||||
|  | // Example: | ||||||
|  | // | ||||||
|  | //	WindowSize: 16, | ||||||
|  | //	Input: [ | ||||||
|  | //		{NumIdentical: 61},              // group 0 | ||||||
|  | //		{NumRemoved: 3, NumInserted: 1}, // group 1 | ||||||
|  | //		{NumIdentical: 6},               // ├── coalesce | ||||||
|  | //		{NumInserted: 2},                // ├── coalesce | ||||||
|  | //		{NumIdentical: 1},               // ├── coalesce | ||||||
|  | //		{NumRemoved: 9},                 // └── coalesce | ||||||
|  | //		{NumIdentical: 64},              // group 2 | ||||||
|  | //		{NumRemoved: 3, NumInserted: 1}, // group 3 | ||||||
|  | //		{NumIdentical: 6},               // ├── coalesce | ||||||
|  | //		{NumInserted: 2},                // ├── coalesce | ||||||
|  | //		{NumIdentical: 1},               // ├── coalesce | ||||||
|  | //		{NumRemoved: 7},                 // ├── coalesce | ||||||
|  | //		{NumIdentical: 1},               // ├── coalesce | ||||||
|  | //		{NumRemoved: 2},                 // └── coalesce | ||||||
|  | //		{NumIdentical: 63},              // group 4 | ||||||
|  | //	] | ||||||
|  | //	Output: [ | ||||||
|  | //		{NumIdentical: 61}, | ||||||
|  | //		{NumIdentical: 7, NumRemoved: 12, NumInserted: 3}, | ||||||
|  | //		{NumIdentical: 64}, | ||||||
|  | //		{NumIdentical: 8, NumRemoved: 12, NumInserted: 3}, | ||||||
|  | //		{NumIdentical: 63}, | ||||||
|  | //	] | ||||||
|  | func coalesceInterveningIdentical(groups []diffStats, windowSize int) []diffStats { | ||||||
|  | 	groups, groupsOrig := groups[:0], groups | ||||||
|  | 	for i, ds := range groupsOrig { | ||||||
|  | 		if len(groups) >= 2 && ds.NumDiff() > 0 { | ||||||
|  | 			prev := &groups[len(groups)-2] // Unequal group | ||||||
|  | 			curr := &groups[len(groups)-1] // Equal group | ||||||
|  | 			next := &groupsOrig[i]         // Unequal group | ||||||
|  | 			hadX, hadY := prev.NumRemoved > 0, prev.NumInserted > 0 | ||||||
|  | 			hasX, hasY := next.NumRemoved > 0, next.NumInserted > 0 | ||||||
|  | 			if ((hadX || hasX) && (hadY || hasY)) && curr.NumIdentical <= windowSize { | ||||||
|  | 				*prev = prev.Append(*curr).Append(*next) | ||||||
|  | 				groups = groups[:len(groups)-1] // Truncate off equal group | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		groups = append(groups, ds) | ||||||
|  | 	} | ||||||
|  | 	return groups | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // cleanupSurroundingIdentical scans through all unequal groups, and | ||||||
|  | // moves any leading sequence of equal elements to the preceding equal group and | ||||||
|  | // moves and trailing sequence of equal elements to the succeeding equal group. | ||||||
|  | // | ||||||
|  | // This is necessary since coalesceInterveningIdentical may coalesce edit groups | ||||||
|  | // together such that leading/trailing spans of equal elements becomes possible. | ||||||
|  | // Note that this can occur even with an optimal diffing algorithm. | ||||||
|  | // | ||||||
|  | // Example: | ||||||
|  | // | ||||||
|  | //	Input: [ | ||||||
|  | //		{NumIdentical: 61}, | ||||||
|  | //		{NumIdentical: 1 , NumRemoved: 11, NumInserted: 2}, // assume 3 leading identical elements | ||||||
|  | //		{NumIdentical: 67}, | ||||||
|  | //		{NumIdentical: 7, NumRemoved: 12, NumInserted: 3},  // assume 10 trailing identical elements | ||||||
|  | //		{NumIdentical: 54}, | ||||||
|  | //	] | ||||||
|  | //	Output: [ | ||||||
|  | //		{NumIdentical: 64}, // incremented by 3 | ||||||
|  | //		{NumRemoved: 9}, | ||||||
|  | //		{NumIdentical: 67}, | ||||||
|  | //		{NumRemoved: 9}, | ||||||
|  | //		{NumIdentical: 64}, // incremented by 10 | ||||||
|  | //	] | ||||||
|  | func cleanupSurroundingIdentical(groups []diffStats, eq func(i, j int) bool) []diffStats { | ||||||
|  | 	var ix, iy int // indexes into sequence x and y | ||||||
|  | 	for i, ds := range groups { | ||||||
|  | 		// Handle equal group. | ||||||
|  | 		if ds.NumDiff() == 0 { | ||||||
|  | 			ix += ds.NumIdentical | ||||||
|  | 			iy += ds.NumIdentical | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		// Handle unequal group. | ||||||
|  | 		nx := ds.NumIdentical + ds.NumRemoved + ds.NumModified | ||||||
|  | 		ny := ds.NumIdentical + ds.NumInserted + ds.NumModified | ||||||
|  | 		var numLeadingIdentical, numTrailingIdentical int | ||||||
|  | 		for j := 0; j < nx && j < ny && eq(ix+j, iy+j); j++ { | ||||||
|  | 			numLeadingIdentical++ | ||||||
|  | 		} | ||||||
|  | 		for j := 0; j < nx && j < ny && eq(ix+nx-1-j, iy+ny-1-j); j++ { | ||||||
|  | 			numTrailingIdentical++ | ||||||
|  | 		} | ||||||
|  | 		if numIdentical := numLeadingIdentical + numTrailingIdentical; numIdentical > 0 { | ||||||
|  | 			if numLeadingIdentical > 0 { | ||||||
|  | 				// Remove leading identical span from this group and | ||||||
|  | 				// insert it into the preceding group. | ||||||
|  | 				if i-1 >= 0 { | ||||||
|  | 					groups[i-1].NumIdentical += numLeadingIdentical | ||||||
|  | 				} else { | ||||||
|  | 					// No preceding group exists, so prepend a new group, | ||||||
|  | 					// but do so after we finish iterating over all groups. | ||||||
|  | 					defer func() { | ||||||
|  | 						groups = append([]diffStats{{Name: groups[0].Name, NumIdentical: numLeadingIdentical}}, groups...) | ||||||
|  | 					}() | ||||||
|  | 				} | ||||||
|  | 				// Increment indexes since the preceding group would have handled this. | ||||||
|  | 				ix += numLeadingIdentical | ||||||
|  | 				iy += numLeadingIdentical | ||||||
|  | 			} | ||||||
|  | 			if numTrailingIdentical > 0 { | ||||||
|  | 				// Remove trailing identical span from this group and | ||||||
|  | 				// insert it into the succeeding group. | ||||||
|  | 				if i+1 < len(groups) { | ||||||
|  | 					groups[i+1].NumIdentical += numTrailingIdentical | ||||||
|  | 				} else { | ||||||
|  | 					// No succeeding group exists, so append a new group, | ||||||
|  | 					// but do so after we finish iterating over all groups. | ||||||
|  | 					defer func() { | ||||||
|  | 						groups = append(groups, diffStats{Name: groups[len(groups)-1].Name, NumIdentical: numTrailingIdentical}) | ||||||
|  | 					}() | ||||||
|  | 				} | ||||||
|  | 				// Do not increment indexes since the succeeding group will handle this. | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			// Update this group since some identical elements were removed. | ||||||
|  | 			nx -= numIdentical | ||||||
|  | 			ny -= numIdentical | ||||||
|  | 			groups[i] = diffStats{Name: ds.Name, NumRemoved: nx, NumInserted: ny} | ||||||
|  | 		} | ||||||
|  | 		ix += nx | ||||||
|  | 		iy += ny | ||||||
|  | 	} | ||||||
|  | 	return groups | ||||||
|  | } | ||||||
							
								
								
									
										432
									
								
								vendor/github.com/google/go-cmp/cmp/report_text.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										432
									
								
								vendor/github.com/google/go-cmp/cmp/report_text.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,432 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"bytes" | ||||||
|  | 	"fmt" | ||||||
|  | 	"math/rand" | ||||||
|  | 	"strings" | ||||||
|  | 	"time" | ||||||
|  | 	"unicode/utf8" | ||||||
|  | 
 | ||||||
|  | 	"github.com/google/go-cmp/cmp/internal/flags" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | var randBool = rand.New(rand.NewSource(time.Now().Unix())).Intn(2) == 0 | ||||||
|  | 
 | ||||||
|  | const maxColumnLength = 80 | ||||||
|  | 
 | ||||||
|  | type indentMode int | ||||||
|  | 
 | ||||||
|  | func (n indentMode) appendIndent(b []byte, d diffMode) []byte { | ||||||
|  | 	// The output of Diff is documented as being unstable to provide future | ||||||
|  | 	// flexibility in changing the output for more humanly readable reports. | ||||||
|  | 	// This logic intentionally introduces instability to the exact output | ||||||
|  | 	// so that users can detect accidental reliance on stability early on, | ||||||
|  | 	// rather than much later when an actual change to the format occurs. | ||||||
|  | 	if flags.Deterministic || randBool { | ||||||
|  | 		// Use regular spaces (U+0020). | ||||||
|  | 		switch d { | ||||||
|  | 		case diffUnknown, diffIdentical: | ||||||
|  | 			b = append(b, "  "...) | ||||||
|  | 		case diffRemoved: | ||||||
|  | 			b = append(b, "- "...) | ||||||
|  | 		case diffInserted: | ||||||
|  | 			b = append(b, "+ "...) | ||||||
|  | 		} | ||||||
|  | 	} else { | ||||||
|  | 		// Use non-breaking spaces (U+00a0). | ||||||
|  | 		switch d { | ||||||
|  | 		case diffUnknown, diffIdentical: | ||||||
|  | 			b = append(b, "  "...) | ||||||
|  | 		case diffRemoved: | ||||||
|  | 			b = append(b, "- "...) | ||||||
|  | 		case diffInserted: | ||||||
|  | 			b = append(b, "+ "...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return repeatCount(n).appendChar(b, '\t') | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type repeatCount int | ||||||
|  | 
 | ||||||
|  | func (n repeatCount) appendChar(b []byte, c byte) []byte { | ||||||
|  | 	for ; n > 0; n-- { | ||||||
|  | 		b = append(b, c) | ||||||
|  | 	} | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // textNode is a simplified tree-based representation of structured text. | ||||||
|  | // Possible node types are textWrap, textList, or textLine. | ||||||
|  | type textNode interface { | ||||||
|  | 	// Len reports the length in bytes of a single-line version of the tree. | ||||||
|  | 	// Nested textRecord.Diff and textRecord.Comment fields are ignored. | ||||||
|  | 	Len() int | ||||||
|  | 	// Equal reports whether the two trees are structurally identical. | ||||||
|  | 	// Nested textRecord.Diff and textRecord.Comment fields are compared. | ||||||
|  | 	Equal(textNode) bool | ||||||
|  | 	// String returns the string representation of the text tree. | ||||||
|  | 	// It is not guaranteed that len(x.String()) == x.Len(), | ||||||
|  | 	// nor that x.String() == y.String() implies that x.Equal(y). | ||||||
|  | 	String() string | ||||||
|  | 
 | ||||||
|  | 	// formatCompactTo formats the contents of the tree as a single-line string | ||||||
|  | 	// to the provided buffer. Any nested textRecord.Diff and textRecord.Comment | ||||||
|  | 	// fields are ignored. | ||||||
|  | 	// | ||||||
|  | 	// However, not all nodes in the tree should be collapsed as a single-line. | ||||||
|  | 	// If a node can be collapsed as a single-line, it is replaced by a textLine | ||||||
|  | 	// node. Since the top-level node cannot replace itself, this also returns | ||||||
|  | 	// the current node itself. | ||||||
|  | 	// | ||||||
|  | 	// This does not mutate the receiver. | ||||||
|  | 	formatCompactTo([]byte, diffMode) ([]byte, textNode) | ||||||
|  | 	// formatExpandedTo formats the contents of the tree as a multi-line string | ||||||
|  | 	// to the provided buffer. In order for column alignment to operate well, | ||||||
|  | 	// formatCompactTo must be called before calling formatExpandedTo. | ||||||
|  | 	formatExpandedTo([]byte, diffMode, indentMode) []byte | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // textWrap is a wrapper that concatenates a prefix and/or a suffix | ||||||
|  | // to the underlying node. | ||||||
|  | type textWrap struct { | ||||||
|  | 	Prefix   string      // e.g., "bytes.Buffer{" | ||||||
|  | 	Value    textNode    // textWrap | textList | textLine | ||||||
|  | 	Suffix   string      // e.g., "}" | ||||||
|  | 	Metadata interface{} // arbitrary metadata; has no effect on formatting | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s *textWrap) Len() int { | ||||||
|  | 	return len(s.Prefix) + s.Value.Len() + len(s.Suffix) | ||||||
|  | } | ||||||
|  | func (s1 *textWrap) Equal(s2 textNode) bool { | ||||||
|  | 	if s2, ok := s2.(*textWrap); ok { | ||||||
|  | 		return s1.Prefix == s2.Prefix && s1.Value.Equal(s2.Value) && s1.Suffix == s2.Suffix | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  | func (s *textWrap) String() string { | ||||||
|  | 	var d diffMode | ||||||
|  | 	var n indentMode | ||||||
|  | 	_, s2 := s.formatCompactTo(nil, d) | ||||||
|  | 	b := n.appendIndent(nil, d)      // Leading indent | ||||||
|  | 	b = s2.formatExpandedTo(b, d, n) // Main body | ||||||
|  | 	b = append(b, '\n')              // Trailing newline | ||||||
|  | 	return string(b) | ||||||
|  | } | ||||||
|  | func (s *textWrap) formatCompactTo(b []byte, d diffMode) ([]byte, textNode) { | ||||||
|  | 	n0 := len(b) // Original buffer length | ||||||
|  | 	b = append(b, s.Prefix...) | ||||||
|  | 	b, s.Value = s.Value.formatCompactTo(b, d) | ||||||
|  | 	b = append(b, s.Suffix...) | ||||||
|  | 	if _, ok := s.Value.(textLine); ok { | ||||||
|  | 		return b, textLine(b[n0:]) | ||||||
|  | 	} | ||||||
|  | 	return b, s | ||||||
|  | } | ||||||
|  | func (s *textWrap) formatExpandedTo(b []byte, d diffMode, n indentMode) []byte { | ||||||
|  | 	b = append(b, s.Prefix...) | ||||||
|  | 	b = s.Value.formatExpandedTo(b, d, n) | ||||||
|  | 	b = append(b, s.Suffix...) | ||||||
|  | 	return b | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // textList is a comma-separated list of textWrap or textLine nodes. | ||||||
|  | // The list may be formatted as multi-lines or single-line at the discretion | ||||||
|  | // of the textList.formatCompactTo method. | ||||||
|  | type textList []textRecord | ||||||
|  | type textRecord struct { | ||||||
|  | 	Diff       diffMode     // e.g., 0 or '-' or '+' | ||||||
|  | 	Key        string       // e.g., "MyField" | ||||||
|  | 	Value      textNode     // textWrap | textLine | ||||||
|  | 	ElideComma bool         // avoid trailing comma | ||||||
|  | 	Comment    fmt.Stringer // e.g., "6 identical fields" | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // AppendEllipsis appends a new ellipsis node to the list if none already | ||||||
|  | // exists at the end. If cs is non-zero it coalesces the statistics with the | ||||||
|  | // previous diffStats. | ||||||
|  | func (s *textList) AppendEllipsis(ds diffStats) { | ||||||
|  | 	hasStats := !ds.IsZero() | ||||||
|  | 	if len(*s) == 0 || !(*s)[len(*s)-1].Value.Equal(textEllipsis) { | ||||||
|  | 		if hasStats { | ||||||
|  | 			*s = append(*s, textRecord{Value: textEllipsis, ElideComma: true, Comment: ds}) | ||||||
|  | 		} else { | ||||||
|  | 			*s = append(*s, textRecord{Value: textEllipsis, ElideComma: true}) | ||||||
|  | 		} | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 	if hasStats { | ||||||
|  | 		(*s)[len(*s)-1].Comment = (*s)[len(*s)-1].Comment.(diffStats).Append(ds) | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s textList) Len() (n int) { | ||||||
|  | 	for i, r := range s { | ||||||
|  | 		n += len(r.Key) | ||||||
|  | 		if r.Key != "" { | ||||||
|  | 			n += len(": ") | ||||||
|  | 		} | ||||||
|  | 		n += r.Value.Len() | ||||||
|  | 		if i < len(s)-1 { | ||||||
|  | 			n += len(", ") | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return n | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s1 textList) Equal(s2 textNode) bool { | ||||||
|  | 	if s2, ok := s2.(textList); ok { | ||||||
|  | 		if len(s1) != len(s2) { | ||||||
|  | 			return false | ||||||
|  | 		} | ||||||
|  | 		for i := range s1 { | ||||||
|  | 			r1, r2 := s1[i], s2[i] | ||||||
|  | 			if !(r1.Diff == r2.Diff && r1.Key == r2.Key && r1.Value.Equal(r2.Value) && r1.Comment == r2.Comment) { | ||||||
|  | 				return false | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		return true | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s textList) String() string { | ||||||
|  | 	return (&textWrap{Prefix: "{", Value: s, Suffix: "}"}).String() | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s textList) formatCompactTo(b []byte, d diffMode) ([]byte, textNode) { | ||||||
|  | 	s = append(textList(nil), s...) // Avoid mutating original | ||||||
|  | 
 | ||||||
|  | 	// Determine whether we can collapse this list as a single line. | ||||||
|  | 	n0 := len(b) // Original buffer length | ||||||
|  | 	var multiLine bool | ||||||
|  | 	for i, r := range s { | ||||||
|  | 		if r.Diff == diffInserted || r.Diff == diffRemoved { | ||||||
|  | 			multiLine = true | ||||||
|  | 		} | ||||||
|  | 		b = append(b, r.Key...) | ||||||
|  | 		if r.Key != "" { | ||||||
|  | 			b = append(b, ": "...) | ||||||
|  | 		} | ||||||
|  | 		b, s[i].Value = r.Value.formatCompactTo(b, d|r.Diff) | ||||||
|  | 		if _, ok := s[i].Value.(textLine); !ok { | ||||||
|  | 			multiLine = true | ||||||
|  | 		} | ||||||
|  | 		if r.Comment != nil { | ||||||
|  | 			multiLine = true | ||||||
|  | 		} | ||||||
|  | 		if i < len(s)-1 { | ||||||
|  | 			b = append(b, ", "...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	// Force multi-lined output when printing a removed/inserted node that | ||||||
|  | 	// is sufficiently long. | ||||||
|  | 	if (d == diffInserted || d == diffRemoved) && len(b[n0:]) > maxColumnLength { | ||||||
|  | 		multiLine = true | ||||||
|  | 	} | ||||||
|  | 	if !multiLine { | ||||||
|  | 		return b, textLine(b[n0:]) | ||||||
|  | 	} | ||||||
|  | 	return b, s | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s textList) formatExpandedTo(b []byte, d diffMode, n indentMode) []byte { | ||||||
|  | 	alignKeyLens := s.alignLens( | ||||||
|  | 		func(r textRecord) bool { | ||||||
|  | 			_, isLine := r.Value.(textLine) | ||||||
|  | 			return r.Key == "" || !isLine | ||||||
|  | 		}, | ||||||
|  | 		func(r textRecord) int { return utf8.RuneCountInString(r.Key) }, | ||||||
|  | 	) | ||||||
|  | 	alignValueLens := s.alignLens( | ||||||
|  | 		func(r textRecord) bool { | ||||||
|  | 			_, isLine := r.Value.(textLine) | ||||||
|  | 			return !isLine || r.Value.Equal(textEllipsis) || r.Comment == nil | ||||||
|  | 		}, | ||||||
|  | 		func(r textRecord) int { return utf8.RuneCount(r.Value.(textLine)) }, | ||||||
|  | 	) | ||||||
|  | 
 | ||||||
|  | 	// Format lists of simple lists in a batched form. | ||||||
|  | 	// If the list is sequence of only textLine values, | ||||||
|  | 	// then batch multiple values on a single line. | ||||||
|  | 	var isSimple bool | ||||||
|  | 	for _, r := range s { | ||||||
|  | 		_, isLine := r.Value.(textLine) | ||||||
|  | 		isSimple = r.Diff == 0 && r.Key == "" && isLine && r.Comment == nil | ||||||
|  | 		if !isSimple { | ||||||
|  | 			break | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	if isSimple { | ||||||
|  | 		n++ | ||||||
|  | 		var batch []byte | ||||||
|  | 		emitBatch := func() { | ||||||
|  | 			if len(batch) > 0 { | ||||||
|  | 				b = n.appendIndent(append(b, '\n'), d) | ||||||
|  | 				b = append(b, bytes.TrimRight(batch, " ")...) | ||||||
|  | 				batch = batch[:0] | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 		for _, r := range s { | ||||||
|  | 			line := r.Value.(textLine) | ||||||
|  | 			if len(batch)+len(line)+len(", ") > maxColumnLength { | ||||||
|  | 				emitBatch() | ||||||
|  | 			} | ||||||
|  | 			batch = append(batch, line...) | ||||||
|  | 			batch = append(batch, ", "...) | ||||||
|  | 		} | ||||||
|  | 		emitBatch() | ||||||
|  | 		n-- | ||||||
|  | 		return n.appendIndent(append(b, '\n'), d) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Format the list as a multi-lined output. | ||||||
|  | 	n++ | ||||||
|  | 	for i, r := range s { | ||||||
|  | 		b = n.appendIndent(append(b, '\n'), d|r.Diff) | ||||||
|  | 		if r.Key != "" { | ||||||
|  | 			b = append(b, r.Key+": "...) | ||||||
|  | 		} | ||||||
|  | 		b = alignKeyLens[i].appendChar(b, ' ') | ||||||
|  | 
 | ||||||
|  | 		b = r.Value.formatExpandedTo(b, d|r.Diff, n) | ||||||
|  | 		if !r.ElideComma { | ||||||
|  | 			b = append(b, ',') | ||||||
|  | 		} | ||||||
|  | 		b = alignValueLens[i].appendChar(b, ' ') | ||||||
|  | 
 | ||||||
|  | 		if r.Comment != nil { | ||||||
|  | 			b = append(b, " // "+r.Comment.String()...) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	n-- | ||||||
|  | 
 | ||||||
|  | 	return n.appendIndent(append(b, '\n'), d) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s textList) alignLens( | ||||||
|  | 	skipFunc func(textRecord) bool, | ||||||
|  | 	lenFunc func(textRecord) int, | ||||||
|  | ) []repeatCount { | ||||||
|  | 	var startIdx, endIdx, maxLen int | ||||||
|  | 	lens := make([]repeatCount, len(s)) | ||||||
|  | 	for i, r := range s { | ||||||
|  | 		if skipFunc(r) { | ||||||
|  | 			for j := startIdx; j < endIdx && j < len(s); j++ { | ||||||
|  | 				lens[j] = repeatCount(maxLen - lenFunc(s[j])) | ||||||
|  | 			} | ||||||
|  | 			startIdx, endIdx, maxLen = i+1, i+1, 0 | ||||||
|  | 		} else { | ||||||
|  | 			if maxLen < lenFunc(r) { | ||||||
|  | 				maxLen = lenFunc(r) | ||||||
|  | 			} | ||||||
|  | 			endIdx = i + 1 | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	for j := startIdx; j < endIdx && j < len(s); j++ { | ||||||
|  | 		lens[j] = repeatCount(maxLen - lenFunc(s[j])) | ||||||
|  | 	} | ||||||
|  | 	return lens | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // textLine is a single-line segment of text and is always a leaf node | ||||||
|  | // in the textNode tree. | ||||||
|  | type textLine []byte | ||||||
|  | 
 | ||||||
|  | var ( | ||||||
|  | 	textNil      = textLine("nil") | ||||||
|  | 	textEllipsis = textLine("...") | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | func (s textLine) Len() int { | ||||||
|  | 	return len(s) | ||||||
|  | } | ||||||
|  | func (s1 textLine) Equal(s2 textNode) bool { | ||||||
|  | 	if s2, ok := s2.(textLine); ok { | ||||||
|  | 		return bytes.Equal([]byte(s1), []byte(s2)) | ||||||
|  | 	} | ||||||
|  | 	return false | ||||||
|  | } | ||||||
|  | func (s textLine) String() string { | ||||||
|  | 	return string(s) | ||||||
|  | } | ||||||
|  | func (s textLine) formatCompactTo(b []byte, d diffMode) ([]byte, textNode) { | ||||||
|  | 	return append(b, s...), s | ||||||
|  | } | ||||||
|  | func (s textLine) formatExpandedTo(b []byte, _ diffMode, _ indentMode) []byte { | ||||||
|  | 	return append(b, s...) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type diffStats struct { | ||||||
|  | 	Name         string | ||||||
|  | 	NumIgnored   int | ||||||
|  | 	NumIdentical int | ||||||
|  | 	NumRemoved   int | ||||||
|  | 	NumInserted  int | ||||||
|  | 	NumModified  int | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s diffStats) IsZero() bool { | ||||||
|  | 	s.Name = "" | ||||||
|  | 	return s == diffStats{} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s diffStats) NumDiff() int { | ||||||
|  | 	return s.NumRemoved + s.NumInserted + s.NumModified | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (s diffStats) Append(ds diffStats) diffStats { | ||||||
|  | 	assert(s.Name == ds.Name) | ||||||
|  | 	s.NumIgnored += ds.NumIgnored | ||||||
|  | 	s.NumIdentical += ds.NumIdentical | ||||||
|  | 	s.NumRemoved += ds.NumRemoved | ||||||
|  | 	s.NumInserted += ds.NumInserted | ||||||
|  | 	s.NumModified += ds.NumModified | ||||||
|  | 	return s | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // String prints a humanly-readable summary of coalesced records. | ||||||
|  | // | ||||||
|  | // Example: | ||||||
|  | // | ||||||
|  | //	diffStats{Name: "Field", NumIgnored: 5}.String() => "5 ignored fields" | ||||||
|  | func (s diffStats) String() string { | ||||||
|  | 	var ss []string | ||||||
|  | 	var sum int | ||||||
|  | 	labels := [...]string{"ignored", "identical", "removed", "inserted", "modified"} | ||||||
|  | 	counts := [...]int{s.NumIgnored, s.NumIdentical, s.NumRemoved, s.NumInserted, s.NumModified} | ||||||
|  | 	for i, n := range counts { | ||||||
|  | 		if n > 0 { | ||||||
|  | 			ss = append(ss, fmt.Sprintf("%d %v", n, labels[i])) | ||||||
|  | 		} | ||||||
|  | 		sum += n | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Pluralize the name (adjusting for some obscure English grammar rules). | ||||||
|  | 	name := s.Name | ||||||
|  | 	if sum > 1 { | ||||||
|  | 		name += "s" | ||||||
|  | 		if strings.HasSuffix(name, "ys") { | ||||||
|  | 			name = name[:len(name)-2] + "ies" // e.g., "entrys" => "entries" | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	// Format the list according to English grammar (with Oxford comma). | ||||||
|  | 	switch n := len(ss); n { | ||||||
|  | 	case 0: | ||||||
|  | 		return "" | ||||||
|  | 	case 1, 2: | ||||||
|  | 		return strings.Join(ss, " and ") + " " + name | ||||||
|  | 	default: | ||||||
|  | 		return strings.Join(ss[:n-1], ", ") + ", and " + ss[n-1] + " " + name | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | type commentString string | ||||||
|  | 
 | ||||||
|  | func (s commentString) String() string { return string(s) } | ||||||
							
								
								
									
										121
									
								
								vendor/github.com/google/go-cmp/cmp/report_value.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										121
									
								
								vendor/github.com/google/go-cmp/cmp/report_value.go
									
										
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							|  | @ -0,0 +1,121 @@ | ||||||
|  | // Copyright 2019, The Go Authors. All rights reserved. | ||||||
|  | // Use of this source code is governed by a BSD-style | ||||||
|  | // license that can be found in the LICENSE file. | ||||||
|  | 
 | ||||||
|  | package cmp | ||||||
|  | 
 | ||||||
|  | import "reflect" | ||||||
|  | 
 | ||||||
|  | // valueNode represents a single node within a report, which is a | ||||||
|  | // structured representation of the value tree, containing information | ||||||
|  | // regarding which nodes are equal or not. | ||||||
|  | type valueNode struct { | ||||||
|  | 	parent *valueNode | ||||||
|  | 
 | ||||||
|  | 	Type   reflect.Type | ||||||
|  | 	ValueX reflect.Value | ||||||
|  | 	ValueY reflect.Value | ||||||
|  | 
 | ||||||
|  | 	// NumSame is the number of leaf nodes that are equal. | ||||||
|  | 	// All descendants are equal only if NumDiff is 0. | ||||||
|  | 	NumSame int | ||||||
|  | 	// NumDiff is the number of leaf nodes that are not equal. | ||||||
|  | 	NumDiff int | ||||||
|  | 	// NumIgnored is the number of leaf nodes that are ignored. | ||||||
|  | 	NumIgnored int | ||||||
|  | 	// NumCompared is the number of leaf nodes that were compared | ||||||
|  | 	// using an Equal method or Comparer function. | ||||||
|  | 	NumCompared int | ||||||
|  | 	// NumTransformed is the number of non-leaf nodes that were transformed. | ||||||
|  | 	NumTransformed int | ||||||
|  | 	// NumChildren is the number of transitive descendants of this node. | ||||||
|  | 	// This counts from zero; thus, leaf nodes have no descendants. | ||||||
|  | 	NumChildren int | ||||||
|  | 	// MaxDepth is the maximum depth of the tree. This counts from zero; | ||||||
|  | 	// thus, leaf nodes have a depth of zero. | ||||||
|  | 	MaxDepth int | ||||||
|  | 
 | ||||||
|  | 	// Records is a list of struct fields, slice elements, or map entries. | ||||||
|  | 	Records []reportRecord // If populated, implies Value is not populated | ||||||
|  | 
 | ||||||
|  | 	// Value is the result of a transformation, pointer indirect, of | ||||||
|  | 	// type assertion. | ||||||
|  | 	Value *valueNode // If populated, implies Records is not populated | ||||||
|  | 
 | ||||||
|  | 	// TransformerName is the name of the transformer. | ||||||
|  | 	TransformerName string // If non-empty, implies Value is populated | ||||||
|  | } | ||||||
|  | type reportRecord struct { | ||||||
|  | 	Key   reflect.Value // Invalid for slice element | ||||||
|  | 	Value *valueNode | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (parent *valueNode) PushStep(ps PathStep) (child *valueNode) { | ||||||
|  | 	vx, vy := ps.Values() | ||||||
|  | 	child = &valueNode{parent: parent, Type: ps.Type(), ValueX: vx, ValueY: vy} | ||||||
|  | 	switch s := ps.(type) { | ||||||
|  | 	case StructField: | ||||||
|  | 		assert(parent.Value == nil) | ||||||
|  | 		parent.Records = append(parent.Records, reportRecord{Key: reflect.ValueOf(s.Name()), Value: child}) | ||||||
|  | 	case SliceIndex: | ||||||
|  | 		assert(parent.Value == nil) | ||||||
|  | 		parent.Records = append(parent.Records, reportRecord{Value: child}) | ||||||
|  | 	case MapIndex: | ||||||
|  | 		assert(parent.Value == nil) | ||||||
|  | 		parent.Records = append(parent.Records, reportRecord{Key: s.Key(), Value: child}) | ||||||
|  | 	case Indirect: | ||||||
|  | 		assert(parent.Value == nil && parent.Records == nil) | ||||||
|  | 		parent.Value = child | ||||||
|  | 	case TypeAssertion: | ||||||
|  | 		assert(parent.Value == nil && parent.Records == nil) | ||||||
|  | 		parent.Value = child | ||||||
|  | 	case Transform: | ||||||
|  | 		assert(parent.Value == nil && parent.Records == nil) | ||||||
|  | 		parent.Value = child | ||||||
|  | 		parent.TransformerName = s.Name() | ||||||
|  | 		parent.NumTransformed++ | ||||||
|  | 	default: | ||||||
|  | 		assert(parent == nil) // Must be the root step | ||||||
|  | 	} | ||||||
|  | 	return child | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (r *valueNode) Report(rs Result) { | ||||||
|  | 	assert(r.MaxDepth == 0) // May only be called on leaf nodes | ||||||
|  | 
 | ||||||
|  | 	if rs.ByIgnore() { | ||||||
|  | 		r.NumIgnored++ | ||||||
|  | 	} else { | ||||||
|  | 		if rs.Equal() { | ||||||
|  | 			r.NumSame++ | ||||||
|  | 		} else { | ||||||
|  | 			r.NumDiff++ | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	assert(r.NumSame+r.NumDiff+r.NumIgnored == 1) | ||||||
|  | 
 | ||||||
|  | 	if rs.ByMethod() { | ||||||
|  | 		r.NumCompared++ | ||||||
|  | 	} | ||||||
|  | 	if rs.ByFunc() { | ||||||
|  | 		r.NumCompared++ | ||||||
|  | 	} | ||||||
|  | 	assert(r.NumCompared <= 1) | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | func (child *valueNode) PopStep() (parent *valueNode) { | ||||||
|  | 	if child.parent == nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 	parent = child.parent | ||||||
|  | 	parent.NumSame += child.NumSame | ||||||
|  | 	parent.NumDiff += child.NumDiff | ||||||
|  | 	parent.NumIgnored += child.NumIgnored | ||||||
|  | 	parent.NumCompared += child.NumCompared | ||||||
|  | 	parent.NumTransformed += child.NumTransformed | ||||||
|  | 	parent.NumChildren += child.NumChildren + 1 | ||||||
|  | 	if parent.MaxDepth < child.MaxDepth+1 { | ||||||
|  | 		parent.MaxDepth = child.MaxDepth + 1 | ||||||
|  | 	} | ||||||
|  | 	return parent | ||||||
|  | } | ||||||
							
								
								
									
										7
									
								
								vendor/modules.txt
									
										
									
									
										vendored
									
									
								
							
							
						
						
									
										7
									
								
								vendor/modules.txt
									
										
									
									
										vendored
									
									
								
							|  | @ -353,6 +353,13 @@ github.com/golang/geo/r2 | ||||||
| github.com/golang/geo/r3 | github.com/golang/geo/r3 | ||||||
| github.com/golang/geo/s1 | github.com/golang/geo/s1 | ||||||
| github.com/golang/geo/s2 | github.com/golang/geo/s2 | ||||||
|  | # github.com/google/go-cmp v0.6.0 | ||||||
|  | ## explicit; go 1.13 | ||||||
|  | github.com/google/go-cmp/cmp | ||||||
|  | github.com/google/go-cmp/cmp/internal/diff | ||||||
|  | github.com/google/go-cmp/cmp/internal/flags | ||||||
|  | github.com/google/go-cmp/cmp/internal/function | ||||||
|  | github.com/google/go-cmp/cmp/internal/value | ||||||
| # github.com/google/uuid v1.6.0 | # github.com/google/uuid v1.6.0 | ||||||
| ## explicit | ## explicit | ||||||
| github.com/google/uuid | github.com/google/uuid | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue