template 함수 추가
This commit is contained in:
parent
c9fbc24e51
commit
08ec19535d
|
@ -0,0 +1,23 @@
|
|||
package tmpl
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
_ "unsafe"
|
||||
)
|
||||
|
||||
//go:linkname indirectInterface text/template.indirectInterface
|
||||
//go:nosplit
|
||||
func indirectInterface(v reflect.Value) reflect.Value
|
||||
|
||||
//go:linkname truth text/template.truth
|
||||
//go:nosplit
|
||||
func truth(arg reflect.Value) bool
|
||||
|
||||
//go:linkname isTrue text/template.isTrue
|
||||
//go:nosplit
|
||||
func isTrue(arg reflect.Value) (truth, ok bool)
|
||||
|
||||
func evalTrue(val reflect.Value) (truth bool) {
|
||||
truth, _ = isTrue(val)
|
||||
return
|
||||
}
|
|
@ -0,0 +1,66 @@
|
|||
package tmpl
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
_ "unsafe"
|
||||
)
|
||||
|
||||
type TmplMap struct {
|
||||
reflect.Value
|
||||
}
|
||||
|
||||
func (m TmplMap) Keys() ([]reflect.Value) {
|
||||
return m.MapKeys()
|
||||
}
|
||||
|
||||
func (m TmplMap) MapWithDefaults(arg0 reflect.Value, args ...reflect.Value) (v reflect.Value) {
|
||||
if iarg0 := indirectInterface(arg0); evalTrue(iarg0) {
|
||||
if v = m.MapIndex(iarg0); evalTrue(v) {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
for _, v = range args {
|
||||
if truth(v) {
|
||||
break
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m TmplMap) Map(arg0 reflect.Value, args ...reflect.Value) (v reflect.Value) {
|
||||
if iarg0 := indirectInterface(arg0); evalTrue(iarg0) {
|
||||
if v = m.MapIndex(iarg0); evalTrue(v) {
|
||||
return
|
||||
}
|
||||
}
|
||||
argLen := len(args)
|
||||
if argLen == 0 {
|
||||
return
|
||||
}
|
||||
keys, defaultValue := args[:argLen-1], args[argLen-1]
|
||||
for _, k := range keys {
|
||||
if ik := indirectInterface(k); !evalTrue(ik) {
|
||||
} else if v = m.Value.MapIndex(ik); evalTrue(v) {
|
||||
return
|
||||
}
|
||||
}
|
||||
v = defaultValue
|
||||
return
|
||||
}
|
||||
|
||||
func MakeMapFunc(arg ...reflect.Value) TmplMap {
|
||||
if len(arg)%2 != 0 {
|
||||
panic("bad makemap")
|
||||
}
|
||||
|
||||
m := reflect.MakeMapWithSize(reflect.TypeOf(map[interface{}]interface{}{}), len(arg)/2)
|
||||
for i := 0; i < len(arg); i += 2 {
|
||||
k, v := indirectInterface(arg[i]), indirectInterface(arg[i+1])
|
||||
if !(evalTrue(k) && evalTrue(v)) {
|
||||
continue
|
||||
}
|
||||
m.SetMapIndex(k, v)
|
||||
}
|
||||
return TmplMap{m}
|
||||
}
|
|
@ -0,0 +1 @@
|
|||
// make compiler happy
|
Reference in New Issue