LINQ-to-SQL: Generic Primary Key function

Introduction

Looking around the web today for a method to generically update a property from a datagrid, using LINQ, I finally found what I was looking for, at

http://csainty.blogspot.com/2008/04/linq-to-sql-generic-primary-key.html

Since it took a little while to convert to VB.NET, I figured I would post it here for others to use.

Using the code

Since I'm not yet a very experienced writer, I'll leave it for the reader to make sense of where and how best to use this code. In a nutshell, this code will work in concert with LINQ to SQL to allow you to retrieve one business entity for whatever purpose needed.

Create your new module called DataContextHelpers, as in the following.

Imports System.Runtime.CompilerServices

Imports System
Imports System.Data
Imports System.Configuration
Imports System.Linq
Imports System.Web
Imports System.Web.Security
Imports System.Web.UI
Imports System.Web.UI.HtmlControls
Imports System.Web.UI.WebControls
Imports System.Web.UI.WebControls.WebParts
Imports System.Xml.Linq
Imports System.Data.Linq
Imports System.Linq.Expressions

Public Module DataContextHelpers

     _
    Public Function GetByPk(Of T As Class)(ByVal context As DataContext, _
                                           ByVal pk As Object) As T
        Dim table = context.GetTable(Of T)()
        Dim mapping = context.Mapping.GetTable(GetType(T))
        Dim pkfield = mapping.RowType.DataMembers.SingleOrDefault(Function(d) d.IsPrimaryKey)

        If (pkfield Is Nothing) Then
            Throw New Exception(String.Format("Table {0} does not contain a Primary Key field", _
                                mapping.TableName))
        End If

        Dim param = Expression.Parameter(GetType(T), "e")
        Dim predicate = Expression.Lambda(Of Func(Of T, Boolean)) _
                        (Expression.Equal(Expression.Property(param, pkfield.Name), _
                                          Expression.Constant(pk)), New ParameterExpression() {param})
        Return table.SingleOrDefault(predicate)
    End Function


End Module

To use this in your code, simply:

Imports yournamespace.DataContextHelpers 

and

Dim thisBusinessEntity As businessEntity = db.GetByPk(Of businessEntity )(key) 

Note that even with the import, VS 2008 still didn't give me full intellisense when I was working on "Common" intellisense mode. It did show up in "All" mode.

Also make sure your project is set to use .NET 3.5, and you have added all the usual references. system.data.linq being the big one to get this to compile.