Skip to content

Commit

Permalink
update ast to support generic
Browse files Browse the repository at this point in the history
  • Loading branch information
aqaurius6666 committed Mar 11, 2024
1 parent e57deea commit fecc592
Show file tree
Hide file tree
Showing 6 changed files with 111 additions and 0 deletions.
7 changes: 7 additions & 0 deletions internal/wire/copyast.go
Expand Up @@ -263,6 +263,13 @@ func copyAST(original ast.Node) ast.Node {
Index: exprFromMap(m, node.Index),
Rbrack: node.Rbrack,
}
case *ast.IndexListExpr:
m[node] = &ast.IndexListExpr{
X: exprFromMap(m, node.X),
Lbrack: node.Lbrack,
Indices: copyExprList(m, node.Indices),
Rbrack: node.Rbrack,
}
case *ast.InterfaceType:
m[node] = &ast.InterfaceType{
Interface: node.Interface,
Expand Down
54 changes: 54 additions & 0 deletions internal/wire/testdata/Generic/foo/foo.go
@@ -0,0 +1,54 @@
// Copyright 2018 The Wire Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main

type Foo string

type FooStorer interface {
Set(key string, value Foo)
Get(key string) (Foo, bool)
}

type Store[TKey comparable, TValue any] struct {
store map[TKey]TValue
}

func NewStore[TKey comparable, TValue any]() Store[TKey, TValue] {
return Store[TKey, TValue]{store: make(map[TKey]TValue)}
}

func (s Store[TKey, TValue]) Set(key TKey, value TValue) {
s.store[key] = value
}

func (s Store[TKey, TValue]) Get(key TKey) (TValue, bool) {
value, ok := s.store[key]
return value, ok
}

func NewFooStore(foo Foo) Store[string, Foo] {
r := NewStore[string, Foo]()
r.Set("foo", foo)
return r
}

func main() {
fooStore := InitializeFooStore()
v, ok := fooStore.Get("foo")
if !ok {
panic("foo not found")
}
print(v)

}
29 changes: 29 additions & 0 deletions internal/wire/testdata/Generic/foo/wire.go
@@ -0,0 +1,29 @@
// Copyright 2018 The Wire Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//go:build wireinject
// +build wireinject

// wire.go
package main

import "github.com/google/wire"

func InitializeFooStore() FooStorer {
wire.Build(
NewFooStore,
wire.Value(Foo("foo hello value")),
wire.Bind(new(FooStorer), new(Store[string, Foo])),
)
return nil
}
1 change: 1 addition & 0 deletions internal/wire/testdata/Generic/pkg
@@ -0,0 +1 @@
example.com/foo
1 change: 1 addition & 0 deletions internal/wire/testdata/Generic/want/program_out.txt
@@ -0,0 +1 @@
foo hello value
19 changes: 19 additions & 0 deletions internal/wire/testdata/Generic/want/wire_gen.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit fecc592

Please sign in to comment.