Method InterceptQuery

proxycore/mockcluster.go:209–275

(hdr *frame.Header, msg *message.Query)

}

func (c *MockClient) InterceptQuery(hdr *frame.Header, msg *message.Query) message.Message {
	handled, stmt, err := parser.IsQueryHandled(parser.IdentifierFromString(c.keyspace), msg.Query)

	if handled {
		if err != nil {
			return &message.Invalid{ErrorMessage: err.Error()}
		}

		switch s := stmt.(type) {
		case *parser.SelectStatement:
			if s.Table == "local" {
				vals := c.makeSystemLocalValues(hdr.Version, mockSchemaVersion)
				localColumns := parser.SystemLocalColumns
				if len(c.server.DseVersion) > 0 {
					localColumns = parser.DseSystemLocalColumns
				}
				if columns, err := parser.FilterColumns(s, localColumns); err != nil {
					return &message.Invalid{ErrorMessage: err.Error()}
				} else if row, err := c.filterValues(hdr.Version, s, localColumns, vals, 1); err != nil {
					return &message.Invalid{ErrorMessage: err.Error()}
				} else {
					return &message.RowsResult{
						Metadata: &message.RowsMetadata{
							ColumnCount: int32(len(columns)),
							Columns:     columns,
						},
						Data: []message.Row{row},
					}
				}
			} else if s.Table == "peers" {
				peersColumns := parser.SystemPeersColumns
				if len(c.server.DseVersion) > 0 {
					peersColumns = parser.DseSystemPeersColumns
				}
				if columns, err := parser.FilterColumns(s, peersColumns); err != nil {
					return &message.Invalid{ErrorMessage: err.Error()}
				} else {
					var data []message.Row
					peers := c.server.copyPeers()
					for _, peer := range peers {
						vals := c.makeSystemPeerValues(hdr.Version, peer.IP, peer.HostID, mockSchemaVersion)
						if row, err := c.filterValues(hdr.Version, s, peersColumns, vals, len(peers)); err != nil {
							return &message.Invalid{ErrorMessage: err.Error()}
						} else {
							data = append(data, row)
						}
					}
					return &message.RowsResult{
						Metadata: &message.RowsMetadata{
							ColumnCount: int32(len(columns)),
							Columns:     columns,
						},
						Data: data,
					}
				}
			} else {
				return &message.Invalid{ErrorMessage: "Doesn't exist"}
			}